Python mock_engine Examples

Example #1

File: test_ddlevents.py Project: clones/sqlalchemy

    def test_conditional_constraint(self):
        metadata, users, engine = self.metadata, self.users, self.engine
        nonpg_mock = engines.mock_engine(dialect_name='sqlite')
        pg_mock = engines.mock_engine(dialect_name='postgresql')
        
        constraint = CheckConstraint('a < b',name="my_test_constraint", table=users)

        # by placing the constraint in an Add/Drop construct,
        # the 'inline_ddl' flag is set to False
        AddConstraint(constraint, on='postgresql').execute_at("after-create", users)
        DropConstraint(constraint, on='postgresql').execute_at("before-drop", users)
        
        metadata.create_all(bind=nonpg_mock)
        strings = " ".join(str(x) for x in nonpg_mock.mock)
        assert "my_test_constraint" not in strings  
        metadata.drop_all(bind=nonpg_mock)
        strings = " ".join(str(x) for x in nonpg_mock.mock)
        assert "my_test_constraint" not in strings

        metadata.create_all(bind=pg_mock)
        strings = " ".join(str(x) for x in pg_mock.mock)
        assert "my_test_constraint" in strings
        metadata.drop_all(bind=pg_mock)
        strings = " ".join(str(x) for x in pg_mock.mock)
        assert "my_test_constraint" in strings

Example #2

File: test_ddlevents.py Project: Mozilla-GitHub-Standards/441e91400508d94b5a6931d7693918293007697b91395acd9460441a437990ec

    def test_conditional_constraint(self):
        metadata, users, engine = self.metadata, self.users, self.engine
        nonpg_mock = engines.mock_engine(dialect_name='sqlite')
        pg_mock = engines.mock_engine(dialect_name='postgresql')
        constraint = CheckConstraint('a < b',
                                     name='my_test_constraint',
                                     table=users)

        # by placing the constraint in an Add/Drop construct, the
        # 'inline_ddl' flag is set to False

        AddConstraint(constraint,
                      on='postgresql').execute_at('after-create', users)
        DropConstraint(constraint,
                       on='postgresql').execute_at('before-drop', users)
        metadata.create_all(bind=nonpg_mock)
        strings = ' '.join(str(x) for x in nonpg_mock.mock)
        assert 'my_test_constraint' not in strings
        metadata.drop_all(bind=nonpg_mock)
        strings = ' '.join(str(x) for x in nonpg_mock.mock)
        assert 'my_test_constraint' not in strings
        metadata.create_all(bind=pg_mock)
        strings = ' '.join(str(x) for x in pg_mock.mock)
        assert 'my_test_constraint' in strings
        metadata.drop_all(bind=pg_mock)
        strings = ' '.join(str(x) for x in pg_mock.mock)
        assert 'my_test_constraint' in strings

Example #3

File: test_constraints.py Project: samucc/kuma

    def test_use_alter(self):
        m = MetaData()
        t = Table(
            't',
            m,
            Column('a', Integer),
        )

        t2 = Table(
            't2',
            m,
            Column('a', Integer, ForeignKey('t.a',
                                            use_alter=True,
                                            name='fk_ta')),
            Column('b', Integer,
                   ForeignKey('t.a',
                              name='fk_tb')),  # to ensure create ordering ...
        )

        e = engines.mock_engine(dialect_name='postgresql')
        m.create_all(e)
        m.drop_all(e)

        e.assert_sql([
            'CREATE TABLE t (a INTEGER)',
            'CREATE TABLE t2 (a INTEGER, b INTEGER, CONSTRAINT fk_tb FOREIGN KEY(b) REFERENCES t (a))',
            'ALTER TABLE t2 ADD CONSTRAINT fk_ta FOREIGN KEY(a) REFERENCES t (a)',
            'ALTER TABLE t2 DROP CONSTRAINT fk_ta', 'DROP TABLE t2',
            'DROP TABLE t'
        ])

Example #4

File: test_ddlevents.py Project: gajop/springgrid

 def setup(self):
     self.engine = engines.mock_engine()
     self.metadata = MetaData(self.engine)
     self.users = Table('users', self.metadata,
                        Column('user_id', Integer, primary_key=True),
                        Column('user_name', String(40)),
                        )

Example #5

 def setup(self):
     self.engine = engines.mock_engine()
     self.metadata = MetaData(self.engine)
     self.users = Table(
         'users',
         self.metadata,
         Column('user_id', Integer, primary_key=True),
         Column('user_name', String(40)),
     )

Example #6

File: test_constraints.py Project: AndryulE/kitsune

    def test_use_alter(self):
        m = MetaData()
        t = Table('t', m,
                  Column('a', Integer),
        )
        
        t2 = Table('t2', m,
                Column('a', Integer, ForeignKey('t.a', use_alter=True, name='fk_ta')),
                Column('b', Integer, ForeignKey('t.a', name='fk_tb')), # to ensure create ordering ...
        )

        e = engines.mock_engine(dialect_name='postgresql')
        m.create_all(e)
        m.drop_all(e)

        e.assert_sql([
            'CREATE TABLE t (a INTEGER)', 
            'CREATE TABLE t2 (a INTEGER, b INTEGER, CONSTRAINT fk_tb FOREIGN KEY(b) REFERENCES t (a))', 
            'ALTER TABLE t2 ADD CONSTRAINT fk_ta FOREIGN KEY(a) REFERENCES t (a)', 
            'ALTER TABLE t2 DROP CONSTRAINT fk_ta', 
            'DROP TABLE t2', 
            'DROP TABLE t'
        ])

Example #7

 def setup(self):
     self.bind = engines.mock_engine()
     self.metadata = MetaData()
     self.table = Table('t', self.metadata, Column('id', Integer))

Example #8

File: test_metadata.py Project: hughperkins/ailadder

 def setup(self):
     self.engine = engines.mock_engine()
     self.metadata = MetaData(self.engine)

Example #9

File: test_ddlevents.py Project: gajop/springgrid

 def setup(self):
     self.bind = engines.mock_engine()
     self.metadata = MetaData()
     self.table = Table('t', self.metadata, Column('id', Integer))

Example #10

File: test_metadata.py Project: furoka/ailadder

 def setup(self):
     self.engine = engines.mock_engine()
     self.metadata = MetaData(self.engine)