Fixed unicode with SQLite

ZuluPro · Nov 16, 2016 · a5960a0 · a5960a0
1 parent e3dcb34
commit a5960a0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/dbbackup/db/sqlite.py b/dbbackup/db/sqlite.py
@@ -1,3 +1,4 @@
+from __future__ import unicode_literals
 import warnings
 from tempfile import SpooledTemporaryFile
 from shutil import copyfileobj

diff --git a/dbbackup/tests/test_connectors.py b/dbbackup/tests/test_connectors.py
@@ -1,3 +1,5 @@
+from __future__ import unicode_literals
+
 import os
 from mock import patch, mock_open
 from tempfile import SpooledTemporaryFile
@@ -11,6 +13,7 @@
 from dbbackup.db.mysql import MysqlDumpConnector
 from dbbackup.db.postgresql import PgDumpConnector, PgDumpGisConnector
 from dbbackup.db.mongodb import MongoDumpConnector
+from dbbackup.tests.testapp.models import CharModel
 
 
 class GetConnectorTest(TestCase):
@@ -98,7 +101,6 @@ def test_run_command_with_parent_env(self):
 
 class SqliteConnectorTest(TestCase):
     def test_write_dump(self):
-
         dump_file = BytesIO()
         connector = SqliteConnector()
         connector._write_dump(dump_file)
@@ -111,6 +113,12 @@ def test_create_dump(self):
         dump = connector.create_dump()
         self.assertTrue(dump.read())
 
+    def test_create_dump_with_unicode(self):
+        CharModel.objects.create(field='\xe9')
+        connector = SqliteConnector()
+        dump = connector.create_dump()
+        self.assertTrue(dump.read())
+
     def test_restore_dump(self):
         connector = SqliteConnector()
         dump = connector.create_dump()