mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-21 01:04:57 +08:00
ee0648c365
restored the possibility of different writing modes in the FileLogger mocked writeToFile in TestFileLogger Related-To: NEO-7006 Signed-off-by: Marcel Skierkowski <marcel.skierkowski@intel.com>
73 lines
2.1 KiB
C++
73 lines
2.1 KiB
C++
/*
|
|
* Copyright (C) 2022-2024 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/helpers/string_helpers.h"
|
|
#include "shared/source/utilities/directory.h"
|
|
#include "shared/source/utilities/logger.h"
|
|
#include "shared/test/common/helpers/mock_file_io.h"
|
|
|
|
#include <map>
|
|
|
|
namespace NEO {
|
|
extern std::map<std::string, std::stringstream> virtualFileList;
|
|
} // namespace NEO
|
|
|
|
template <DebugFunctionalityLevel debugLevel>
|
|
class TestFileLogger : public NEO::FileLogger<debugLevel> {
|
|
public:
|
|
using NEO::FileLogger<debugLevel>::FileLogger;
|
|
|
|
TestFileLogger(std::string filename, const NEO::DebugVariables &flags) : NEO::FileLogger<debugLevel>(filename, flags) {
|
|
if (NEO::FileLogger<debugLevel>::enabled() && virtualFileExists(this->getLogFileName())) {
|
|
removeVirtualFile(this->getLogFileName());
|
|
}
|
|
}
|
|
|
|
~TestFileLogger() override {
|
|
if (virtualFileExists(this->getLogFileName())) {
|
|
removeVirtualFile(this->getLogFileName());
|
|
}
|
|
}
|
|
|
|
void useRealFiles(bool value) {
|
|
mockFileSystem = !value;
|
|
}
|
|
|
|
void writeToFile(std::string filename,
|
|
const char *str,
|
|
size_t length,
|
|
std::ios_base::openmode mode) override {
|
|
|
|
if (mockFileSystem == false) {
|
|
NEO::FileLogger<debugLevel>::writeToFile(filename, str, length, mode);
|
|
return;
|
|
}
|
|
writeDataToFile(filename.c_str(), str, length);
|
|
}
|
|
|
|
int32_t createdFilesCount() {
|
|
return static_cast<int32_t>(NEO::virtualFileList.size());
|
|
}
|
|
|
|
bool wasFileCreated(std::string filename) {
|
|
return virtualFileExists(filename);
|
|
}
|
|
|
|
std::string getFileString(std::string filename) {
|
|
return NEO::virtualFileList[filename].str();
|
|
}
|
|
|
|
protected:
|
|
bool mockFileSystem = false;
|
|
};
|
|
|
|
using FullyEnabledFileLogger = TestFileLogger<DebugFunctionalityLevel::full>;
|
|
using FullyDisabledFileLogger = TestFileLogger<DebugFunctionalityLevel::none>;
|
|
using ReleaseInternalFileLogger = TestFileLogger<DebugFunctionalityLevel::regKeys>;
|