TestOutput_module.cc

Go to the documentation of this file.

#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Core/OutputModule.h"
#include "cetlib_except/exception.h"
#include "fhiclcpp/types/Atom.h"
#include "fhiclcpp/types/ConfigurationTable.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include <cassert>

namespace arttest {
  class TestOutput;
}

class arttest::TestOutput : public art::OutputModule {
public:
  struct Config {
    fhicl::TableFragment<art::OutputModule::Config> omConfig;
    fhicl::Atom<unsigned> shouldPass{fhicl::Name("shouldPass")};
  };

  using Parameters =
    fhicl::WrappedTable<Config, art::OutputModule::Config::KeysToIgnore>;
  explicit TestOutput(Parameters const&);

private:
  void write(art::EventPrincipal& e) override;
  void
  writeSubRun(art::SubRunPrincipal&) override
  {}
  void
  writeRun(art::RunPrincipal&) override
  {}
  void endJob() override;

  unsigned num_pass_;
  unsigned total_;
};

arttest::TestOutput::TestOutput(arttest::TestOutput::Parameters const& ps)
  : art::OutputModule(ps().omConfig, ps.get_PSet())
  , num_pass_{ps().shouldPass()}
  , total_{0u}
{}

void
arttest::TestOutput::write(art::EventPrincipal&)
{
  ++total_;
}

void
arttest::TestOutput::endJob()
{
  mf::LogAbsolute("TestOutputReport") << "TestOutput: should pass " << num_pass_
                                      << ", did pass " << total_ << "\n";

  if (total_ != num_pass_) {
    throw cet::exception("TestOutputFailure")
      << "Number passed should be " << num_pass_ << ", but got " << total_
      << "\n";
  }
}

DEFINE_ART_MODULE(arttest::TestOutput)