Add a flag and FileReader in a test

Author: durswd

src_test/TestHelper.cpp

@@ -9,11 +9,48 @@
 #include <regex>
 #include <string>
 
+namespace
+{
+class DefaultTestFileReader : public TestFileReader
+{
+public:
+	std::vector<uint8_t> Read(const char* path) override
+	{
+		std::vector<uint8_t> ret;
+
+#ifdef _WIN32
+		FILE* fp = nullptr;
+		fopen_s(&fp, path, "rb");
+
+#else
+		FILE* fp = fopen(path, "rb");
+#endif
+
+		if (fp == nullptr)
+		{
+			std::cout << "Not found : " << path << std::endl;
+			return ret;
+		}
+
+		fseek(fp, 0, SEEK_END);
+		auto size = ftell(fp);
+		fseek(fp, 0, SEEK_SET);
+
+		ret.resize(size);
+		fread(ret.data(), 1, size, fp);
+		fclose(fp);
+
+		return ret;
+	}
+};
+} // namespace
+
 struct InternalTestHelper
 {
 	std::string Root;
 	bool IsCaptureRequired = false;
 	std::map<std::string, std::function<void(LLGI::DeviceType)>> tests;
+	std::shared_ptr<TestFileReader> FileReader;
 };
 
 std::shared_ptr<InternalTestHelper> TestHelper::Get()
@@ -66,6 +103,18 @@ ParsedArgs TestHelper::ParseArg(int argc, char* argv[])
 	return args;
 }
 
+std::shared_ptr<TestFileReader> TestHelper::GetFileReader()
+{
+	auto helper = Get();
+	if (helper->FileReader == nullptr)
+	{
+		helper->FileReader = std::make_shared<DefaultTestFileReader>();
+	}
+	return helper->FileReader;
+}
+
+void TestHelper::SetFileReader(std::shared_ptr<TestFileReader> fileReader) { Get()->FileReader = fileReader; }
+
 std::vector<uint8_t> TestHelper::CreateDummyTextureData(LLGI::Vec2I size, LLGI::TextureFormatType format)
 {
 	std::vector<uint8_t> ret;
@@ -149,34 +198,7 @@ std::vector<uint8_t> TestHelper::LoadData(const char* path)
 	return LoadDataWithoutRoot(path_.c_str());
 }
 
-std::vector<uint8_t> TestHelper::LoadDataWithoutRoot(const char* path)
-{
-	std::vector<uint8_t> ret;
-
-#ifdef _WIN32
-	FILE* fp = nullptr;
-	fopen_s(&fp, path, "rb");
-
-#else
-	FILE* fp = fopen(path, "rb");
-#endif
-
-	if (fp == nullptr)
-	{
-		std::cout << "Not found : " << path << std::endl;
-		return ret;
-	}
-
-	fseek(fp, 0, SEEK_END);
-	auto size = ftell(fp);
-	fseek(fp, 0, SEEK_SET);
-
-	ret.resize(size);
-	fread(ret.data(), 1, size, fp);
-	fclose(fp);
-
-	return ret;
-}
+std::vector<uint8_t> TestHelper::LoadDataWithoutRoot(const char* path) { return GetFileReader()->Read(path); }
 
 void TestHelper::SetRoot(const char* root)
 {

src_test/TestHelper.h

@@ -16,12 +16,22 @@ struct ParsedArgs
 	std::string Filter;
 };
 
+class TestFileReader
+{
+public:
+	virtual ~TestFileReader() = default;
+	virtual std::vector<uint8_t> Read(const char* path) = 0;
+};
+
 class TestHelper
 {
 private:
 	static std::shared_ptr<InternalTestHelper> Get();
+	static std::shared_ptr<TestFileReader> GetFileReader();
 
 public:
+	static void SetFileReader(std::shared_ptr<TestFileReader> fileReader);
+
 	static ParsedArgs ParseArg(int argc, char* argv[]);
 
 	static std::vector<uint8_t> CreateDummyTextureData(LLGI::Vec2I size, LLGI::TextureFormatType format);

src_test/capture.cpp

@@ -188,7 +188,9 @@ void test_capture(LLGI::DeviceType deviceType, LLGI::Vec2I windowSize)
 	pips.clear();
 }
 
-#if defined(__linux__) || defined(__APPLE__) || defined(WIN32)
+#ifndef _DISABLE_CAPTURE_TEST
+
+#if defined(__linux__) || defined(__APPLE__) || defined(_WIN32)
 
 TestRegister Capture_Size1279("Capture.Size1279", [](LLGI::DeviceType device) -> void { test_capture(device, LLGI::Vec2I(1279, 719)); });
 
@@ -199,3 +201,5 @@ TestRegister Capture_Size1280("Capture.Size1280", [](LLGI::DeviceType device) ->
 TestRegister Capture_Texture("Capture.Texture", [](LLGI::DeviceType device) -> void { test_capture_texture(device); });
 
 #endif
+
+#endif

src_test/main.cpp

@@ -1,3 +1,4 @@
+#ifndef _DISABLE_MAIN
 
 #include "TestHelper.h"
 #include "test.h"
@@ -13,7 +14,7 @@
 
 #endif
 
-#if defined(__linux__) || defined(__APPLE__) || defined(WIN32)
+#if defined(__linux__) || defined(__APPLE__) || defined(_WIN32)
 
 int main(int argc, char* argv[])
 {
@@ -67,3 +68,5 @@ int main(int argc, char* argv[])
 	return 0;
 }
 #endif
+
+#endif