Add basic test coverage for some _repr_html_ cases #668/#771

Author: soxofaan

tests/rest/models/test_general.py

@@ -1,3 +1,6 @@
+import re
+
+import dirty_equals
 import pytest
 
 from openeo.rest.models.general import (
@@ -68,6 +71,13 @@ def test_federation_missing(self, data, expected):
         collections = CollectionListingResponse(data)
         assert collections.ext_federation_missing() == expected
 
+    def test_repr_html_basic(self):
+        data = {"collections": [{"id": "S2"}]}
+        collections = CollectionListingResponse(data)
+        assert collections._repr_html_() == dirty_equals.IsStr(
+            regex=r'.*<openeo-collections>.*"collections":\s*\[{"id":\s*"S2".*', regex_flags=re.DOTALL
+        )
+
 
 class TestProcessListingResponse:
     def test_basic(self):
@@ -108,6 +118,14 @@ def test_federation_missing(self, data, expected):
         assert processes.ext_federation_missing() == expected
 
 
+    def test_repr_html_basic(self):
+        data = {"processes": [{"id": "ndvi"}, {"id": "s2mask"}]}
+        processes = ProcessListingResponse(data)
+        assert processes._repr_html_() == dirty_equals.IsStr(
+            regex=r'.*<openeo-processes>.*"processes":\s*\[{"id":\s*"ndvi".*', regex_flags=re.DOTALL
+        )
+
+
 class TestJobListingResponse:
     def test_basic(self):
         data = {"jobs": [{"id": "job-01"}, {"id": "job-02"}]}
@@ -146,6 +164,13 @@ def test_federation_missing(self, data, expected):
         jobs = JobListingResponse(data)
         assert jobs.ext_federation_missing() == expected
 
+    def test_repr_html_basic(self):
+        data = {"jobs": [{"id": "job-01"}, {"id": "job-02"}]}
+        jobs = JobListingResponse(data)
+        assert jobs._repr_html_() == dirty_equals.IsStr(
+            regex=r'.*<openeo-data-table>.*"data":\s*\[{"id":\s*"job-01".*', regex_flags=re.DOTALL
+        )
+
 
 class TestLogsResponse:
     def test_basic(self):
@@ -187,3 +212,10 @@ def test_links(self):
     def test_federation_missing(self, data, expected):
         logs = LogsResponse(data)
         assert logs.ext_federation_missing() == expected
+
+    def test_repr_html_basic(self):
+        data = {"logs": [{"id": "log-01", "level": "info", "message": "hello"}]}
+        logs = LogsResponse(data)
+        assert logs._repr_html_() == dirty_equals.IsStr(
+            regex=r'.*<openeo-logs>.*"logs":\s*\[{"id":\s*"log-01".*', regex_flags=re.DOTALL
+        )

tests/rest/test_connection.py

@@ -10,6 +10,7 @@
 from contextlib import nullcontext
 from pathlib import Path
 
+import dirty_equals
 import pytest
 import requests
 import requests_mock
@@ -3601,6 +3602,29 @@ def test_list_collections_extra_metadata(requests_mock, caplog):
     assert "Partial collection listing: missing federation components: ['oeob']." in caplog.text
 
 
+def test_collection_metadata_repr_html(requests_mock):
+    requests_mock.get(
+        API_URL,
+        json={
+            "api_version": "1.0.0",
+            "federation": {"b1": {"url": "https://b1.test/"}, "b2": {"url": "https://b2.test/"}},
+        },
+    )
+    requests_mock.get(
+        API_URL + "collections/S2",
+        json={"id": "S2", "federation:backends": ["b2"]},
+    )
+    con = Connection(API_URL)
+    metadata = con.describe_collection("S2")
+    result = metadata._repr_html_()
+    assert result == dirty_equals.IsStr(
+        regex=r'.*<openeo-collection>.*"federation":.*"https://b1.test/".*', regex_flags=re.DOTALL
+    )
+    assert result == dirty_equals.IsStr(
+        regex=r'.*<openeo-collection>.*"federation:backends":\s*\["b2"\].*', regex_flags=re.DOTALL
+    )
+
+
 def test_describe_collection(requests_mock):
     requests_mock.get(API_URL, json={"api_version": "1.0.0"})
     requests_mock.get(

tests/rest/test_udp.py

@@ -1,5 +1,7 @@
+import re
 import warnings
 
+import dirty_equals
 import pytest
 
 import openeo
@@ -27,6 +29,14 @@ def test_describe(con100, requests_mock, test_data):
     assert details == expected_details
 
 
+def test_repr_html(con100, requests_mock):
+    requests_mock.get(API_URL + "/process_graphs/add1", json={"id": "add1"})
+    udp = con100.user_defined_process("add1")
+    assert udp._repr_html_() == dirty_equals.IsStr(
+        regex=r'.*<openeo-process>.*"process":\s*{"id":\s*"add1".*', regex_flags=re.DOTALL
+    )
+
+
 def test_store_simple(con100, requests_mock):
     requests_mock.get(API_URL + "/processes", json={"processes": [{"id": "add"}]})
 

tests/test_metadata.py

@@ -43,6 +43,15 @@
 )
 
 
+class TestCollectionMetadata:
+
+    def test_repr_html_basic(self):
+        metadata = CollectionMetadata({"hello": "world"})
+        assert metadata._repr_html_() == dirty_equals.IsStr(
+            regex=r'.*<openeo-collection>.*"hello":\s*"world".*</openeo-collection>.*', regex_flags=re.DOTALL
+        )
+
+
 def test_metadata_get():
     metadata = CollectionMetadata({"foo": "bar", "very": {"deeply": {"nested": {"path": {"to": "somewhere"}}}}})
     assert metadata.get("foo") == "bar"