add test for html instructions

Matt Sokoloff · Matt Sokoloff · commit ef4bdc8f7071 · 2022-01-10T20:06:42.000-05:00
diff --git a/tests/integration/test_project.py b/tests/integration/test_project.py
@@ -1,6 +1,7 @@
 import json
 
 import pytest
+import requests
 
 from labelbox import Project, LabelingFrontend
 from labelbox.exceptions import InvalidQueryError
@@ -97,7 +98,22 @@ def test_attach_instructions(client, project):
 
     with pytest.raises(ValueError) as exc_info:
         project.upsert_instructions('/tmp/file.invalid_file_extension')
-    assert "instructions_file must be a pdf. Found" in str(exc_info.value)
+    assert "instructions_file must be a pdf or html file. Found" in str(
+        exc_info.value)
+
+
+def test_html_instructions(configured_project):
+    html_file_path = '/tmp/instructions.html'
+    sample_html_str = "<html></html>"
+
+    with open(html_file_path, 'w') as file:
+        file.write(sample_html_str)
+
+    configured_project.upsert_instructions(html_file_path)
+    updated_ontology = configured_project.ontology().normalized
+
+    instructions = updated_ontology.pop('projectInstructions')
+    assert requests.get(instructions).text == sample_html_str
 
 
 def test_same_ontology_after_instructions(
