class CreateTableDialectTestCase(TestCase): _fixtures = [sample.SampleFixture] def setUp(self): super().setUp() from aiorm.orm.dialect.postgresql import CreateTableDialect self._dialect = CreateTableDialect() def test_render_create_table(self): self._dialect.render_create_table(sample.UserGroup) self.assertEqual(self._dialect.query, """\ CREATE TABLE IF NOT EXISTS"user_group" ( "group_id" int NOT NULL, "user_id" int NOT NULL, CONSTRAINT "user_group_group_id_fkey" FOREIGN KEY ("group_id") REFERENCES "group" ("id") MATCH SIMPLE ON UPDATE NO ACTION \ ON DELETE NO ACTION, CONSTRAINT "user_group_user_id_fkey" FOREIGN KEY ("user_id") REFERENCES "user" ("id") MATCH SIMPLE ON UPDATE NO ACTION ON DELETE NO ACTION ) """)
def setUp(self): super().setUp() from aiorm.orm.dialect.postgresql import CreateTableDialect self._dialect = CreateTableDialect()