Merge pull request #15 from gabrielmscampos/feat/oms-proxy-client-tests

tests: add oms proxy client tests

tests/integration/test_h2d_client.py

@@ -8,7 +8,7 @@
 
 def test_get_h2d() -> None:
     dials = setup_dials_object()
-    data = dials.h2d.get(dataset_id=14677060, run_number=367112, ls_number=10, me_id=96)
+    data = dials.h2d.get(dataset_id=14677060, run_number=367112, ls_number=10, me_id=133)
     assert isinstance(data, LumisectionHistogram2D)
 
 

tests/integration/test_oms_proxy_client.py

@@ -0,0 +1,37 @@
+from cmsdials.filters import OMSFilter, OMSPage
+
+from .utils import setup_dials_object
+
+
+def test_query_runs() -> None:
+    dials = setup_dials_object()
+    data = dials.oms.query(
+        endpoint="runs", filters=[OMSFilter(attribute_name="run_number", value=382921, operator="EQ")]
+    )
+    assert isinstance(data, dict)
+    assert data["meta"]["totalResourceCount"] == 1
+
+
+def test_query_lumisections() -> None:
+    dials = setup_dials_object()
+    data = dials.oms.query(
+        endpoint="lumisections",
+        filters=[OMSFilter(attribute_name="run_number", value=382921, operator="EQ")],
+        pages=[OMSPage(attribute_name="limit", value=5000)],
+    )
+    assert isinstance(data, dict)
+    assert data["meta"]["totalResourceCount"] > 100
+
+
+def test_query_datasetrates() -> None:
+    dials = setup_dials_object()
+    data = dials.oms.query(
+        endpoint="datasetrates",
+        filters=[
+            OMSFilter(attribute_name="run_number", value=382921, operator="EQ"),
+            OMSFilter(attribute_name="dataset_name", value="ZeroBias", operator="EQ"),
+        ],
+        pages=[OMSPage(attribute_name="limit", value=5000)],
+    )
+    assert isinstance(data, dict)
+    assert data["meta"]["totalResourceCount"] > 100