test_ObjectStore.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#include <TH1.h>
#include <gsl/span>
#define BOOST_TEST_MODULE Test Utilities MergerObjectStore
#define BOOST_TEST_MAIN
#define BOOST_TEST_DYN_LINK
 
#include "Mergers/ObjectStore.h"
#include "Mergers/CustomMergeableObject.h"
#include "Mergers/CustomMergeableTObject.h"
#include "Headers/DataHeader.h"
#include "Framework/DataRef.h"
 
#include <TMessage.h>
#include <boost/test/unit_test.hpp>
 
#include <memory>
 
using namespace o2::framework;
using namespace o2::mergers;
 
// Simple test to do root deserialization.
BOOST_AUTO_TEST_SUITE(TestObjectExtraction)
 
template <typename TypeToDataRef>
DataRef makeDataRef(TypeToDataRef* obj)
{
  DataRef ref;
  TMessage* tm = new TMessage(kMESS_OBJECT);
  tm->WriteObject(obj);
 
  ref.payload = tm->Buffer();
 
  auto dh = new o2::header::DataHeader{};
  dh->payloadSerializationMethod = o2::header::gSerializationMethodROOT;
  dh->payloadSize = tm->BufferSize();
  ref.header = reinterpret_cast<char const*>(dh->data());
 
  return ref;
}
 
BOOST_AUTO_TEST_CASE(MergeableObject)
{
  auto* obj = new CustomMergeableTObject("obj1", 123);
  DataRef ref = makeDataRef(obj);
 
  auto objStore = object_store_helpers::extractObjectFrom(ref);
  BOOST_REQUIRE(std::holds_alternative<MergeInterfacePtr>(objStore));
 
  auto objExtractedCustom = dynamic_cast<CustomMergeableTObject*>(std::get<MergeInterfacePtr>(objStore).get());
  BOOST_REQUIRE(objExtractedCustom != nullptr);
  BOOST_CHECK_EQUAL(objExtractedCustom->getSecret(), 123);
 
  delete ref.header;
  delete ref.payload;
}
 
BOOST_AUTO_TEST_CASE(NamedMergeableObject)
{
  auto* obj = new CustomMergeableObject(123);
  DataRef ref = makeDataRef(obj);
 
  auto objStore = object_store_helpers::extractObjectFrom(ref);
  BOOST_REQUIRE(std::holds_alternative<MergeInterfacePtr>(objStore));
 
  auto objExtractedCustom = dynamic_cast<CustomMergeableObject*>(std::get<MergeInterfacePtr>(objStore).get());
  BOOST_REQUIRE(objExtractedCustom != nullptr);
  BOOST_CHECK_EQUAL(objExtractedCustom->getSecret(), 123);
 
  delete ref.header;
  delete ref.payload;
}
 
BOOST_AUTO_TEST_CASE(Histo1D)
{
  TH1I* obj = new TH1I("histo", "histo", 100, 0, 100);
  obj->Fill(4);
 
  DataRef ref = makeDataRef(obj);
 
  auto objStore = object_store_helpers::extractObjectFrom(ref);
  BOOST_CHECK(std::holds_alternative<TObjectPtr>(objStore));
 
  auto objExtractedHisto = dynamic_cast<TH1I*>(std::get<TObjectPtr>(objStore).get());
  BOOST_REQUIRE(objExtractedHisto != nullptr);
  BOOST_CHECK_EQUAL(objExtractedHisto->GetEntries(), 1);
 
  delete ref.header;
  delete ref.payload;
  delete obj;
}
 
BOOST_AUTO_TEST_CASE(TArrayOfHisto1D)
{
  TObjArray* array = new TObjArray();
  array->SetOwner(true);
 
  TH1I* histo = new TH1I("histo 1d", "histo 1d", 100, 0, 100);
  histo->Fill(5);
  array->Add(histo);
 
  DataRef ref = makeDataRef(array);
  auto objStore = object_store_helpers::extractObjectFrom(ref);
  BOOST_CHECK(std::holds_alternative<TObjectPtr>(objStore));
 
  auto objExtractedArray = dynamic_cast<TObjArray*>(std::get<TObjectPtr>(objStore).get());
  BOOST_REQUIRE(objExtractedArray != nullptr);
  BOOST_CHECK_EQUAL(objExtractedArray->GetEntries(), 1);
 
  delete ref.header;
  delete ref.payload;
  delete array;
}
 
BOOST_AUTO_TEST_CASE(VectorOfHistos1D)
{
  auto histo = std::make_shared<TH1F>("histo 1d", "histo 1d", 100, 0, 100);
  histo->Fill(5);
  VectorOfTObjectPtrs vectorWithData{histo};
 
  auto vectorToDataRef = object_store_helpers::toRawObserverPointers(vectorWithData);
 
  DataRef ref = makeDataRef(&vectorToDataRef);
  auto objStore = object_store_helpers::extractObjectFrom(ref);
  BOOST_CHECK(std::holds_alternative<VectorOfTObjectPtrs>(objStore));
  auto extractedVector = std::get<VectorOfTObjectPtrs>(objStore);
  BOOST_CHECK(extractedVector.size() == 1);
  auto* extractedHisto = dynamic_cast<TH1F*>(extractedVector[0].get());
  BOOST_CHECK(gsl::span(histo->GetArray(), histo->GetSize()) == gsl::span(extractedHisto->GetArray(), extractedHisto->GetSize()));
}
 
BOOST_AUTO_TEST_SUITE_END()