|
5 | 5 | from model_mommy import mommy |
6 | 6 |
|
7 | 7 | from pythonpro.cohorts.models import Cohort, LiveClass |
8 | | -from pythonpro.django_assertions import dj_assert_contains, dj_assert_template_used |
| 8 | +from pythonpro.django_assertions import dj_assert_contains |
9 | 9 |
|
10 | 10 |
|
11 | 11 | @pytest.fixture |
@@ -45,24 +45,22 @@ def test_cohort_title(cohort, resp): |
45 | 45 |
|
46 | 46 | @pytest.fixture |
47 | 47 | def resp_client(client_with_client, live_class: LiveClass, mocker, logged_user): |
48 | | - tag_as = mocker.patch('pythonpro.cohorts.views.tag_as') |
49 | | - yield client_with_client.get(reverse('cohorts:live_class', kwargs={'pk': live_class.id}), secure=True) |
50 | | - tag_as.assert_called_once_with(logged_user.email, 'potencial-member') |
| 48 | + return client_with_client.get(reverse('cohorts:live_class', kwargs={'pk': live_class.id}), secure=True) |
51 | 49 |
|
52 | 50 |
|
53 | 51 | def test_live_class_landing_for_client(cohort, resp_client): |
54 | | - dj_assert_template_used(resp_client, 'cohorts/live_class_landing_page.html') |
| 52 | + assert resp_client.status_code == 302 |
| 53 | + assert resp_client.url == reverse('payments:member_landing_page') |
55 | 54 |
|
56 | 55 |
|
57 | 56 | @pytest.fixture |
58 | 57 | def resp_lead(client_with_lead, live_class: LiveClass, mocker, logged_user): |
59 | | - tag_as = mocker.patch('pythonpro.cohorts.views.tag_as') |
60 | | - yield client_with_lead.get(reverse('cohorts:live_class', kwargs={'pk': live_class.id}), secure=True) |
61 | | - tag_as.assert_called_once_with(logged_user.email, 'potencial-member') |
| 58 | + return client_with_lead.get(reverse('cohorts:live_class', kwargs={'pk': live_class.id}), secure=True) |
62 | 59 |
|
63 | 60 |
|
64 | 61 | def test_live_class_landing_for_lead(cohort, resp_lead): |
65 | | - dj_assert_template_used(resp_lead, 'cohorts/live_class_landing_page.html') |
| 62 | + assert resp_lead.status_code == 302 |
| 63 | + assert resp_lead.url == reverse('payments:member_landing_page') |
66 | 64 |
|
67 | 65 |
|
68 | 66 | def test_cohort_url(cohort: Cohort, resp): |
|
0 commit comments