fix a load of test

2023-11-27 12:08:45 +00:00
parent 95c136d3f3
commit 5cc57587aa
16 changed files with 209 additions and 67 deletions
@@ -44,6 +44,7 @@ def test_trainee_stuff(client, create_cid_manager, create_users):
    print(UserProfile.objects.all().count())

    assert UserProfile.objects.all().count() == 5
+    print(UserProfile.objects.filter(peninsula_trainee=True))
    assert UserProfile.objects.filter(peninsula_trainee=True).count() == 4

    response = client.get(reverse(f"trainees"))
@@ -59,9 +60,10 @@ def test_trainee_stuff(client, create_cid_manager, create_users):
    response = client.get(reverse(f"trainees"))
    assert response.status_code == 200

-    trainee_rows = BeautifulSoup(response.content, "html.parser").find(
+    trainee_rows = BeautifulSoup(response.content, "html.parser").find_all(
        "tr", {"class": "trainee"}
    )
+    print(trainee_rows)

    assert (
        len(trainee_rows) == UserProfile.objects.filter(peninsula_trainee=True).count()