add tests

app/service_invite/rest.py

@@ -226,7 +226,7 @@ def validate_service_invitation_token(token):
 
 
 @service_invite.route("/service/invite/redis/<redis_key>", methods=["GET"])
-def get_redis_data(redis_key):
+def get_service_invite_data(redis_key):
     service_invite_data = redis_store.raw_get(redis_key)
     # We can't log this because key may contain PII (email address)
     if service_invite_data is None:

tests/app/service_invite/test_service_invite_rest.py

@@ -398,3 +398,35 @@ def test_get_invited_user_404s_if_invite_doesnt_exist(
         _expected_status=404,
     )
     assert json_resp["result"] == "error"
+
+
+def test_get_service_invite_data_with_invite(
+    admin_request,
+    mocker
+):
+    redis_key =  "service-invite-j.k@fake.gov"
+    expected_user_data = b'{"from_user_id": ["7480cdcf-fa31-42b8-a4bf-2cd4d7a9b4f4"], "service_id": "721b0aa6-2447-4bcd-91fc-26d576f2bbff", "permissions": ["manage_api_keys"], "folder_permissions": []}' # noqa
+    expected_status = 200
+
+    mocker.patch("app.service_invite.rest.redis_store.raw_get", return_value=expected_user_data)
+    json_resp = json.loads(admin_request.get(
+        "service_invite.get_service_invite_data",
+        redis_key=redis_key,
+        _expected_status=expected_status,
+    ))
+    assert json_resp["permissions"] == ["manage_api_keys"]
+
+
+def test_get_service_invite_data_without_invite(
+    admin_request,
+    mocker
+):
+    redis_key =  "service-invite-j.k@fake.gov"
+
+
+    mocker.patch("app.service_invite.rest.redis_store.raw_get", return_value=None)
+    with pytest.raises(Exception):
+        json_resp = json.loads(admin_request.get(
+            "service_invite.get_service_invite_data",
+            redis_key=redis_key,
+        ))