🎨 [pre-commit.ci] Auto format from pre-commit.com hooks

astronomer · Dec 2, 2023 · 0f2ca10 · 0f2ca10
1 parent 71c30ac
commit 0f2ca10
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/plugin/test_plugin.py b/tests/plugin/test_plugin.py
@@ -65,9 +65,11 @@ def conf_get(section, key, *args, **kwargs):
     mock_open_file.assert_called_once()
     assert response.status_code == 200
     if artifact == "dbt_docs_index.html":
-        assert iframe_script in response.text
-
-
+        # Airflow < 2.4 uses an old version of Werkzeug that does not have Response.text.
+        if not hasattr(response, "text"):
+            assert iframe_script in response.get_data(as_text=True)
+        else:
+            assert iframe_script in response.text
 
 
 @pytest.mark.parametrize("artifact", ["dbt_docs_index.html", "manifest.json", "catalog.json"])