Skip to content

Commit 2becf27

Browse files
committed
rename TestModel in test so pytest does not try to collect it
1 parent cae2875 commit 2becf27

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

tests/test_core.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import pytest
44
import sys
55
from typing import Type, TYPE_CHECKING, TypeVar
6-
from tests.django_routines_tests.models import TestModel
6+
from tests.django_routines_tests.models import TestModel as _TestModel
77

88
from django.core.management import call_command, CommandError
99
from django_typer.management import get_command, TyperCommand
@@ -47,13 +47,13 @@ class CoreTests(with_typehint(TestCase)):
4747
def setUp(self):
4848
if track_file.is_file():
4949
os.remove(track_file)
50-
TestModel.objects.create(id=0, name="Brian")
50+
_TestModel.objects.create(id=0, name="Brian")
5151
super().setUp()
5252

5353
def tearDown(self):
5454
if track_file.is_file():
5555
os.remove(track_file)
56-
TestModel.objects.all().delete()
56+
_TestModel.objects.all().delete()
5757
super().setUp()
5858

5959
def strip_ansi(self, text):
@@ -890,41 +890,41 @@ def test_settings_format(self):
890890

891891
def test_non_atomic(self):
892892
call_command("routine", "atomic-pass")
893-
self.assertEqual(TestModel.objects.get(id=0).name, "Name3")
894-
self.assertEqual(TestModel.objects.get(id=1).name, "Name4")
895-
self.assertEqual(TestModel.objects.count(), 2)
893+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name3")
894+
self.assertEqual(_TestModel.objects.get(id=1).name, "Name4")
895+
self.assertEqual(_TestModel.objects.count(), 2)
896896

897897
def test_force_nonatomic(self):
898898
call_command("routine", "atomic-pass", "--non-atomic")
899-
self.assertEqual(TestModel.objects.get(id=0).name, "Name3")
900-
self.assertEqual(TestModel.objects.get(id=1).name, "Name4")
901-
self.assertEqual(TestModel.objects.count(), 2)
899+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name3")
900+
self.assertEqual(_TestModel.objects.get(id=1).name, "Name4")
901+
self.assertEqual(_TestModel.objects.count(), 2)
902902

903903
def test_atomic_fail(self):
904904
with self.assertRaises(TestError):
905905
call_command("routine", "atomic-fail")
906-
self.assertEqual(TestModel.objects.get(id=0).name, "Brian")
907-
self.assertEqual(TestModel.objects.count(), 1)
906+
self.assertEqual(_TestModel.objects.get(id=0).name, "Brian")
907+
self.assertEqual(_TestModel.objects.count(), 1)
908908

909909
def test_force_nonatomic_fail(self):
910910
with self.assertRaises(TestError):
911911
call_command("routine", "atomic-fail", "--non-atomic")
912-
self.assertEqual(TestModel.objects.get(id=0).name, "Name3")
913-
self.assertEqual(TestModel.objects.count(), 1)
912+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name3")
913+
self.assertEqual(_TestModel.objects.count(), 1)
914914

915915
def test_force_atomic_continue_fail(self):
916916
call_command("routine", "atomic-fail", "--continue")
917-
self.assertEqual(TestModel.objects.get(id=0).name, "Name3")
918-
self.assertEqual(TestModel.objects.count(), 1)
917+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name3")
918+
self.assertEqual(_TestModel.objects.count(), 1)
919919

920920
def test_continue_on_error(self):
921921
call_command("routine", "test-continue")
922-
self.assertEqual(TestModel.objects.get(id=0).name, "Name3")
922+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name3")
923923

924924
def test_force_halt_on_error(self):
925925
with self.assertRaises(TestError):
926926
call_command("routine", "test-continue", "--halt")
927-
self.assertEqual(TestModel.objects.get(id=0).name, "Name1")
927+
self.assertEqual(_TestModel.objects.get(id=0).name, "Name1")
928928

929929

930930
class Test(CoreTests, TestCase):

0 commit comments

Comments
 (0)