From 715dc6e9b54e3509f23820f63b0972b29f5b5a62 Mon Sep 17 00:00:00 2001
From: JackeyGao <gaojunqi@outlook.com>
Date: Fri, 30 Jun 2017 17:34:47 +0800
Subject: [PATCH] test: add sqlite test

---
 test_poetry.py | 29 +++++++++++++++++++++--------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/test_poetry.py b/test_poetry.py
index 7647499..27454ec 100644
--- a/test_poetry.py
+++ b/test_poetry.py
@@ -1,11 +1,5 @@
 #! -*- coding: utf-8 -*-
-"""
-
-   测试古诗JSON文件是否有效
-
-"""
-
-import os, json
+import os, json, sqlite3
 
 def check_json(f):
     filepath = os.path.join('./json', f)
@@ -14,8 +8,27 @@ def check_json(f):
             _ = json.loads(file.read())
             return True
         except:
-            assert False, "校验(%s)失败" % f
+            assert False, u"校验(%s)失败" % f
 
 
 def test_json():
+    """
+        测试古诗JSON文件是否有效
+    """
     map(check_json, os.listdir('./json'))
+
+
+
+def test_sqlite():
+    """
+        测试ci数据库文件是否有效
+    """
+    conn = sqlite3.connect('./ci/ci.db')
+
+    c = conn.cursor()
+
+    c.execute("SELECT name FROM sqlite_master WHERE type='table'")
+
+    tables = c.fetchall()
+    
+    assert len(tables) == 2, u"Sqlite文件异常"