1 files changed, 18 insertions, 27 deletions

diff --git a/test/test_config/test_sa.py b/test/test_config/test_sa.py
index d6ff940..57459bc 100644
--- a/test/test_config/test_sa.py
+++ b/test/test_config/test_sa.py
@@ -25,6 +25,20 @@ from dodai.config.databases.sa import Sa
 
 class TestSa(unittest.TestCase):
 
+    DB_ONE = {}
+    DB_ONE['protocol'] ='postgresql'
+    DB_ONE['protocol_extra'] ='psycopg2'
+    DB_ONE['hostname'] ='127.0.0.1'
+    DB_ONE['username'] ='test'
+    DB_ONE['password'] ='test'
+    DB_ONE['port'] ='12345'
+    DB_ONE['database'] ='testing'
+    DB_ONE['schema'] ='foo'
+
+    DB_TWO = {}
+    DB_TWO['protocol'] ='sqlite'
+    DB_TWO['filename'] ='/tmp/test'
+
     def setUp(self):
         self.create_engine = Dingus('create_engine')
         self.session_maker = Dingus('session_maker')
@@ -32,33 +46,10 @@ class TestSa(unittest.TestCase):
         self.obj = Sa(self.create_engine, self.session_maker,
                       self.result_object)
 
-
     def test_sa_one(self):
-        name = 'test_db_two'
-        obj = DbOne()
-        db = self.obj(name, obj)
-
+        name = 'test_db_one'
+        db = self.obj(self.DB_ONE, name)
 
     def test_sa_two(self):
-        name = 'test_db_six'
-        obj = DbTwo()
-        db = self.obj(name, obj)
-
-
-class DbOne(object):
-
-    def __init__(self):
-        self.protocol='postgresql'
-        self.protocol_extra='psycopg2'
-        self.hostname='127.0.0.1'
-        self.username='test'
-        self.password='test'
-        self.port='12345'
-        self.database='testing'
-        self.schema='foo'
-
-class DbTwo(object):
-
-    def __init__(self):
-        self.protocol='sqlite'
-        self.filename='/tmp/test'
+        name = 'test_db_two'
+        db = self.obj(self.DB_TWO, name)