fix bad test

* fixes #675

Author: fantix

tests/test_schema.py

@@ -1,41 +1,40 @@
 from enum import Enum
 
 import pytest
+from asyncpg import UndefinedTableError
 
 import gino
 from gino.dialects.asyncpg import AsyncEnum
 
 pytestmark = pytest.mark.asyncio
-db = gino.Gino()
 
 
 class MyEnum(Enum):
     ONE = "one"
     TWO = "two"
 
 
-class Blog(db.Model):
-    __tablename__ = "s_blog"
-
-    id = db.Column(db.BigInteger(), primary_key=True)
-    title = db.Column(db.Unicode(), index=True, comment="Title Comment")
-    visits = db.Column(db.BigInteger(), default=0)
-    comment_id = db.Column(db.ForeignKey("s_comment.id"))
-    number = db.Column(db.Enum(MyEnum), nullable=False, default=MyEnum.TWO)
-    number2 = db.Column(AsyncEnum(MyEnum), nullable=False, default=MyEnum.TWO)
-
+async def test(engine, define=True):
+    db = gino.Gino()
 
-class Comment(db.Model):
-    __tablename__ = "s_comment"
+    class Blog(db.Model):
+        __tablename__ = "s_blog"
 
-    id = db.Column(db.BigInteger(), primary_key=True)
-    blog_id = db.Column(db.ForeignKey("s_blog.id", name="blog_id_fk"))
+        id = db.Column(db.BigInteger(), primary_key=True)
+        title = db.Column(db.Unicode(), index=True, comment="Title Comment")
+        visits = db.Column(db.BigInteger(), default=0)
+        comment_id = db.Column(db.ForeignKey("s_comment.id"))
+        number = db.Column(db.Enum(MyEnum), nullable=False, default=MyEnum.TWO)
+        number2 = db.Column(AsyncEnum(MyEnum), nullable=False, default=MyEnum.TWO)
 
+    class Comment(db.Model):
+        __tablename__ = "s_comment"
 
-blog_seq = db.Sequence("blog_seq", metadata=db, schema="schema_test")
+        id = db.Column(db.BigInteger(), primary_key=True)
+        blog_id = db.Column(db.ForeignKey("s_blog.id", name="blog_id_fk"))
 
+    blog_seq = db.Sequence("blog_seq", metadata=db, schema="schema_test")
 
-async def test(engine, define=True):
     try:
         async with engine.acquire() as conn:
             assert not await engine.dialect.has_schema(conn, "schema_test")
@@ -78,6 +77,5 @@ class Comment2(db.Model):
 
 async def test_no_alter(engine, mocker):
     engine.dialect.supports_alter = False
-    warn = mocker.patch("warnings.warn")
-    await test(engine, define=False)
-    assert warn.called
+    with pytest.raises(UndefinedTableError):
+        await test(engine, define=False)