extended test coverage for the freeze package, thereby fixing some bugs.

test/test_freeze.py

@@ -60,3 +60,18 @@ class FreezeTestCase(unittest.TestCase):
                     else:
                         v1 = '%s' % v1
                 self.assertEqual(v2, v1)
+
+
+class SerializerTestCase(unittest.TestCase):
+    def test_Serializer(self):
+        from dataset.freeze.format.common import Serializer
+        from dataset.freeze.config import Export
+        from dataset.util import FreezeException
+
+        self.assertRaises(FreezeException, Serializer, {}, {})
+        s = Serializer(Export({'filename': 'f'}, {'mode': 'nomode'}), '')
+        self.assertRaises(FreezeException, getattr, s, 'wrap')
+        s = Serializer(Export({'filename': 'f'}, {}), '')
+        s.wrap
+        s = Serializer(Export({'filename': '-'}, {}), '')
+        self.assert_(s.fileobj)

test/test_freeze_app.py

@@ -0,0 +1,63 @@
+"""
+Test CLI following the recipe at http://dustinrcollins.com/testing-python-command-line-apps
+"""
+import os
+from unittest import TestCase
+from tempfile import mkdtemp
+from shutil import rmtree
+from copy import copy
+
+from six import StringIO
+
+from dataset import connect
+from dataset.util import FreezeException
+from dataset.freeze.config import Configuration, Export
+from dataset.freeze.app import create_parser, freeze_with_config, freeze_export
+from .sample_data import TEST_DATA
+
+
+class FreezeTestCase(TestCase):
+    """
+    Base TestCase class, sets up a CLI parser
+    """
+    def setUp(self):
+        parser = create_parser()
+        self.parser = parser
+        self.d = mkdtemp()
+        self.db_path = os.path.abspath(os.path.join(self.d, 'db.sqlite'))
+        self.db = 'sqlite:///' + self.db_path
+        _db = connect(self.db)
+        tbl = _db['weather']
+        for i, row in enumerate(TEST_DATA):
+            _row = copy(row)
+            _row['count'] = i
+            _row['bool'] = True
+            _row['none'] = None
+            tbl.insert(_row)
+
+    def tearDown(self):
+        rmtree(self.d, ignore_errors=True)
+
+    def test_with_empty_args(self):
+        """
+        User passes no args, should fail with SystemExit
+        """
+        with self.assertRaises(SystemExit):
+            self.parser.parse_args([])
+
+    def test_with_config(self):
+        """
+        """
+        cfg = Configuration(os.path.join(os.path.dirname(__file__), 'Freezefile.yaml'))
+        cfg.data['common']['database'] = self.db
+        cfg.data['common']['prefix'] = self.d
+        cfg.data['common']['query'] = 'SELECT * FROM weather'
+        cfg.data['exports'] = [
+            {'filename': '{{identity:count}}.json', 'mode': 'item', 'transform': {'bool': 'identity'}},
+            {'filename': 'weather.json', 'format': 'tabson'},
+            {'filename': 'weather.csv', 'fileobj': StringIO(), 'format': 'csv'},
+            {'filename': 'weather.json', 'fileobj': StringIO(), 'format': 'tabson'},
+            {'filename': 'weather.json', 'format': 'tabson', 'callback': 'read'},
+            {'skip': True}]
+        freeze_with_config(cfg, db=self.db)
+        self.assertRaises(FreezeException, freeze_export, Export(cfg.data['common'], {'query': 'SELECT * FROM notable'}))

test/test_persistence.py

@@ -4,7 +4,7 @@ from datetime import datetime
 
 try:
     from collections import OrderedDict
-except ImportError:
+except ImportError:  # pragma: no cover
     from ordereddict import OrderedDict  # Python < 2.7 drop-in
 
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError