1 files changed, 103 insertions, 0 deletions
diff --git a/python/mozperftest/mozperftest/tests/test_ir_schema.py b/python/mozperftest/mozperftest/tests/test_ir_schema.py
new file mode 100644
index 0000000000..41f3ad7804
--- /dev/null
+++ b/python/mozperftest/mozperftest/tests/test_ir_schema.py
@@ -0,0 +1,103 @@
+#!/usr/bin/env python
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+import mozunit
+import pytest
+from jsonschema.exceptions import ValidationError
+
+from mozperftest.metrics.utils import validate_intermediate_results
+
+
+def test_results_with_directory():
+    test_result = {"results": "path-to-results", "name": "the-name"}
+    validate_intermediate_results(test_result)
+
+
+def test_results_with_measurements():
+    test_result = {
+        "results": [
+            {"name": "metric-1", "values": [0, 1, 1, 0]},
+            {"name": "metric-2", "values": [0, 1, 1, 0]},
+        ],
+        "name": "the-name",
+    }
+    validate_intermediate_results(test_result)
+
+
+def test_results_with_suite_perfherder_options():
+    test_result = {
+        "results": [
+            {"name": "metric-1", "values": [0, 1, 1, 0]},
+            {"name": "metric-2", "values": [0, 1, 1, 0]},
+        ],
+        "name": "the-name",
+        "extraOptions": ["an-extra-option"],
+        "value": 9000,
+    }
+    validate_intermediate_results(test_result)
+
+
+def test_results_with_subtest_perfherder_options():
+    test_result = {
+        "results": [
+            {"name": "metric-1", "shouldAlert": True, "values": [0, 1, 1, 0]},
+            {"name": "metric-2", "alertThreshold": 1.0, "values": [0, 1, 1, 0]},
+        ],
+        "name": "the-name",
+        "extraOptions": ["an-extra-option"],
+        "value": 9000,
+    }
+    validate_intermediate_results(test_result)
+
+
+def test_results_with_bad_suite_property():
+    test_result = {
+        "results": "path-to-results",
+        "name": "the-name",
+        "I'll cause a failure,": "an expected failure",
+    }
+    with pytest.raises(ValidationError):
+        validate_intermediate_results(test_result)
+
+
+def test_results_with_bad_subtest_property():
+    test_result = {
+        "results": [
+            # Error is in "shouldalert", it should be "shouldAlert"
+            {"name": "metric-1", "shouldalert": True, "values": [0, 1, 1, 0]},
+            {"name": "metric-2", "alertThreshold": 1.0, "values": [0, 1, 1, 0]},
+        ],
+        "name": "the-name",
+        "extraOptions": ["an-extra-option"],
+        "value": 9000,
+    }
+    with pytest.raises(ValidationError):
+        validate_intermediate_results(test_result)
+
+
+def test_results_with_missing_suite_property():
+    test_result = {
+        # Missing "results"
+        "name": "the-name"
+    }
+    with pytest.raises(ValidationError):
+        validate_intermediate_results(test_result)
+
+
+def test_results_with_missing_subtest_property():
+    test_result = {
+        "results": [
+            # Missing "values"
+            {"name": "metric-2", "alertThreshold": 1.0}
+        ],
+        "name": "the-name",
+        "extraOptions": ["an-extra-option"],
+        "value": 9000,
+    }
+    with pytest.raises(ValidationError):
+        validate_intermediate_results(test_result)
+
+
+if __name__ == "__main__":
+    mozunit.main()