mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-03 21:44:30 +08:00
2957ab4576
* switch to catch2 v3 Fixes #6894 * fix a couple of problems noticed in the CI builds * more warning cleanup * changes in response to review
126 lines
3.7 KiB
C++
126 lines
3.7 KiB
C++
//
|
|
// Copyright (C) 2022 Greg Landrum
|
|
//
|
|
// @@ All Rights Reserved @@
|
|
// This file is part of the RDKit.
|
|
// The contents are covered by the terms of the BSD license
|
|
// which is included in the file license.txt, found at the root
|
|
// of the RDKit source tree.
|
|
//
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <catch2/catch_all.hpp>
|
|
#include "RDLog.h"
|
|
|
|
TEST_CASE("LogStateSetter") {
|
|
RDLog::RDLoggerList allLogs({rdErrorLog, rdWarningLog, rdInfoLog});
|
|
SECTION("disable all") {
|
|
for (auto strm : allLogs) {
|
|
// explicitly enable the stream so that we know something is happening
|
|
std::stringstream ostrm;
|
|
strm->df_enabled = true;
|
|
strm->SetTee(ostrm);
|
|
{
|
|
RDLog::LogStateSetter disabler;
|
|
BOOST_LOG(strm) << "should be silent" << std::endl;
|
|
auto txt = ostrm.str();
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
}
|
|
BOOST_LOG(strm) << "should not be silent" << std::endl;
|
|
auto txt = ostrm.str();
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
strm->ClearTee();
|
|
}
|
|
}
|
|
SECTION("enable one") {
|
|
for (auto strm : allLogs) {
|
|
strm->df_enabled = true;
|
|
RDLog::LogStateSetter disabler;
|
|
{
|
|
RDLog::LogStateSetter justone(RDLog::RDLoggerList({strm}));
|
|
for (auto strm2 : allLogs) {
|
|
std::stringstream ostrm;
|
|
strm2->SetTee(ostrm);
|
|
BOOST_LOG(strm2) << "should not be silent" << std::endl;
|
|
auto txt = ostrm.str();
|
|
if (strm == strm2) {
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
} else {
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
}
|
|
|
|
strm2->ClearTee();
|
|
}
|
|
}
|
|
|
|
// should be disabled again
|
|
std::stringstream ostrm;
|
|
strm->SetTee(ostrm);
|
|
BOOST_LOG(strm) << "should be silent" << std::endl;
|
|
auto txt = ostrm.str();
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
strm->ClearTee();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST_CASE("GitHub Issue #5172", "[bug][logging]") {
|
|
std::stringstream err_ostrm;
|
|
std::stringstream warn_ostrm;
|
|
rdErrorLog->df_enabled = true;
|
|
rdWarningLog->df_enabled = true;
|
|
rdErrorLog->SetTee(err_ostrm);
|
|
rdWarningLog->SetTee(warn_ostrm);
|
|
|
|
{
|
|
RDLog::LogStateSetter disabler;
|
|
|
|
BOOST_LOG(rdErrorLog) << "should be silent" << std::endl;
|
|
auto txt = err_ostrm.str();
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
|
|
BOOST_LOG(rdWarningLog) << "should be silent" << std::endl;
|
|
txt = warn_ostrm.str();
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
|
|
{
|
|
// The second setter overrides the first one:
|
|
RDLog::LogStateSetter disabler2({rdWarningLog});
|
|
|
|
BOOST_LOG(rdErrorLog) << "should be silent" << std::endl;
|
|
txt = err_ostrm.str();
|
|
CHECK(txt.find("should") == std::string::npos);
|
|
|
|
BOOST_LOG(rdWarningLog) << "should not be silent" << std::endl;
|
|
txt = warn_ostrm.str();
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
warn_ostrm.clear();
|
|
}
|
|
}
|
|
|
|
// Both setters are destroyed, and we revert to initial state
|
|
|
|
BOOST_LOG(rdErrorLog) << "should not be silent" << std::endl;
|
|
auto txt = err_ostrm.str();
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
|
|
BOOST_LOG(rdWarningLog) << "should not be silent" << std::endl;
|
|
txt = warn_ostrm.str();
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
|
|
rdErrorLog->ClearTee();
|
|
rdWarningLog->ClearTee();
|
|
}
|
|
|
|
TEST_CASE("Tee to file") {
|
|
const std::string filename = "error_log.txt";
|
|
rdErrorLog->SetTee(filename);
|
|
BOOST_LOG(rdErrorLog) << "should not be silent" << std::endl;
|
|
std::ifstream istrm(filename);
|
|
std::string txt;
|
|
CHECK(std::getline(istrm, txt));
|
|
CHECK(txt.find("should") != std::string::npos);
|
|
}
|