Add unit test for Pool Output/Source

This uses the new TestSourceProcessor and changes in TestProcessor to support OutputModules.
cms-sw · Jul 17, 2024 · d08336b · d08336b
1 parent 8b9b9fa
commit d08336b
Show file tree

Hide file tree

Showing 3 changed files with 152 additions and 0 deletions.
diff --git a/IOPool/Common/test/BuildFile.xml b/IOPool/Common/test/BuildFile.xml
@@ -1,3 +1,7 @@
 <test name="TestEdmFastMerge" command="TestEdmFastMerge.sh"/>
 <test name="TestEdmConfigDump" command="run_TestEdmConfigDump.sh"/>
 <test name="testIOPoolCommonLumiOverlap" command="test_overlap_lumi_fast_copy.sh"/>
+<bin file="test_catch2_*.cc" name="TestIOPoolCommonTP">
+  <use name="FWCore/TestProcessor"/>
+  <use name="catch2"/>
+</bin>
diff --git a/IOPool/Common/test/test_catch2_main.cc b/IOPool/Common/test/test_catch2_main.cc
@@ -0,0 +1,2 @@
+#define CATCH_CONFIG_MAIN
+#include "catch.hpp"
diff --git a/IOPool/Common/test/test_catch2_output2input.cc b/IOPool/Common/test/test_catch2_output2input.cc
@@ -0,0 +1,146 @@
+#include "FWCore/TestProcessor/interface/TestProcessor.h"
+#include "FWCore/TestProcessor/interface/TestSourceProcessor.h"
+#include "FWCore/Utilities/interface/Exception.h"
+
+#include "DataFormats/TestObjects/interface/Thing.h"
+#include <vector>
+#include <filesystem>
+#include "catch.hpp"
+
+static constexpr auto s_tag = "[PoolOutputSource]";
+
+namespace {
+  std::string setOutputFile(std::string const& iConfig, std::string const& iFileName) {
+    using namespace std::string_literals;
+    return iConfig + "\nprocess.out.fileName = '"s + iFileName + "'\n";
+  }
+
+  std::string setInputFile(std::string const& iConfig, std::string const& iFileName) {
+    using namespace std::string_literals;
+    return iConfig + "\nprocess.source.fileNames = ['file:"s + iFileName + "']\n";
+  }
+}  // namespace
+TEST_CASE("Tests of PoolOuput -> PoolSource", s_tag) {
+  const std::string baseOutConfig{
+      R"_(from FWCore.TestProcessor.TestProcess import *
+process = TestProcess()
+process.out = cms.OutputModule('PoolOutputModule',
+                      fileName = cms.untracked.string('')
+)
+process.add_(cms.Service("InitRootHandlers"))
+process.add_(cms.Service("JobReportService"))
+
+process.moduleToTest(process.out)
+)_"};
+
+  const std::string baseSourceConfig{
+      R"_(from FWCore.TestProcessor.TestSourceProcess import *
+process = TestSourceProcess()
+process.source = cms.Source("PoolSource", fileNames = cms.untracked.vstring(''))
+process.add_(cms.Service("InitRootHandlers"))
+process.add_(cms.Service("SiteLocalConfigService"))
+process.add_(cms.Service("JobReportService"))
+    )_"};
+
+  SECTION("OneEmptyEvent") {
+    const std::string fileName = "one_event.root";
+    {
+      auto configString = setOutputFile(baseOutConfig, fileName);
+
+      edm::test::TestProcessor::Config config{configString};
+
+      edm::test::TestProcessor tester(config);
+      tester.test();
+    }
+    {
+      auto config = setInputFile(baseSourceConfig, fileName);
+      edm::test::TestSourceProcessor tester(config);
+
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n == edm::InputSource::ItemType::IsFile);
+        auto f = tester.openFile();
+        REQUIRE(bool(f));
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsRun);
+        auto r = tester.readRun();
+        REQUIRE(r.run() == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsLumi);
+        auto r = tester.readLuminosityBlock();
+        REQUIRE(r.run() == 1);
+        REQUIRE(r.luminosityBlock() == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsEvent);
+        auto r = tester.readEvent();
+        REQUIRE(r.run() == 1);
+        REQUIRE(r.luminosityBlock() == 1);
+        REQUIRE(r.event() == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsStop);
+      }
+    }
+    std::filesystem::remove(fileName);
+  }
+
+  SECTION("EventWithThing") {
+    const std::string fileName = "thing.root";
+    {
+      auto configString = setOutputFile(baseOutConfig, fileName);
+
+      edm::test::TestProcessor::Config config{configString};
+      auto thingToken = config.produces<std::vector<edmtest::Thing>>("thing");
+
+      edm::test::TestProcessor tester(config);
+      tester.test(std::make_pair(thingToken, std::make_unique<std::vector<edmtest::Thing>>(1, edmtest::Thing{1})));
+    }
+    {
+      auto config = setInputFile(baseSourceConfig, fileName);
+      edm::test::TestSourceProcessor tester(config);
+
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n == edm::InputSource::ItemType::IsFile);
+        auto f = tester.openFile();
+        REQUIRE(bool(f));
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsRun);
+        auto r = tester.readRun();
+        REQUIRE(r.run() == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsLumi);
+        auto r = tester.readLuminosityBlock();
+        REQUIRE(r.run() == 1);
+        REQUIRE(r.luminosityBlock() == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsEvent);
+        auto r = tester.readEvent();
+        REQUIRE(r.run() == 1);
+        REQUIRE(r.luminosityBlock() == 1);
+        REQUIRE(r.event() == 1);
+        auto v = r.get<std::vector<edmtest::Thing>>("thing", "", "TEST");
+        REQUIRE(v->size() == 1);
+        REQUIRE((*v)[0].a == 1);
+      }
+      {
+        auto n = tester.findNextTransition();
+        REQUIRE(n.itemType() == edm::InputSource::ItemType::IsStop);
+      }
+    }
+    std::filesystem::remove(fileName);
+  }
+}