TestTasks.h

Go to the documentation of this file.

#include "FairTask.h"
#include "FairTimeStamp.h"

#include "PndInputContainer.h"
#include "PndOutputContainer.h"
// #include "PndPersistencyTask.h" // for PndPersistencyTask
#include "PndSTDOutputContainer.h"
#include "PndTCAOutputContainer.h"

#include "gtest/gtest.h" // for AssertHelper, EXPECT_EQ, EXPECT_TRUE

//______________________________________________________________________________
class TestInTask : public FairTask {
  PndInputContainer<FairTimeStamp> fInCont;
  Int_t fgNumber{111};
  TString fBranchName;

 public:
  TestInTask(const TString &t_branch = "TestBranch") : FairTask(), fBranchName(t_branch){};
  virtual ~TestInTask(){};

  virtual InitStatus Init() /*override*/ final
  {
    EXPECT_TRUE(fInCont.Init(fBranchName));
    return kSUCCESS;
  };

  virtual void Exec(Option_t * /*option*/) /*override*/
  {
    ASSERT_EQ(fInCont.GetSize(), 1);
    EXPECT_EQ(fInCont.GetConstElementPtr(0)->GetTimeStamp(), fgNumber--);
  };

 private:
  // ClassDef(TestInTask, 1);
};

//______________________________________________________________________________
class TestOutTask : public FairTask {
  PndOutputContainer<FairTimeStamp> fOutCont;
  Int_t fgNumber{111};
  TString fBranchName;

 public:
  TestOutTask(const TString &t_branch = "TestBranch") : FairTask(), fBranchName(t_branch) {}
  virtual ~TestOutTask() {}

  virtual InitStatus Init() /*override*/ final
  {
    EXPECT_TRUE(fOutCont.Init(fBranchName));
    return kSUCCESS;
  }

  virtual void Exec(Option_t * /*option*/) /*override*/
  {
    fOutCont.Reset();
    FairTimeStamp stamp;
    stamp.SetTimeStamp(fgNumber--);
    fOutCont.CreateCopy(stamp);
  }
  // ClassDef(TestOutTask, 1);
};

//______________________________________________________________________________
class TestOutWithSTDTask : public FairTask {
  PndSTDOutputContainer<FairTimeStamp> fOutCont;
  Int_t fgNumber{111};
  TString fBranchName;

 public:
  TestOutWithSTDTask(const TString &t_branch = "TestBranch") : FairTask(), fBranchName(t_branch) {}
  virtual ~TestOutWithSTDTask() {}

  virtual InitStatus Init() /*override*/ final
  {
    EXPECT_TRUE(fOutCont.Init(fBranchName));
    return kSUCCESS;
  }

  virtual void Exec(Option_t * /*option*/) /*override*/
  {
    fOutCont.Reset();
    FairTimeStamp stamp;
    stamp.SetTimeStamp(fgNumber--);
    fOutCont.CreateCopy(stamp);
  }
  // ClassDef(TestOutWithSTDTask, 1);
};

//______________________________________________________________________________
class TestOutWithTCATask : public FairTask {
  PndTCAOutputContainer<FairTimeStamp> fOutCont;
  Int_t fgNumber{111};
  TString fBranchName;

 public:
  TestOutWithTCATask(const TString &t_branch = "TestBranch") : FairTask(), fBranchName(t_branch) {}

  virtual ~TestOutWithTCATask() {}

  virtual InitStatus Init() /*override*/ final
  {
    EXPECT_TRUE(fOutCont.Init(fBranchName));
    return kSUCCESS;
  }

  virtual void Exec(Option_t * /*option*/) /*override*/
  {
    fOutCont.Reset();
    FairTimeStamp stamp;
    stamp.SetTimeStamp(fgNumber--);
    fOutCont.CreateCopy(stamp);
  }
  // ClassDef(TestOutWithTCATask, 1);
};

//______________________________________________________________________________
class TestPeekTask : public FairTask {
  PndInputContainer<FairTimeStamp> fInCont;
  Int_t fgNumber{111};
  TString fBranchName;

 public:
  TestPeekTask(const TString &t_branch = "TestBranch") : FairTask(), fBranchName(t_branch) {}
  virtual ~TestPeekTask() {}

  virtual InitStatus Init() /*override*/ final
  {
    EXPECT_TRUE(fInCont.Init(fBranchName));
    return kSUCCESS;
  }

  virtual void Exec(Option_t * /*option*/) /*override*/
  {
    ASSERT_EQ(fInCont.GetSize(), 1);
    EXPECT_EQ(fInCont.GetConstElementPtr(0)->GetTimeStamp(), fgNumber--);
  }
  // ClassDef(TestPeekTask, 1);
};