@@ -601,32 +601,44 @@ class Migration(migrations.Migration):
601
601
assert result .ret == 0
602
602
603
603
604
- class Test_django_db_blocker :
605
- @pytest .mark .django_db
606
- def test_block_manually (self , django_db_blocker ):
607
- try :
608
- django_db_blocker .block ()
609
- with pytest .raises (RuntimeError ):
610
- Item .objects .exists ()
611
- finally :
612
- django_db_blocker .restore ()
604
+ def test_django_db_blocker (testdir ):
605
+ p1 = testdir .makepyfile ("""
606
+ import django.db.utils
607
+ import pytest
613
608
614
- @pytest .mark .django_db
615
- def test_block_with_block (self , django_db_blocker ):
616
- with django_db_blocker .block ():
617
- with pytest .raises (RuntimeError ):
618
- Item .objects .exists ()
609
+ from pytest_django_test.app.models import Item
619
610
620
- def test_unblock_manually (self , django_db_blocker ):
621
- try :
622
- django_db_blocker .unblock ()
623
- Item .objects .exists ()
624
- finally :
625
- django_db_blocker .restore ()
626
611
627
- def test_unblock_with_block (self , django_db_blocker ):
628
- with django_db_blocker .unblock ():
629
- Item .objects .exists ()
612
+ @pytest.mark.django_db
613
+ def test_block_manually(django_db_blocker):
614
+ try:
615
+ django_db_blocker.block()
616
+ with pytest.raises(RuntimeError):
617
+ Item.objects.exists()
618
+ finally:
619
+ django_db_blocker.restore()
620
+
621
+ @pytest.mark.django_db
622
+ def test_block_with_block(django_db_blocker):
623
+ with django_db_blocker.block():
624
+ with pytest.raises(RuntimeError):
625
+ Item.objects.exists()
626
+
627
+ def test_unblock_manually(django_db_blocker):
628
+ try:
629
+ django_db_blocker.unblock()
630
+ with pytest.raises(django.db.utils.OperationalError):
631
+ Item.objects.exists()
632
+ finally:
633
+ django_db_blocker.restore()
634
+
635
+ def test_unblock_with_block(django_db_blocker):
636
+ with django_db_blocker.unblock():
637
+ with pytest.raises(django.db.utils.OperationalError):
638
+ Item.objects.exists()
639
+ """ )
640
+ result = testdir .runpytest_subprocess ("-vv" , "-rA" , "--tb=short" , str (p1 ))
641
+ assert result .ret == 0
630
642
631
643
632
644
def test_mail (mailoutbox ):
0 commit comments