update tests

acm-uiuc · Aug 4, 2024 · dad9196 · dad9196
1 parent 699767c
commit dad9196
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/live_integration/test_checkMembership.py b/tests/live_integration/test_checkMembership.py
@@ -2,11 +2,18 @@ def test_valid_member(api_client):
     response = api_client.get("/api/v1/checkMembership?netId=dsingh14")
     assert response.status_code == 200
     assert response.json() == {"netId": "dsingh14", "isPaidMember": True}
+    assert response.headers['x-acm-membership-source']  # it should at least exist, may or may not be cached
+    response = api_client.get("/api/v1/checkMembership?netId=dsingh14")
+    assert response.status_code == 200
+    assert response.json() == {"netId": "dsingh14", "isPaidMember": True}
+    assert response.headers['x-acm-membership-source'] == 'dynamo'
 
 def test_invalid_member(api_client):
     response = api_client.get("/api/v1/checkMembership?netId=invalid")
     assert response.status_code == 200
     assert response.json() == {"netId": "invalid", "isPaidMember": False,}
+    assert response.headers['x-acm-membership-source'] == 'aad' # non-member results should always be double checked against aad
+
 
 def test_invalid_parameters(api_client):
     response = api_client.get("/api/v1/checkMembership")