Refactor code. Tests still failing

tests/app/main/test_beta_redirect.py

@@ -0,0 +1,18 @@
+from flask import current_app
+from unittest.mock import patch
+
+from app import create_beta_url
+
+
+def test_create_beta_url():
+    url_for_redirect = create_beta_url("https://notify.gov/using-notify/get-started")
+    assert url_for_redirect == "https://beta.notify.gov/using-notify/get-started"
+
+
+def test_redirect_notify_to_beta(monkeypatch, client_request):
+    monkeypatch.setitem(current_app.config, "NOTIFY_ENVIRONMENT", "production")
+    # import pdb
+    # pdb.set_trace()
+    # resp = client_request.get_response_from_url("https://notify.gov/using-notify/get-started")
+    # assert resp.status_code == 301
+    assert current_app.config["NOTIFY_ENVIRONMENT"] == "production"

tests/app/test_event_handlers.py

@@ -1,7 +1,6 @@
 import uuid
 from unittest.mock import ANY
 
-from app import create_url
 from app.event_handlers import (
     create_add_user_to_service_event,
     create_archive_service_event,
@@ -130,15 +129,3 @@ def test_set_user_permissions(client_request, mock_events):
 
     create_set_user_permissions_event(**kwargs)
     mock_events.assert_called_with("set_user_permissions", event_dict(**kwargs))
-
-
-def test_create_url():
-    url_for_redirect = create_url(
-        "production", "https://notify.gov/using-notify/get-started"
-    )
-    assert url_for_redirect == "https://beta.notify.gov/using-notify/get-started"
-
-
-def test_create_url_non_production():
-    url = create_url("development", "https://notify.gov/using-notify/get-started")
-    assert url == "https://notify.gov/using-notify/get-started"