Add db file is set UT

[hubacji1/coffee-getter.git] / ut / test_db.py

diff --git a/ut/test_db.py b/ut/test_db.py
index b6afcf45ffe149a39f3371b0326afc3083abdc41..c1577f4e0bb1565cf377dc60244c89fd9d5378b7 100644 (file)
--- a/ut/test_db.py
+++ b/ut/test_db.py
@@ -4,7 +4,7 @@ from os import path, remove
 from sqlite3 import connect
 from unittest import TestCase
 from cbconf import Conf
-from cbdb import Db
+from cbdb import Db, FileNotSetError
 
 TESTDB_FILENAME = ".tmptest.db"
 
@@ -94,3 +94,7 @@ class CoffeeDb(TestCase):
             self.assertIsNotNone(db.cur)
         finally:
             delete_test_db(TESTDB_FILENAME)
+
+class FailCoffeeDb(TestCase):
+    def test_db_path_not_set(self):
+        self.assertRaises(FileNotSetError, Db)