From 2f2ea2b21026cf2eb6b50d925484ee0a13bd0d02 Mon Sep 17 00:00:00 2001
From: Alexey Milovidov <milovidov@yandex-team.ru>
Date: Mon, 7 Mar 2016 05:40:59 +0300
Subject: [PATCH] Fixed build [#METR-2807].

---
 libs/libcommon/include/common/Test.h    | 125 ++++++++++++++++++++++++
 libs/libcommon/src/tests/CMakeLists.txt |   2 +-
 libs/libcommon/src/tests/json_test.cpp  |   3 +-
 3 files changed, 128 insertions(+), 2 deletions(-)
 create mode 100644 libs/libcommon/include/common/Test.h

diff --git a/libs/libcommon/include/common/Test.h b/libs/libcommon/include/common/Test.h
new file mode 100644
index 00000000000..35e2ad8363c
--- /dev/null
+++ b/libs/libcommon/include/common/Test.h
@@ -0,0 +1,125 @@
+#pragma once
+
+#include <stdexcept>
+#include <sstream>
+
+/// Часть функциональности - только для сборки из репозитория Метрики.
+#ifndef NO_METRIKA
+	#include <ssqls/IDbObject.h>
+#else
+	struct IDbObject {};
+#endif
+
+#include <regex>
+#include <DB/Common/StackTrace.h>
+#include <DB/IO/WriteBufferFromString.h>
+
+#include <Poco/ConsoleChannel.h>
+#include <Poco/Logger.h>
+#include <Poco/AutoPtr.h>
+
+
+namespace Test
+{
+	template <class Result, class Reference, class Enable = void>
+	struct Comparator
+	{
+		void check(const Result & result, const Reference & reference, const std::string & check_name = "")
+		{
+			if (result != reference)
+			{
+				std::stringstream ss;
+				if (check_name.size())
+					ss << "Check name: " << check_name << ". ";
+
+				StackTrace stacktrace;
+				ss << "Result ";
+				if (check_name.size())
+					ss << "for \"" << check_name << "\" ";
+
+				ss << "differs from reference" <<
+				". Result: " << result << ", reference: " << reference << "\n. Stacktrace: " << stacktrace.toString();
+				throw std::logic_error(ss.str());
+			}
+		}
+	};
+
+	template <class Result, class Reference, class Enable = void, class ... Args>
+	void compare(const Result & result, const Reference & reference, Args && ... args)
+	{
+		Comparator<Result, Reference> comp;
+		comp.check(result, reference, std::forward<Args>(args)...);
+	}
+
+	template <class ContainerA, class ContainerB>
+	void compareContainers(const ContainerA & result, const ContainerB & reference)
+	{
+		if (result != reference)
+		{
+			std::stringstream ss;
+			ss << "Result differs from reference. Result: {";
+			for (auto a : result)
+				ss << " " << a;
+			ss << " }, reference {";
+			for (auto b : reference)
+				ss << " " << b;
+			ss << " }";
+
+			StackTrace stacktrace;
+			ss << "\n. Stacktrace: " << stacktrace.toString();
+
+			throw std::logic_error(ss.str());
+		}
+	}
+
+	template <class Result, class Reference>
+	struct Comparator<Result, Reference, typename std::enable_if<
+					std::is_base_of<IDbObject, Result>::value &&
+					std::is_base_of<IDbObject, Reference>::value>::type>
+	{
+		void check(const Result & result, const Reference & reference, const std::string field_name_regexp_str = ".*")
+		{
+			std::string res_s;
+			{
+				DB::WriteBufferFromString res_b(res_s);
+				result.writeTSKV(res_b, "", "", "");
+			}
+
+			std::string ref_s;
+			{
+				DB::WriteBufferFromString ref_b(ref_s);
+				reference.writeTSKV(ref_b, "", "", "");
+			}
+
+			size_t ref_pos = 0;
+			size_t res_pos = 0;
+
+			while (ref_pos != std::string::npos && res_pos != std::string::npos)
+			{
+				size_t new_ref_pos = ref_s.find('\t', ref_pos);
+				size_t new_res_pos = res_s.find('\t', res_pos);
+
+				auto ref_field = ref_s.substr(ref_pos,
+											  new_ref_pos != std::string::npos ? new_ref_pos - ref_pos : new_ref_pos);
+
+				if (std::regex_match(ref_field, std::regex(field_name_regexp_str + "=.*")))
+				{
+					auto res_field = res_s.substr(res_pos, new_res_pos != std::string::npos ? new_res_pos - res_pos : new_res_pos);
+					compare(res_field, ref_field);
+				}
+
+				res_pos = new_res_pos != res_s.size() && new_res_pos != std::string::npos ?
+					new_res_pos + 1 : std::string::npos;
+				ref_pos = new_ref_pos != ref_s.size() && new_ref_pos != std::string::npos ?
+					new_ref_pos + 1 : std::string::npos;
+			}
+		}
+	};
+
+	inline void initLogger()
+	{
+		Poco::AutoPtr<Poco::ConsoleChannel> channel = new Poco::ConsoleChannel(std::cerr);
+		Poco::Logger::root().setChannel(channel);
+		Poco::Logger::root().setLevel("trace");
+	}
+};
diff --git a/libs/libcommon/src/tests/CMakeLists.txt b/libs/libcommon/src/tests/CMakeLists.txt
index 3f20a5f2698..11056819424 100644
--- a/libs/libcommon/src/tests/CMakeLists.txt
+++ b/libs/libcommon/src/tests/CMakeLists.txt
@@ -12,6 +12,6 @@ target_link_libraries (date_lut2 common libicui18n.a libicuuc.a libicudata.a dl)
 target_link_libraries (date_lut3 common libicui18n.a libicuuc.a libicudata.a dl)
 target_link_libraries (date_lut4 common libicui18n.a libicuuc.a libicudata.a dl)
 target_link_libraries (multi_version common libboost_thread.a libboost_system.a rt)
-target_link_libraries (json_test statdaemons common libboost_thread.a libboost_system.a rt)
+target_link_libraries (json_test dbms libboost_thread.a libboost_system.a rt)
 
 add_check (json_test)
diff --git a/libs/libcommon/src/tests/json_test.cpp b/libs/libcommon/src/tests/json_test.cpp
index a507fc5ad33..4d5c6a91ece 100644
--- a/libs/libcommon/src/tests/json_test.cpp
+++ b/libs/libcommon/src/tests/json_test.cpp
@@ -1,4 +1,5 @@
-#include <statdaemons/Test.h>
+#define NO_METRIKA
+#include <common/Test.h>
 #include <vector>
 #include <string>
 #include <common/JSON.h>