summaryrefslogtreecommitdiff
path: root/tests/dbshell/test_sqlite.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/dbshell/test_sqlite.py')
-rw-r--r--tests/dbshell/test_sqlite.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/tests/dbshell/test_sqlite.py b/tests/dbshell/test_sqlite.py
new file mode 100644
index 0000000000..46d09895c8
--- /dev/null
+++ b/tests/dbshell/test_sqlite.py
@@ -0,0 +1,31 @@
+from pathlib import Path
+from subprocess import CompletedProcess
+from unittest import mock, skipUnless
+
+from django.db import connection
+from django.db.backends.sqlite3.client import DatabaseClient
+from django.test import SimpleTestCase
+
+
+@skipUnless(connection.vendor == 'sqlite', 'SQLite tests.')
+class SqliteDbshellCommandTestCase(SimpleTestCase):
+ def _run_dbshell(self):
+ """Run runshell command and capture its arguments."""
+ def _mock_subprocess_run(*args, **kwargs):
+ self.subprocess_args = list(*args)
+ return CompletedProcess(self.subprocess_args, 0)
+
+ client = DatabaseClient(connection)
+ with mock.patch('subprocess.run', new=_mock_subprocess_run):
+ client.runshell()
+ return self.subprocess_args
+
+ def test_path_name(self):
+ with mock.patch.dict(
+ connection.settings_dict,
+ {'NAME': Path('test.db.sqlite3')},
+ ):
+ self.assertEqual(
+ self._run_dbshell(),
+ ['sqlite3', 'test.db.sqlite3'],
+ )