64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
|
"""
|
||
|
|
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'}))
|