Skip to content

Commit

Permalink
Additional tests for insert/many
Browse files Browse the repository at this point in the history
  • Loading branch information
coleifer committed Nov 1, 2022
1 parent 2d7ed54 commit 60f5c0b
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions tests/models.py
Expand Up @@ -2487,6 +2487,20 @@ def test_simple_returning(self):
.returning(User.username.alias('new_username')))
self.assertEqual([x.new_username for x in query.execute()], ['sipp'])

# Minimal test with insert_many.
query = User.insert_many([('u7',), ('u8',)])
self.assertEqual([r for r, in query.execute()], [7, 8])

# Test with insert / on conflict.
query = (User
.insert_many([(7, 'u7',), (9, 'u9',)],
[User.id, User.username])
.on_conflict(conflict_target=[User.id],
update={User.username: User.username + 'x'})
.returning(User))
self.assertEqual([(x.id, x.username) for x in query],
[(7, 'u7x'), (9, 'u9')])

def test_simple_returning_insert_update_delete(self):
res = User.insert(username='charlie').returning(User).execute()
self.assertEqual([u.username for u in res], ['charlie'])
Expand Down

0 comments on commit 60f5c0b

Please sign in to comment.