Don't append .db to sql database name

gunthercox · Aug 15, 2017 · 9f4b385 · 9f4b385
1 parent d20263a
commit 9f4b385
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/chatterbot/storage/sql_storage.py b/chatterbot/storage/sql_storage.py
@@ -52,7 +52,7 @@ def __init__(self, **kwargs):
 
         # Create a sqlite file if a database name is provided
         if database_name:
-            self.database_uri = "sqlite:///" + database_name + ".db"
+            self.database_uri = "sqlite:///" + database_name
 
         self.engine = create_engine(self.database_uri, convert_unicode=True)
 

diff --git a/tests/storage_adapter_tests/test_sqlalchemy_adapter.py b/tests/storage_adapter_tests/test_sqlalchemy_adapter.py
@@ -32,8 +32,8 @@ def test_set_database_name_none(self):
         self.assertEqual(adapter.database_uri, 'sqlite://')
 
     def test_set_database_name(self):
-        adapter = SQLStorageAdapter(database='test')
-        self.assertEqual(adapter.database_uri, 'sqlite:///test.db')
+        adapter = SQLStorageAdapter(database='test.sqlite3')
+        self.assertEqual(adapter.database_uri, 'sqlite:///test.sqlite3')
 
     def test_set_database_uri(self):
         adapter = SQLStorageAdapter(database_uri='sqlite:///db.sqlite3')