Django-單元測驗物件的選擇不像預期的那樣作業

對于我的兩個模型，用戶和組，我有洗掉物件的視圖。他們每個人的代碼幾乎相同，它在應用程式中作業，但單元測驗有不同的結果：它適用于用戶，而不適用于組。
預先感謝您的建議。

以下是單元測驗：

class TestAdmUsers(TestCase):
    def setUp(self):
        self.company = create_dummy_company("Société de test")
        self.user_staff = create_dummy_user(self.company, "staff", admin=True)
        self.usr11 = create_dummy_user(self.company, "user11")
        self.usr12 = create_dummy_user(self.company, "user12", admin=True)
        self.usr13 = create_dummy_user(self.company, "user13")

        self.client.force_login(self.user_staff.user)

    def test_adm_delete_user(self):
        test_usercomp_id = self.usr13.id
        usrcomp = UserComp.objects.get(user__username="user13")
        self.assertEqual(usrcomp.id, test_usercomp_id)

        url = reverse("polls:adm_delete_user", args=[self.company.comp_slug, test_usercomp_id])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 302)
        with self.assertRaises(User.DoesNotExist):
            User.objects.get(id=test_usercomp_id)
        with self.assertRaises(UserComp.DoesNotExist):
            UserComp.objects.get(id=test_usercomp_id)

class TestAdmGroups(TestCase):
    def setUp(self):
        self.company = create_dummy_company("Société de test")

        self.user_staff = create_dummy_user(self.company, "staff", admin=True)
        self.usr11 = create_dummy_user(self.company, "user11")
        self.usr12 = create_dummy_user(self.company, "user12", admin=True)
        self.usr13 = create_dummy_user(self.company, "user13")
        self.usr14 = create_dummy_user(self.company, "user14")

        user_list = [self.usr11.id, self.usr12.id, self.usr13.id, self.usr14.id]
        users = UserComp.objects.filter(id__in=user_list)
        self.group1 = UserGroup.create_group({
            "company": self.company,
            "group_name": "Groupe 1",
            "weight": 40,
            },
            user_list=users)

    def test_adm_delete_group(self):
        test_group_id = self.group1.id
        grp = UserGroup.objects.get(group_name="Groupe 1")
        self.assertEqual(grp.id, test_group_id)

        url = reverse("polls:adm_delete_group", args=[self.company.comp_slug, test_group_id])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 302)
        with self.assertRaises(UserGroup.DoesNotExist):
            UserGroup.objects.get(id=test_group_id)

該測驗對用戶運行良好，但我對組有此錯誤：

回溯（最近一次呼叫）：檔案“D:\Mes 檔案\Informatique\Developpement\Votes AG\projet_votes\polls\tests_admin.py”，第 354 行，在 test_adm_delete_group UserGroup.objects.get(id=test_group_id) 斷言錯誤：DoesNotExist沒有提出

以下是相關型號：

class UserComp(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, verbose_name="Utilisateur")
    company = models.ForeignKey(Company, on_delete=models.CASCADE, verbose_name="Société")
    phone_regex = RegexValidator(regex=r'^0[0-9]([ .-]?[0-9]{2}){4}$', message=("Format de numéro de téléphone invalide"))
    phone_num = models.CharField("numéro de téléphone", validators=[phone_regex], max_length=14, null=True, blank=True)
    is_admin = models.BooleanField("administrateur", default=False)

class UserGroup(models.Model):
    company = models.ForeignKey(
        Company, on_delete=models.CASCADE, verbose_name="société"
    )
    users = models.ManyToManyField(UserComp, verbose_name="utilisateurs", blank=True)
    group_name = models.CharField("nom", max_length=100)
    weight = models.IntegerField("poids", default=0)
    hidden = models.BooleanField(default=False)

和意見：

@user_passes_test(lambda u: u.is_superuser or (u.id is not None and u.usercomp.is_admin))
def adm_delete_user(request, comp_slug, usr_id):
    del_usr = User.objects.get(pk=usr_id)
    msg = "Utilisateur {0} {1} supprimé.".\
            format(del_usr.last_name, del_usr.first_name)

    User.objects.get(pk=usr_id).delete()

    messages.success(request, msg)
    return redirect("polls:adm_users", comp_slug=comp_slug)

@user_passes_test(lambda u: u.is_superuser or (u.id is not None and u.usercomp.is_admin))
def adm_delete_group(request, comp_slug, grp_id):
    del_grp = UserGroup.objects.get(pk=grp_id)
    msg = "Groupe {0} supprimé.".format(del_grp.group_name)

    UserGroup.objects.get(pk=grp_id).delete()

    messages.success(request, msg)
    return redirect("polls:adm_groups", comp_slug=comp_slug)

uj5u.com熱心網友回復：

您的TestAdmGroups測驗班上沒有登錄人員，因此它很可能不會運行洗掉，但會失敗user_passes_test（回傳 302）。

所以要修復，只需登錄員工用戶TestAdmGroups：

class TestAdmGroups(TestCase):
    ...
    self.client.force_login(self.user_staff.user)

標籤：Python 姜戈 单元测试 django-views django-unittest

上一篇：HowtosubstitutePrismContainerLocator.ContainerinNUnit

下一篇：我如何測驗Json.parser沒有被mockitojava呼叫？