예제 #1
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_many = [
                [range(3), range(10, 13)],
            ]

            yield from registry.connect('/sample')
            groups = yield from (stmt.Select(sample.Group).join(
                sample.User).left_join(
                    sample.UserPreference,
                    sample.User.id == sample.UserPreference.user_id).where(
                        Mock()).run())
            yield from registry.disconnect('sample')

            groups = [group for group in groups]
            self.assertEqual(len(groups), 2)
            self.assertEqual(groups[0].to_dict(), {
                'id': 1,
                'created_at': 0,
                'name': 2
            })
            self.assertEqual(groups[1].to_dict(), {
                'id': 11,
                'created_at': 10,
                'name': 12
            })
예제 #2
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_many = [[range(3),
                                               range(10, 13)],
                                              ]

            yield from registry.connect('/sample')
            groups = yield from (stmt.Select(sample.Group)
                                     .join(sample.User)
                                     .left_join(sample.UserPreference,
                                                sample.User.id ==
                                                sample.UserPreference.user_id)
                                     .where(Mock()).run()
                                     )
            yield from registry.disconnect('sample')

            groups = [group for group in groups]
            self.assertEqual(len(groups), 2)
            self.assertEqual(groups[0].to_dict(),
                             {'id': 1, 'created_at': 0, 'name': 2})
            self.assertEqual(groups[1].to_dict(),
                             {'id': 11, 'created_at': 10, 'name': 12})
예제 #3
0
 def aiotest():
     from aiorm import registry
     from aiorm.orm.query.schema import CreateTable
     dummy_dialect = dialect.DummyCreateTableDialect()
     yield from registry.connect('/sample')
     yield from CreateTable(sample.Group).run()
     dummy_dialect.render_create_table.assert_called_once_with(sample.Group)
     # last query is a mock, we just ensure that a query has been
     # executed, not that the query is valid, this is a dialect test
     self.assertIsNotNone(driver.DummyCursor.last_query)
     yield from registry.disconnect('sample')
예제 #4
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm import Transaction

            yield from registry.connect('/sample')

            transaction = Transaction('sample')
            self.assertIsInstance(transaction.driver, driver.DummyDriver)
            self.assertEqual(transaction.driver.database, 'sample')
            yield from registry.disconnect('sample')
예제 #5
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm import Transaction

            yield from registry.connect('/sample')

            transaction = Transaction('sample')
            self.assertIsInstance(transaction.driver, driver.DummyDriver)
            self.assertEqual(transaction.driver.database, 'sample')
            yield from registry.disconnect('sample')
예제 #6
0
 def aiotest():
     from aiorm import registry
     from aiorm.orm.query.schema import CreateTable
     dummy_dialect = dialect.DummyCreateTableDialect()
     yield from registry.connect('/sample')
     yield from CreateTable(sample.Group).run()
     dummy_dialect.render_sql.assert_called_once()
     # last query is a mock, we just ensure that a query has been
     # executed, not that the query is valid, this is a dialect test
     self.assertIsNotNone(driver.DummyCursor.last_query)
     yield from registry.disconnect('sample')
예제 #7
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')
            driver.DummyCursor.return_many = [[('one',), ('two',)]]

            transaction = yield from Transaction('sample').begin()
            result = yield from transaction.fetchall()
            self.assertEqual(result, [('one',), ('two',)])
            yield from registry.disconnect('sample')
예제 #8
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')
            driver.DummyCursor.return_one = ['one']

            transaction = yield from Transaction('sample').begin()
            result = yield from transaction.fetchone()
            self.assertEqual(result, 'one')
            yield from registry.disconnect('sample')
예제 #9
0
 def aiotest():
     from aiorm import registry
     from aiorm.orm.query.schema import CreateSchema
     dummy_dialect = dialect.DummyCreateTableDialect()
     yield from registry.connect('/sample')
     yield from CreateSchema('sample').run()
     dummy_dialect.render_sql.assert_called()
     # last query is a mock, we just ensure that at least, a query
     # has been executed, not that the query is valid, this is a
     # dialect test
     self.assertIsNotNone(driver.DummyCursor.last_query)
     yield from registry.disconnect('sample')
예제 #10
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_one = [range(3)]
            yield from registry.connect('/sample')
            group = yield from stmt.Get(sample.Group, 1).run()
            yield from registry.disconnect('sample')

            self.assertEqual(group.to_dict(),
                             {'id': 1, 'created_at': 0, 'name': 2})
예제 #11
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')
            driver.DummyCursor.return_many = [[('one', ), ('two', )]]

            transaction = yield from Transaction('sample').begin()
            result = yield from transaction.fetchall()
            self.assertEqual(result, [('one', ), ('two', )])
            yield from registry.disconnect('sample')
예제 #12
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')
            driver.DummyCursor.return_one = ['one']

            transaction = yield from Transaction('sample').begin()
            result = yield from transaction.fetchone()
            self.assertEqual(result, 'one')
            yield from registry.disconnect('sample')
예제 #13
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')

            transaction = yield from Transaction('sample').begin()
            yield from transaction.execute('dummy_query')
            self.assertEqual(driver.DummyCursor.last_query, 'dummy_query')

            yield from registry.disconnect('sample')
예제 #14
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')

            transaction = yield from Transaction('sample').begin()
            yield from transaction.execute('dummy_query')
            self.assertEqual(driver.DummyCursor.last_query,
                             'dummy_query')

            yield from registry.disconnect('sample')
예제 #15
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_one = [None]

            yield from registry.connect('/sample')
            groups = yield from stmt.Get(sample.Group, 1).where(Mock()).run()
            yield from registry.disconnect('sample')

            driver.DummyDriver.mock = driver.DummyCursor()
            self.assertIsNone(groups)
예제 #16
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_one = [None]

            yield from registry.connect('/sample')
            groups = yield from stmt.Get(sample.Group, 1).where(Mock()).run()
            yield from registry.disconnect('sample')

            driver.DummyDriver.mock = driver.DummyCursor()
            self.assertIsNone(groups)
예제 #17
0
        def aiotest():

            from aiorm import registry
            from aiorm.orm.query import statements as stmt

            driver.DummyCursor.return_one = [range(3)]
            yield from registry.connect('/sample')
            group = yield from stmt.Get(sample.Group, 1).run()
            yield from registry.disconnect('sample')

            self.assertEqual(group.to_dict(), {
                'id': 1,
                'created_at': 0,
                'name': 2
            })
예제 #18
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')

            cursor = registry.get(interfaces.IDialect)()
            cursor.configure_mock(**{'render_rollback_transaction.return_value':
                                          'rollback_transaction'})

            transaction = yield from Transaction('sample').begin()
            yield from transaction.rollback()
            cursor.render_rollback_transaction.assert_called_once_with()
            self.assertEqual(driver.DummyCursor.last_query,
                             'rollback_transaction')

            yield from registry.disconnect('sample')
예제 #19
0
        def aiotest():
            from aiorm import registry
            from aiorm.orm import Transaction
            from aiorm.orm import interfaces

            yield from registry.connect('/sample')

            cursor = registry.get(interfaces.IDialect)()
            cursor.configure_mock(
                **
                {'render_begin_transaction.return_value': 'begin_transaction'})

            transaction = Transaction('sample')
            yield from transaction.begin()
            cursor.render_begin_transaction.assert_called_once_with()
            self.assertEqual(driver.DummyCursor.last_query,
                             'begin_transaction')

            yield from registry.disconnect('sample')
예제 #20
0
 def aioTearDown(self):
     from aiorm import registry
     from aiorm.orm.dialect.postgresql import Dialect, CreateTableDialect
     registry.unregister(Dialect)
     registry.unregister(CreateTableDialect)
     yield from registry.disconnect('sample')
예제 #21
0
 def aioTearDown(self):
     from aiorm import registry
     from aiorm.orm.dialect.postgresql import Dialect, CreateTableDialect
     registry.unregister(Dialect)
     registry.unregister(CreateTableDialect)
     yield from registry.disconnect('sample')