Add test for response header (#1058)

Follow up from: #1055
astronomer · Jun 20, 2024 · 47613f4 · 47613f4
1 parent 10471cd
commit 47613f4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/plugin/test_plugin.py b/tests/plugin/test_plugin.py
@@ -101,6 +101,8 @@ def test_dbt_docs_artifact(mock_open_file, monkeypatch, app, artifact):
     assert response.status_code == 200
     if artifact == "dbt_docs_index.html":
         assert iframe_script in _get_text_from_response(response)
+        assert "Content-Security-Policy" in response.headers
+        assert response.headers["Content-Security-Policy"] == "frame-ancestors 'self'"
 
 
 @pytest.mark.integration