Add tests for database name and uri

gunthercox · Jul 21, 2017 · 95efe1b · 95efe1b
1 parent 4c690ac
commit 95efe1b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/storage_adapter_tests/test_sqlalchemy_adapter.py b/tests/storage_adapter_tests/test_sqlalchemy_adapter.py
@@ -27,6 +27,18 @@ def tearDown(self):
 
 class SQLStorageAdapterTestCase(SQLAlchemyAdapterTestCase):
 
+    def test_set_database_name_none(self):
+        adapter = SQLStorageAdapter(database=None)
+        self.assertEqual(adapter.database_uri, 'sqlite://')
+
+    def test_set_database_name(self):
+        adapter = SQLStorageAdapter(database='test')
+        self.assertEqual(adapter.database_uri, 'sqlite:///test.db')
+
+    def test_set_database_uri(self):
+        adapter = SQLStorageAdapter(database_uri='sqlite:///db.sqlite3')
+        self.assertEqual(adapter.database_uri, 'sqlite:///db.sqlite3')
+
     def test_count_returns_zero(self):
         """
         The count method should return a value of 0