testCcdbApiMultipleUrls.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.
 
#define BOOST_TEST_MODULE CCDB
#define BOOST_TEST_MAIN
#define BOOST_TEST_DYN_LINK
 
#include "CCDB/CcdbApi.h"
#include "CCDB/CCDBTimeStampUtils.h"
#include <boost/test/unit_test.hpp>
#include <cstdio>
#include <TH1F.h>
 
using namespace std;
using namespace o2::ccdb;
namespace utf = boost::unit_test;
namespace tt = boost::test_tools;
 
static std::string ccdbUrl;
static std::string basePath;
bool hostReachable = false;
 
struct Fixture {
  Fixture()
  {
    CcdbApi api;
    // Deliberately NOT reading ALICEO2_CCDB_HOST like the other suites: this
    // one is skipped in practice, because isHostReachable() cannot report a
    // multi-host pool reachable (see CcdbApi::isHostReachable). Making it
    // configurable only matters once that is fixed and the suite runs.
    ccdbUrl = "https://localhost:22,https://localhost:8080,http://ccdb-test.cern.ch:8080";
    api.init(ccdbUrl);
    cout << "ccdb url: " << ccdbUrl << endl;
    hostReachable = api.isHostReachable();
    cout << "Is host reachable ? --> " << hostReachable << endl;
    basePath = std::string("Test/pid") + getpid() + "/";
    cout << "Path we will use in this test suite : " + basePath << endl;
  }
  ~Fixture()
  {
    if (hostReachable) {
      CcdbApi api;
      std::map<std::string, std::string> metadata;
      api.init(ccdbUrl);
      api.truncate(basePath + "*");
      cout << "Test data truncated (" << basePath << ")" << endl;
    }
  }
};
BOOST_GLOBAL_FIXTURE(Fixture);
 
struct if_reachable {
  tt::assertion_result operator()(utf::test_unit_id)
  {
    return hostReachable;
  }
};
 
struct test_fixture {
  test_fixture()
  {
    api.init(ccdbUrl);
    metadata["Hello"] = "World";
    std::cout << "*** " << boost::unit_test::framework::current_test_case().p_name << " ***" << std::endl;
  }
  ~test_fixture() = default;
 
  CcdbApi api;
  std::map<std::string, std::string> metadata;
};
 
BOOST_AUTO_TEST_CASE(storeAndRetrieve, *utf::precondition(if_reachable()))
{
  test_fixture f;
 
  TH1F h1("th1name", "th1name", 100, 0, 99);
  h1.FillRandom("gaus", 10000);
  BOOST_CHECK_EQUAL(h1.ClassName(), "TH1F");
  cout << "ccdb/TObject/TEST" << endl;
  long timestamp = getCurrentTimestamp();
  f.api.storeAsTFile(&h1, "ccdb/TObject/TEST", f.metadata, timestamp);
 
  TObject* returnResult = f.api.retrieve("ccdb/TObject/TEST/" + to_string(timestamp), f.metadata, timestamp);
  BOOST_CHECK(returnResult != nullptr);
}