test: 3 functions in tool_quit.h

Merge pull request #658 from hongriTianqi/develop

Author: caic99

source/module_base/test/CMakeLists.txt

@@ -14,3 +14,7 @@ AddTest(
   TARGET tool_check
   SOURCES tool_check_test.cpp
 )
+AddTest(
+  TARGET tool_quit
+  SOURCES tool_quit_test.cpp
+)

source/module_base/test/tool_quit_test.cpp

@@ -0,0 +1,105 @@
+#include "../tool_quit.h"
+#include "../global_variable.h"
+#include "gtest/gtest.h"
+#include "gmock/gmock.h"
+#include "mpi.h"
+
+/************************************************
+ *  unit test of functions in tool_quit.h
+ ***********************************************/
+
+/**
+ * - Tested Function
+ *   - ModuleBase::WARNING
+ *     - print out warning info. in warning.log
+ *
+ *   - ModuleBase::QUIT
+ *     - close log files and exit
+ *
+ *   - ModuleBase::WARNING_QUIT
+ *     - combine the above 2 functions
+ *
+ * - Author: Tianqi Zhao
+ */
+
+class ToolQuitTest : public testing::Test
+{
+  protected:
+	std::ifstream ifs;
+	// for capturing output on screen
+	std::string output;
+	void SetUp()
+	{
+		GlobalV::ofs_warning.open("warning.log");
+		GlobalV::ofs_running.open("running.log");
+		GlobalV::global_out_dir = "OUT/";
+	}
+	void TearDown()
+	{
+		remove("warning.log");
+		remove("running.log");
+	}
+};
+
+
+TEST_F(ToolQuitTest,warning)
+{
+	ModuleBase::WARNING("INPUT","bad input parameter");
+	GlobalV::ofs_warning.close();
+	ifs.open("warning.log");
+	getline(ifs,output);
+	// test output in warning.log file
+	EXPECT_THAT(output,testing::HasSubstr("warning"));
+	ifs.close();
+}
+
+TEST_F(ToolQuitTest,quit)
+{
+	testing::internal::CaptureStdout();
+	EXPECT_EXIT(ModuleBase::QUIT(), ::testing::ExitedWithCode(0), "");
+	output = testing::internal::GetCapturedStdout();
+	// test output on screen
+	EXPECT_THAT(output,testing::HasSubstr("TIME(Sec)-----"));
+}
+
+// use EXPECT_EXIT to test exit codes
+TEST_F(ToolQuitTest,warningquit)
+{
+	testing::internal::CaptureStdout();
+	EXPECT_EXIT(ModuleBase::WARNING_QUIT("INPUT","bad input parameter"), 
+			::testing::ExitedWithCode(0), "");
+	output = testing::internal::GetCapturedStdout();
+	// test output on screening
+	EXPECT_THAT(output,testing::HasSubstr("TIME(Sec)-----"));
+	GlobalV::ofs_warning.close();
+	GlobalV::ofs_running.close();
+	ifs.open("warning.log");
+	getline(ifs,output);
+	// test output in warning.log file
+	EXPECT_THAT(output,testing::HasSubstr("warning"));
+	ifs.close();
+	ifs.open("running.log");
+	getline(ifs,output);
+	// test output in running.log file
+	EXPECT_THAT(output,testing::HasSubstr("!!!!!!!"));
+	ifs.close();
+}
+
+// use __MPI to activate parallel environment
+int main(int argc, char **argv)
+{
+
+#ifdef __MPI
+	MPI_Init(&argc,&argv);
+#endif
+
+	testing::InitGoogleTest(&argc,argv);
+	int result = RUN_ALL_TESTS();
+
+#ifdef __MPI
+	MPI_Finalize();
+#endif
+
+	return result;
+}
+