From 86b78e723d8c4a1b7d67371991329daaf8056970 Mon Sep 17 00:00:00 2001
From: wyy <wuyanyi09@gmail.com>
Date: Sat, 14 Dec 2013 23:46:17 -0800
Subject: [PATCH] add unittest

---
 CMakeLists.txt                    |  3 +++
 test/curl.sh                      |  1 -
 test/unittest/CMakeLists.txt      | 10 +++++---
 test/unittest/ChineseFilterUt.cpp | 38 -------------------------------
 test/unittest/TChineseFilter.cpp  | 36 +++++++++++++++++++++++++++++
 5 files changed, 46 insertions(+), 42 deletions(-)
 delete mode 100755 test/curl.sh
 delete mode 100644 test/unittest/ChineseFilterUt.cpp
 create mode 100644 test/unittest/TChineseFilter.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 96db1b8..74e0126 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -13,3 +13,6 @@ ADD_SUBDIRECTORY(dicts)
 ADD_SUBDIRECTORY(scripts)
 ADD_SUBDIRECTORY(conf)
 ADD_SUBDIRECTORY(test)
+
+ENABLE_TESTING()
+ADD_TEST(NAME mytest COMMAND test1)
diff --git a/test/curl.sh b/test/curl.sh
deleted file mode 100755
index 52bfbf8..0000000
--- a/test/curl.sh
+++ /dev/null
@@ -1 +0,0 @@
-curl "http://127.0.0.1:11200/?key=南京市长江大桥"
diff --git a/test/unittest/CMakeLists.txt b/test/unittest/CMakeLists.txt
index 97ec91f..2ec018a 100644
--- a/test/unittest/CMakeLists.txt
+++ b/test/unittest/CMakeLists.txt
@@ -1,7 +1,11 @@
 SET(EXECUTABLE_OUTPUT_PATH ${PROJECT_BINARY_DIR}/test)
+SET(LIBRARY_OUTPUT_PATH ${PROJECT_BINARY_DIR}/test/lib)
 
 SET(GTEST_ROOT_DIR gtest-1.6.0)
 
-INCLUDE_DIRECTORIES(${GTEST_ROOT_DIR} ${GTEST_ROOT_DIR}/include)
-ADD_EXECUTABLE(test gtest_main.cc ${GTEST_ROOT_DIR}/src/gtest-all.cc)
-TARGET_LINK_LIBRARIES(test pthread)
+INCLUDE_DIRECTORIES(${GTEST_ROOT_DIR} ${GTEST_ROOT_DIR}/include ${PROJECT_SOURCE_DIR})
+ADD_LIBRARY(gtest STATIC ${GTEST_ROOT_DIR}/src/gtest-all.cc)
+ADD_EXECUTABLE(test1 gtest_main.cc TChineseFilter.cpp)
+TARGET_LINK_LIBRARIES(gtest pthread)
+TARGET_LINK_LIBRARIES(test1 gtest pthread)
+
diff --git a/test/unittest/ChineseFilterUt.cpp b/test/unittest/ChineseFilterUt.cpp
deleted file mode 100644
index 0051439..0000000
--- a/test/unittest/ChineseFilterUt.cpp
+++ /dev/null
@@ -1,38 +0,0 @@
-#include "../src/ChineseFilter.hpp"
-
-using namespace CppJieba;
-
-int main(int argc, char** argv)
-{
-    //ChineseFilter chFilter;
-    ifstream ifs(argv[1]);
-    string line;
-    string s;
-    while(getline(ifs, line))
-    {
-        const char * str = line.c_str();
-        uint size = line.size();
-        uint offset = 0;
-        while(offset < size)
-        {
-            uint len;
-            const char* t =  str+offset;
-        int ret = filterAscii(t, size, len);
-        s.assign(t, len);
-        cout<<s<<","<<ret<<","<<len<<endl;
-        offset += len;
-        }
-        
-        
-        //chFilter.feed(line);
-        //for(ChineseFilter::iterator it = chFilter.begin(); it != chFilter.end(); it++)
-        //{
-        //    //cout<<__FILE__<<__LINE__<<endl;
-        //    string tmp;
-        //    TransCode::encode(it.begin, it.end, tmp);
-        //    cout<<tmp<<endl;
-        //}
-    }
-    return 0;
-}
-
diff --git a/test/unittest/TChineseFilter.cpp b/test/unittest/TChineseFilter.cpp
new file mode 100644
index 0000000..ad6e3f8
--- /dev/null
+++ b/test/unittest/TChineseFilter.cpp
@@ -0,0 +1,36 @@
+#include "src/ChineseFilter.hpp"
+#include "gtest/gtest.h"
+
+using namespace CppJieba;
+
+TEST(ChineseFilterTest, Test1)
+{
+    const char* str = "heheh你好...hh";
+    string s;
+    vector<string> buf;
+    buf.push_back("heheh");
+    buf.push_back("你好");
+    buf.push_back("...hh");
+    vector<string> res;
+    uint size = strlen(str);
+    uint offset = 0;
+    while(offset < size)
+    {
+        uint len;
+        const char* t =  str + offset;
+        int ret = filterAscii(t, size - offset, len);
+        s.assign(t, len);
+        res.push_back(s);
+        //cout<<s<<","<<ret<<","<<len<<endl;
+        //cout<<str<<endl;
+        offset += len;
+    }
+    EXPECT_EQ(res, buf);
+}
+
+//int main(int argc, char** argv)
+//{
+//    //ChineseFilter chFilter;
+//    return 0;
+//}
+