fix test cases

python-gino · Aug 26, 2018 · cf73072 · cf73072
1 parent 9587e75
commit cf73072
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 10 deletions.
diff --git a/tests/test_aiohttp.py b/tests/test_aiohttp.py
@@ -22,7 +22,7 @@ class User(db.Model):
         __tablename__ = 'gino_users'
 
         id = db.Column(db.BigInteger(), primary_key=True)
-        nickname = db.Column(db.Unicode(), default='noname')
+        nickname = db.Column('name', db.Unicode(), default='noname')
 
     routes = web.RouteTableDef()
 
@@ -107,8 +107,7 @@ async def _test(test_client):
         response = await test_client.get('/users/1?method=' + method)
         assert response.status == 404
 
-    response = await test_client.post('/users',
-                                      data=dict(name='fantix'))
+    response = await test_client.post('/users', data=dict(name='fantix'))
     assert response.status == 200
     assert await response.json() == dict(id=1, nickname='fantix')
 

diff --git a/tests/test_executemany.py b/tests/test_executemany.py
@@ -20,34 +20,49 @@ async def test_status(bind):
 # noinspection PyUnusedLocal
 async def test_all(bind):
     result = await User.insert().returning(User.nickname).gino.all(
-        dict(nickname='1'), dict(nickname='2'))
+        dict(name='1'), dict(name='2'))
     assert result is None
-    assert len(await User.query.gino.all()) == 2
+    rows = await User.query.gino.all()
+    assert len(rows) == 2
+    assert set(u.nickname for u in rows) == {'1', '2'}
 
     result = await User.insert().gino.all(
-        dict(nickname='3'), dict(nickname='4'))
+        dict(name='3'), dict(name='4'))
     assert result is None
+    rows = await User.query.gino.all()
+    assert len(rows) == 4
+    assert set(u.nickname for u in rows) == {'1', '2', '3', '4'}
 
 
 # noinspection PyUnusedLocal
 async def test_first(bind):
     result = await User.insert().returning(User.nickname).gino.first(
-        dict(nickname='1'), dict(nickname='2'))
+        dict(name='1'), dict(name='2'))
     assert result is None
+    rows = await User.query.gino.all()
     assert len(await User.query.gino.all()) == 2
+    assert set(u.nickname for u in rows) == {'1', '2'}
 
     result = await User.insert().gino.first(
-        dict(nickname='3'), dict(nickname='4'))
+        dict(name='3'), dict(name='4'))
     assert result is None
+    rows = await User.query.gino.all()
+    assert len(rows) == 4
+    assert set(u.nickname for u in rows) == {'1', '2', '3', '4'}
 
 
 # noinspection PyUnusedLocal
 async def test_scalar(bind):
     result = await User.insert().returning(User.nickname).gino.scalar(
-        dict(nickname='1'), dict(nickname='2'))
+        dict(name='1'), dict(name='2'))
     assert result is None
+    rows = await User.query.gino.all()
     assert len(await User.query.gino.all()) == 2
+    assert set(u.nickname for u in rows) == {'1', '2'}
 
     result = await User.insert().gino.scalar(
-        dict(nickname='3'), dict(nickname='4'))
+        dict(name='3'), dict(name='4'))
     assert result is None
+    rows = await User.query.gino.all()
+    assert len(rows) == 4
+    assert set(u.nickname for u in rows) == {'1', '2', '3', '4'}