Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
#include "FWCore/Catalog/interface/SiteLocalConfig.h"
#include "FWCore/Services/src/SiteLocalConfigService.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/Exception.h"

#define CATCH_CONFIG_MAIN
#include "catch.hpp"

TEST_CASE("Test SiteLocalConfigService", "[sitelocalconfig]") {
  std::string dirString;
  auto dir = std::getenv("LOCAL_TEST_DIR");
  if (dir) {
    dirString = dir;
  } else {
    auto base_dir = std::getenv("CMSSW_BASE");
    if (base_dir) {
      dirString = base_dir;
      dirString += "/src/FWCore/Services/test";
    }
  }
  REQUIRE(not dirString.empty());
  SECTION("sitelocalconfig/full/site-local-config.xml") {
    edm::ParameterSet pset;
    pset.addUntrackedParameter<std::string>("siteLocalConfigFileUrl",
                                            dirString + "/sitelocalconfig/full/site-local-config.xml");

    edm::service::SiteLocalConfigService slc(pset);

    CHECK(slc.trivialDataCatalogs()[0] == "trivialcatalog_file:/dummy/storage.xml?protocol=dcap");
    edm::CatalogAttributes tmp("DUMMY", "DUMMY_SUB_SITE", "DUMMY_CROSS_SITE", "CMSXrootdFederation", "XRootD");
    CHECK(slc.dataCatalogs()[0].site == tmp.site);
    CHECK(slc.dataCatalogs()[0].subSite == tmp.subSite);
    CHECK(slc.dataCatalogs()[0].storageSite == tmp.storageSite);
    CHECK(slc.dataCatalogs()[0].volume == tmp.volume);
    CHECK(slc.dataCatalogs()[0].protocol == tmp.protocol);
    REQUIRE(slc.sourceCacheTempDir() != nullptr);
    CHECK(*slc.sourceCacheTempDir() == "/a/b/c");
    CHECK(slc.sourceCacheMinFree() == nullptr);
    REQUIRE(slc.sourceCacheHint() != nullptr);
    CHECK(*slc.sourceCacheHint() == "application-only");
    REQUIRE(slc.sourceCloneCacheHint() == nullptr);
    REQUIRE(slc.sourceReadHint() != nullptr);
    CHECK(*slc.sourceReadHint() == "read-ahead-buffered");
    REQUIRE(slc.sourceTTreeCacheSize() != nullptr);
    CHECK(*slc.sourceTTreeCacheSize() == 10000);
    REQUIRE(slc.sourceTimeout() != nullptr);
    CHECK(*slc.sourceTimeout() == 7200);
    CHECK(slc.enablePrefetching());
    REQUIRE(slc.sourceNativeProtocols() != nullptr);
    REQUIRE(slc.sourceNativeProtocols()->size() == 2);
    CHECK((*slc.sourceNativeProtocols())[0] == "dcache");
    CHECK((*slc.sourceNativeProtocols())[1] == "file");
    REQUIRE(slc.statisticsDestination() != nullptr);
    REQUIRE(slc.statisticsInfo() != nullptr);
    CHECK(slc.statisticsInfo()->size() == 1);
    CHECK(slc.statisticsInfo()->find("dn") != slc.statisticsInfo()->end());
    CHECK(slc.siteName() == "DUMMY");
    CHECK(slc.subSiteName() == "DUMMY_SUB_SITE");
    REQUIRE(slc.useLocalConnectString() == true);
    REQUIRE(slc.localConnectPrefix() == "Test:Prefix");
    REQUIRE(slc.localConnectSuffix() == "Test.Suffix");
  }

  SECTION("overrides") {
    edm::ParameterSet pset;
    pset.addUntrackedParameter<std::string>("siteLocalConfigFileUrl",
                                            dirString + "/sitelocalconfig/full/site-local-config.xml");

    pset.addUntrackedParameter<std::string>("overrideSourceCacheTempDir", "/a/d");
    pset.addUntrackedParameter<double>("overrideSourceCacheMinFree", 10.);
    pset.addUntrackedParameter<std::string>("overrideSourceCacheHintDir", "dummy");
    pset.addUntrackedParameter<std::string>("overrideSourceCloneCacheHintDir", "foo");
    pset.addUntrackedParameter<std::string>("overrideSourceReadHint", "read-ahead-unbuffered");
    pset.addUntrackedParameter<std::vector<std::string> >("overrideSourceNativeProtocols", {{"http"}});
    pset.addUntrackedParameter<unsigned int>("overrideSourceTTreeCacheSize", 20000);
    pset.addUntrackedParameter<unsigned int>("overrideSourceTimeout", 0);
    pset.addUntrackedParameter<bool>("overridePrefetching", false);
    pset.addUntrackedParameter<std::string>("overrideStatisticsDestination", "");
    pset.addUntrackedParameter<std::vector<std::string> >("overrideStatisticsInfo", {{"nodn"}});
    pset.addUntrackedParameter<bool>("overrideUseLocalConnectString", false);
    pset.addUntrackedParameter<std::string>("overrideLocalConnectPrefix", "OverridePrefix");
    pset.addUntrackedParameter<std::string>("overrideLocalConnectSuffix", "OverrideSuffix");

    edm::service::SiteLocalConfigService slc(pset);

    CHECK(slc.trivialDataCatalogs()[0] == "trivialcatalog_file:/dummy/storage.xml?protocol=dcap");
    edm::CatalogAttributes tmp("DUMMY", "DUMMY_SUB_SITE", "DUMMY_CROSS_SITE", "CMSXrootdFederation", "XRootD");
    CHECK(slc.dataCatalogs()[0].site == tmp.site);
    CHECK(slc.dataCatalogs()[0].subSite == tmp.subSite);
    CHECK(slc.dataCatalogs()[0].storageSite == tmp.storageSite);
    CHECK(slc.dataCatalogs()[0].volume == tmp.volume);
    CHECK(slc.dataCatalogs()[0].protocol == tmp.protocol);
    REQUIRE(slc.sourceCacheTempDir() != nullptr);
    CHECK(*slc.sourceCacheTempDir() == "/a/d");
    REQUIRE(slc.sourceCacheMinFree() != nullptr);
    CHECK(*slc.sourceCacheMinFree() == 10.);
    REQUIRE(slc.sourceCacheHint() != nullptr);
    CHECK(*slc.sourceCacheHint() == "dummy");
    REQUIRE(slc.sourceCloneCacheHint() != nullptr);
    CHECK(*slc.sourceCloneCacheHint() == "foo");
    REQUIRE(slc.sourceReadHint() != nullptr);
    CHECK(*slc.sourceReadHint() == "read-ahead-unbuffered");
    REQUIRE(slc.sourceTTreeCacheSize() != nullptr);
    CHECK(*slc.sourceTTreeCacheSize() == 20000);
    REQUIRE(slc.sourceTimeout() != nullptr);
    CHECK(*slc.sourceTimeout() == 0);
    CHECK(not slc.enablePrefetching());
    REQUIRE(slc.sourceNativeProtocols() != nullptr);
    REQUIRE(slc.sourceNativeProtocols()->size() == 1);
    CHECK((*slc.sourceNativeProtocols())[0] == "http");
    REQUIRE(slc.statisticsDestination() == nullptr);
    REQUIRE(slc.statisticsInfo() != nullptr);
    CHECK(slc.statisticsInfo()->size() == 1);
    CHECK(slc.statisticsInfo()->find("nodn") != slc.statisticsInfo()->end());
    CHECK(slc.siteName() == "DUMMY");
    CHECK(slc.subSiteName() == "DUMMY_SUB_SITE");
    REQUIRE(slc.useLocalConnectString() == false);
    REQUIRE(slc.localConnectPrefix() == "OverridePrefix");
    REQUIRE(slc.localConnectSuffix() == "OverrideSuffix");
  }

  SECTION("sitelocalconfig/throw/site-local-config.xml") {
    edm::ParameterSet pset;
    pset.addUntrackedParameter<std::string>("siteLocalConfigFileUrl",
                                            dirString + "/sitelocalconfig/throw/site-local-config.xml");

    REQUIRE_THROWS_AS(edm::service::SiteLocalConfigService(pset), cms::Exception);
  }
}