creating new lead landing page that not redirects to OTO

moacirmoda · moacirmoda · commit ef211429b7b8 · 2020-02-27T13:43:06.000-03:00
diff --git a/pythonpro/core/tests/test_lead_landing_page.py b/pythonpro/core/tests/test_lead_landing_page.py
@@ -178,3 +178,19 @@ def test_should_redirect_to_OTO_page(resp_lead_creation):
 
 def test_should_has_a_lite_version(client):
     assert client.get(reverse('core:lead_landing_lite'), secure=True)
+
+
+@pytest.fixture
+def resp_lead_creation_without_OTO(client, db, fake: Faker, create_lead_mock, email):
+    return client.post(
+        reverse('core:lead_landing_without_OTO'),
+        data={
+            'first_name': fake.name(),
+            'email': email,
+        },
+        secure=True
+    )
+
+
+def test_should_has_a_version_that_not_redirect_to_OTO_page(resp_lead_creation_without_OTO):
+    assert resp_lead_creation_without_OTO['Location'] == reverse('core:thanks')
diff --git a/pythonpro/core/urls.py b/pythonpro/core/urls.py
@@ -12,6 +12,7 @@
     path('podcast', views.podcast, name='podcast'),
     path('curso-de-python-gratis', views.lead_landing, name='lead_landing'),
     path('curso-de-python-gratis-lite', views.lead_landing_lite, name='lead_landing_lite'),
+    path('curso-de-python-gratis-no-oto', views.lead_form_without_OTO, name='lead_landing_without_OTO'),
     path('cadastro-python-birds', views.lead_form, name='lead_form'),
     path('definir-senha', views.lead_change_password, name='lead_change_password'),
     path('obrigado', views.thanks, name='thanks'),
diff --git a/pythonpro/core/views.py b/pythonpro/core/views.py
@@ -148,16 +148,31 @@ def programmer_week_ty(request):
     return render(request, 'core/lead_landing_page.html', context={'form': UserSignupForm()})
 
 
-def lead_form(request):
+def _lead_form(request, redirect_to_OTO=True, *args, **kwargs):
     if request.method == 'GET':
         form = UserSignupForm()
         return render(request, 'core/lead_form_errors.html', context={'form': form})
+
     source = request.GET.get('utm_source', default='unknown')
     first_name = request.POST.get('first_name')
     email = request.POST.get('email')
+
     try:
         user = user_facade.register_lead(first_name, email, source)
     except user_facade.UserCreationException as e:
         return render(request, 'core/lead_form_errors.html', context={'form': e.form}, status=400)
+
     login(request, user)
-    return redirect(reverse('payments:client_landing_page_oto'))
+
+    if redirect_to_OTO:
+        return redirect(reverse('payments:client_landing_page_oto'))
+    else:
+        return redirect(reverse('core:thanks'))
+
+
+def lead_form(request):
+    return _lead_form(request)
+
+
+def lead_form_without_OTO(request):
+    return _lead_form(request, redirect_to_OTO=False)
