Python Loader.load_from_yamlsの例

コード例 #1

ファイル: tests.py プロジェクト: clueproject/clue_sqlaloader

class LoaderTests(unittest.TestCase):

    def setUp(self):
        from clue_sqlaloader import Loader
        self.session = MockSession()
        self.loader = Loader(self.session)

    def test_load_from_list(self):
        self.loader.load_from_list([
                {'model': 'clue_sqlaloader.tests.MockDataObject',
                 'fields': {'abc': 'ghi'},
                 'execute': {'set_foo': 1}}
                ])
        self.assertEqual(self.session.objects,
                         [MockDataObject(abc='ghi',
                                         foo=1)])

    def test_load_from_yamls(self):
        self.loader.load_from_yamls(
            '- model: clue_sqlaloader.tests.MockDataObject')
        self.assertEqual(self.session.objects,
                         [MockDataObject()])

    def test_load_from_yamlf(self):
        data = '- model: clue_sqlaloader.tests.MockDataObject'

        def mockfile(filename, data=data):
            return MockFile(filename, data)
        self.loader.open = mockfile

        self.loader.load_from_yamlf('foo.yaml')
        self.assertEqual(self.session.objects,
                         [MockDataObject()])

コード例 #2

class LoaderTests(unittest.TestCase):
    def setUp(self):
        from clue_sqlaloader import Loader
        self.session = MockSession()
        self.loader = Loader(self.session)

    def test_load_from_list(self):
        self.loader.load_from_list([{
            'model': 'clue_sqlaloader.tests.MockDataObject',
            'fields': {
                'abc': 'ghi'
            },
            'execute': {
                'set_foo': 1
            }
        }])
        self.assertEqual(self.session.objects,
                         [MockDataObject(abc='ghi', foo=1)])

    def test_load_from_yamls(self):
        self.loader.load_from_yamls(
            '- model: clue_sqlaloader.tests.MockDataObject')
        self.assertEqual(self.session.objects, [MockDataObject()])

    def test_load_from_yamlf(self):
        data = '- model: clue_sqlaloader.tests.MockDataObject'

        def mockfile(filename, data=data):
            return MockFile(filename, data)

        self.loader.open = mockfile

        self.loader.load_from_yamlf('foo.yaml')
        self.assertEqual(self.session.objects, [MockDataObject()])