test_main.cc

Go to the documentation of this file.

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/

 
#include "gtest/gtest.h"
#include "framework/core/RandomNumbers.h"
#include <framework/logging/LogSystem.h>
#include <framework/logging/LogConfig.h>
#include <framework/dbobjects/MagneticField.h>
#include <framework/dbobjects/MagneticFieldComponentConstant.h>
#include <mdst/dbobjects/CollisionInvariantMass.h>
#include <mdst/dbobjects/CollisionBoostVector.h>
#include <mdst/dbobjects/CollisionAxisCMS.h>
#include <mdst/dbobjects/BeamSpot.h>
#include <framework/database/DBStore.h>
 
namespace {
  class TestEventListener: public ::testing::EmptyTestEventListener {
  public:
    explicit TestEventListener(bool debug): ::testing::EmptyTestEventListener(), m_enableDebug{debug} {}
  private:
    virtual void OnTestStart(const ::testing::TestInfo& test) final override
    {
      if (m_enableDebug) {
        Belle2::LogSystem::Instance().getLogConfig()->setDebugLevel(10000);
        Belle2::LogSystem::Instance().getLogConfig()->setLogLevel(Belle2::LogConfig::c_Debug);
      }
      std::string name = test.test_case_name();
      name += "/";
      name += test.name();
      Belle2::RandomNumbers::initialize(name);
      Belle2::MagneticField* field = new Belle2::MagneticField();
      field->addComponent(new Belle2::MagneticFieldComponentConstant({0, 0, 1.5 * Belle2::Unit::T}));
      Belle2::DBStore::Instance().addConstantOverride("MagneticField", field, false);
 
      // These 6 collision parameters correspond to following HER & LER momenta
      // pHER = (0.290582573157898, 0, 6.99796952744559,  7.004)
      // pLER = (0.166035330010433, 0, -3.99855423973071, 4.002)
      // These are in BeamParameters (rev=59449 & rev=141509) which is default in main GT
 
      auto* collisionInvariantMass = new Belle2::CollisionInvariantMass();
      collisionInvariantMass->setMass(10.579557836806245064, 0, 0);
      Belle2::DBStore::Instance().addConstantOverride("CollisionInvariantMass", collisionInvariantMass);
 
      auto* collisionBoostVector = new Belle2::CollisionBoostVector();
      collisionBoostVector->setBoost(TVector3(0.041488088603337358595, 0, 0.27252546681036521337), TMatrixDSym(3));
      Belle2::DBStore::Instance().addConstantOverride("CollisionBoostVector", collisionBoostVector);
 
      auto* collisionAxisCMS = new Belle2::CollisionAxisCMS();
      collisionAxisCMS->setAngles(0.0057699660560828307243, 0, TMatrixDSym(2));
      collisionAxisCMS->setSpread(TMatrixDSym(2), 0, 0, 0);
      Belle2::DBStore::Instance().addConstantOverride("CollisionAxisCMS", collisionAxisCMS);
 
      // "dummy" beam spot with center at (0,0,0)
      auto* beamSpot = new Belle2::BeamSpot();
      Belle2::DBStore::Instance().addConstantOverride("BeamSpot", beamSpot);
 
    }
    virtual void OnTestEnd(const ::testing::TestInfo&) final override
    {
      Belle2::LogSystem::Instance().resetLogging();
    }

    bool m_enableDebug;
  };
}

int main(int argc, char** argv)
{
  ::testing::InitGoogleTest(&argc, argv);
  ::testing::UnitTest& unit_test = *::testing::UnitTest::GetInstance();
  bool debug{false};
  for (int i = 1; i < argc; ++i) {
    if (std::string(argv[i]) == "--debug") debug = true;
  }
  unit_test.listeners().Append(new TestEventListener(debug));
  return RUN_ALL_TESTS();
}