bug 1407806: Add test for Vary: Cookie header

This header shouldn't be present, but is added by the BanMiddleware.
mdn · Oct 18, 2017 · b723ca6 · b723ca6
1 parent e275ec0
commit b723ca6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/kuma/attachments/tests/test_views.py b/kuma/attachments/tests/test_views.py
@@ -184,6 +184,7 @@ def test_legacy_redirect(client, file_attachment):
     assert response.status_code == 301
     assert 'Location' in response
     assert response['Location'] == file_attachment['attachment'].get_file_url()
+    assert response['Vary'] == 'Cookie'
 
 
 def test_edit_attachment_get(admin_client, root_doc):

diff --git a/kuma/wiki/tests/test_views_code.py b/kuma/wiki/tests/test_views_code.py
@@ -64,6 +64,7 @@ def test_code_sample(code_sample_doc, constance_config, client, settings):
     assert response.status_code == 200
     assert response['Access-Control-Allow-Origin'] == '*'
     assert response.content.startswith('<!DOCTYPE html>')
+    assert response['Vary'] == 'Cookie'
 
     normalized = normalize_html(response.content)
     expected = (
@@ -137,3 +138,4 @@ def test_raw_code_sample_file(code_sample_doc, constance_config,
     response = admin_client.get(file_url)
     assert response.status_code == 302
     assert response.url == attachment.get_file_url()
+    assert response['Vary'] == 'Cookie'