added dbutils tests

2 files changed, 18 insertions, 2 deletions

diff --git a/dbutils.py b/dbutils.py
index 2dd8e66..31d56ab 100644
--- a/dbutils.py
+++ b/dbutils.py
@@ -1,10 +1,11 @@
 import cx_Oracle
 import dbconfig
 
-connStr = '%s/%s@%s' % (dbconfig.USER, dbconfig.PASS, dbconfig.SID)
+CONN_STR = '%s/%s@%s' % (dbconfig.USER, dbconfig.PASS, dbconfig.SID)
 
 def get_tables():
-    with cx_Oracle.connect(connStr) as conn:
+    with cx_Oracle.connect(CONN_STR) as conn:
         cur = conn.cursor()
         cur.execute('SELECT owner, table_name FROM all_tables')
         return [table for owner, table in cur.fetchall()]
+        
diff --git a/dbutils_test.py b/dbutils_test.py
new file mode 100644
index 0000000..ba4d6bb
--- /dev/null
+++ b/dbutils_test.py
@@ -0,0 +1,15 @@
+import unittest
+
+import dbutils
+
+class DBUtilsTest(unittest.TestCase):
+
+    def setUp(self):
+        self.seq = range(10)
+
+    def test_get_tables(self):
+        tables = dbutils.get_tables()
+        self.assertTrue(len(tables) >= 1)
+
+if __name__ == '__main__':
+    unittest.main()