Skip to content

Commit

Permalink
Adjust Django and is_sqlite errors
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrew-Chen-Wang committed Feb 11, 2020
1 parent 62e1d3e commit e724b6a
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions cachalot/tests/write.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def test_update_or_create(self):
self.assertEqual(t.name, 'test')
self.assertEqual(t.public, True)

with self.assertNumQueries(3 if self.is_dj_21_below_and_is_sqlite() else 2):
with self.assertNumQueries(3 if self.is_sqlite else 2):
t, created = Test.objects.update_or_create(
name='test', defaults={'public': False})
self.assertFalse(created)
Expand Down Expand Up @@ -167,7 +167,7 @@ def test_delete(self):
for i in range(2, 11)])
with self.assertNumQueries(1):
self.assertEqual(Test.objects.count(), 10)
with self.assertNumQueries(2 if self.is_dj_21_below_and_is_sqlite() else 1):
with self.assertNumQueries(2 if self.is_sqlite else 1):
Test.objects.all().delete()
with self.assertNumQueries(1):
self.assertEqual(Test.objects.count(), 0)
Expand Down Expand Up @@ -604,7 +604,7 @@ def test_invalidate_select_related(self):
self.assertEqual(data3[2].owner, u2)
self.assertEqual(data3[3].owner, u1)

with self.assertNumQueries(2 if self.is_dj_21_below_and_is_sqlite() else 1):
with self.assertNumQueries(2 if self.is_sqlite else 1):
Test.objects.filter(name__in=['test1', 'test2']).delete()
with self.assertNumQueries(1):
data4 = list(Test.objects.select_related('owner'))
Expand Down Expand Up @@ -636,7 +636,7 @@ def test_invalidate_prefetch_related(self):
self.assertEqual(data3[0].owner, u)
self.assertListEqual(list(data3[0].owner.groups.all()), [])

with self.assertNumQueries(9 if self.is_dj_21_below_and_is_sqlite() else 6):
with self.assertNumQueries(9 if self.is_sqlite else 6):
group = Group.objects.create(name='test_group')
permissions = list(Permission.objects.all()[:5])
group.permissions.add(*permissions)
Expand Down Expand Up @@ -867,7 +867,7 @@ def test_invalidate_table_inheritance(self):
with self.assertNumQueries(1):
self.assertEqual(TestChild.objects.get(), t_child)

with self.assertNumQueries(2 if self.is_sqlite else 1):
with self.assertNumQueries(2 if self.is_dj_21_below_and_is_sqlite() else 1):
TestParent.objects.filter(pk=t_child.pk).update(name='modified')

with self.assertNumQueries(1):
Expand Down

0 comments on commit e724b6a

Please sign in to comment.