mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-07 22:44:25 +08:00
3b2916c9a4
* add test * fix LogStateSetter * fix BlockLogs() usage in Draw module * allow BlockLogs() as a contestmgr * move test to more suitable location * guarantee log restoration on __exit__ * skip test if logs cannot be grabbed before block
113 lines
3.3 KiB
C++
113 lines
3.3 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 <sstream>
|
|
#include "catch.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();
|
|
} |