TestPRegisterModule2

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
// -*- C++ -*-
//
// Package:     FWCore/Framework
// Class  :     TestPRegisterModule2
//
/**\class TestPRegisterModule2

 Description:

 Usage:

   \author Stefano ARGIRO, Chris Jones
   \date 19 May 2005
*/

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Provenance/interface/StableProvenance.h"
#include "DataFormats/TestObjects/interface/ToyProducts.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/TypeID.h"

#include "cppunit/extensions/HelperMacros.h"

#include <memory>
#include <vector>

class TestPRegisterModule2 : public edm::global::EDProducer<> {
public:
  explicit TestPRegisterModule2(edm::ParameterSet const&);
  void produce(edm::StreamID, edm::Event&, edm::EventSetup const&) const override;
};

TestPRegisterModule2::TestPRegisterModule2(edm::ParameterSet const&) {
  produces<edmtest::DoubleProduct>();
  consumes<edmtest::StringProduct>(edm::InputTag{"m2"});
}

void TestPRegisterModule2::produce(edm::StreamID, edm::Event& event, edm::EventSetup const&) const {
  std::vector<edm::StableProvenance const*> plist;
  event.getAllStableProvenance(plist);

  std::vector<edm::StableProvenance const*>::const_iterator pd = plist.begin();

  CPPUNIT_ASSERT(0 != plist.size());
  CPPUNIT_ASSERT(2 == plist.size());
  CPPUNIT_ASSERT(pd != plist.end());
  if (pd == plist.end())
    return;  // To silence Coverity
  edmtest::StringProduct stringprod;
  edm::TypeID stringID(stringprod);
  CPPUNIT_ASSERT(stringID.friendlyClassName() == (*pd)->friendlyClassName());
  CPPUNIT_ASSERT((*pd)->moduleLabel() == "m1");

  ++pd;
  CPPUNIT_ASSERT(pd != plist.end());
  if (pd == plist.end())
    return;  // To silence Coverity

  edmtest::DoubleProduct dprod;
  edm::TypeID dID(dprod);
  CPPUNIT_ASSERT(dID.friendlyClassName() == (*pd)->friendlyClassName());
  CPPUNIT_ASSERT((*pd)->moduleLabel() == "m2");

  edm::Handle<edmtest::StringProduct> stringp;
  event.getByLabel("m2", stringp);
  CPPUNIT_ASSERT(stringp->name_ == "m1");

  event.put(std::make_unique<edmtest::DoubleProduct>());
}

DEFINE_FWK_MODULE(TestPRegisterModule2);