Merge 0dec4b6 into 618c19a

fireant/tests/database/test_base_database.py

@@ -1,9 +1,8 @@
 from unittest import TestCase
 
 from fireant.database import Database
-from pypika import Field
-
 from fireant.middleware.concurrency import ThreadPoolConcurrencyMiddleware
+from pypika import Field
 
 
 class DatabaseTests(TestCase):
@@ -27,3 +26,5 @@ def test_no_concurrency_middleware_specified_gives_default_threadpool(self):
 
         self.assertIsInstance(db.concurrency_middleware, ThreadPoolConcurrencyMiddleware)
         self.assertEquals(db.concurrency_middleware.max_processes, 5)
+
+

fireant/tests/database/test_pickle.py

@@ -0,0 +1,29 @@
+import pickle
+
+from unittest.case import TestCase
+
+from fireant import (
+    MySQLDatabase,
+    PostgreSQLDatabase,
+    RedshiftDatabase,
+    SnowflakeDatabase,
+    VerticaDatabase,
+)
+
+
+class DatabasePickleTests(TestCase):
+    def test_that_database_inst_can_be_pickled(self):
+        for db_inst in (
+              MySQLDatabase(),
+              PostgreSQLDatabase(),
+              RedshiftDatabase(),
+              SnowflakeDatabase(),
+              VerticaDatabase(),
+        ):
+            with self.subTest(f'serialization {db_inst.__class__.__name__}'):
+                dump = pickle.dumps(db_inst, pickle.HIGHEST_PROTOCOL)
+                self.assertIsNotNone(dump, dump)
+
+            with self.subTest(f'deserialization {db_inst.__class__.__name__}'):
+                db_pickle = pickle.loads(dump)
+                self.assertIsNotNone(db_pickle, dump)