wuttasync/tests/cli/test_base.py

#-*- coding: utf-8; -*-

import inspect
from unittest import TestCase
from unittest.mock import patch

from wuttasync.cli import base as mod
from wuttjamaican.testing import DataTestCase


class TestImportCommandHandler(DataTestCase):

    def make_handler(self, **kwargs):
        return mod.ImportCommandHandler(self.config, **kwargs)

    def test_import_handler(self):

        # none
        handler = self.make_handler()
        self.assertIsNone(handler.import_handler)

        FromCsvToWutta = self.app.load_object('wuttasync.importing.csv:FromCsvToWutta')

        # as spec
        handler = self.make_handler(import_handler=FromCsvToWutta.get_spec())
        self.assertIsInstance(handler.import_handler, FromCsvToWutta)

        # as factory
        handler = self.make_handler(import_handler=FromCsvToWutta)
        self.assertIsInstance(handler.import_handler, FromCsvToWutta)

        # as instance
        myhandler = FromCsvToWutta(self.config)
        handler = self.make_handler(import_handler=myhandler)
        self.assertIs(handler.import_handler, myhandler)

    def test_run(self):
        handler = self.make_handler(import_handler='wuttasync.importing.csv:FromCsvToWutta')

        with patch.object(handler, 'list_models') as list_models:
            handler.run({'list_models': True})
            list_models.assert_called_once_with({'list_models': True})

        with patch.object(handler, 'import_handler') as import_handler:
            handler.run({'models': []})
            import_handler.process_data.assert_called_once_with()

    def test_list_models(self):
        handler = self.make_handler(import_handler='wuttasync.importing.csv:FromCsvToWutta')

        with patch.object(mod, 'sys') as sys:
            handler.list_models({})
            # just test a few random things we expect to see
            self.assertTrue(sys.stdout.write.has_call('ALL MODELS:\n'))
            self.assertTrue(sys.stdout.write.has_call('Person'))
            self.assertTrue(sys.stdout.write.has_call('User'))
            self.assertTrue(sys.stdout.write.has_call('Upgrade'))


class TestImporterCommand(TestCase):

    def test_basic(self):
        def myfunc(ctx, **kwargs):
            pass

        sig1 = inspect.signature(myfunc)
        self.assertIn('kwargs', sig1.parameters)
        self.assertNotIn('dry_run', sig1.parameters)
        wrapt = mod.import_command(myfunc)
        sig2 = inspect.signature(wrapt)
        self.assertNotIn('kwargs', sig2.parameters)
        self.assertIn('dry_run', sig2.parameters)


class TestFileImporterCommand(TestCase):

    def test_basic(self):
        def myfunc(ctx, **kwargs):
            pass

        sig1 = inspect.signature(myfunc)
        self.assertIn('kwargs', sig1.parameters)
        self.assertNotIn('dry_run', sig1.parameters)
        self.assertNotIn('input_file_path', sig1.parameters)
        wrapt = mod.file_import_command(myfunc)
        sig2 = inspect.signature(wrapt)
        self.assertNotIn('kwargs', sig2.parameters)
        self.assertIn('dry_run', sig2.parameters)
        self.assertIn('input_file_path', sig2.parameters)