1 files changed, 153 insertions, 0 deletions
diff --git a/python/mozperftest/mozperftest/tests/conftest.py b/python/mozperftest/mozperftest/tests/conftest.py
new file mode 100644
index 0000000000..19435c5369
--- /dev/null
+++ b/python/mozperftest/mozperftest/tests/conftest.py
@@ -0,0 +1,153 @@
+import json
+import os
+import pathlib
+import sys
+from unittest import mock
+
+import pytest
+
+from mozperftest.metrics.notebook.perftestetl import PerftestETL
+from mozperftest.metrics.notebook.perftestnotebook import PerftestNotebook
+from mozperftest.tests.support import HERE, get_running_env
+from mozperftest.utils import temp_dir
+
+
+@pytest.fixture
+def patched_mozperftest_tools():
+    tools_mock = mock.MagicMock(name="tools-mock")
+    _module = mock.MagicMock(name="mozperftest_tools")
+    _module.SideBySide.return_value = tools_mock
+
+    try:
+        sys.modules["mozperftest_tools.side_by_side"] = _module
+        yield tools_mock
+    finally:
+        del sys.modules["mozperftest_tools.side_by_side"]
+
+
+@pytest.fixture(scope="session", autouse=True)
+def data():
+    data_1 = {
+        "browserScripts": [
+            {"timings": {"firstPaint": 101}},
+            {"timings": {"firstPaint": 102}},
+            {"timings": {"firstPaint": 103}},
+        ],
+    }
+
+    data_2 = {
+        "browserScripts": [
+            {"timings": {"firstPaint": 201}},
+            {"timings": {"firstPaint": 202}},
+            {"timings": {"firstPaint": 203}},
+        ],
+    }
+
+    data_3 = {
+        "browserScripts": [
+            {"timings": {"firstPaint": 301}},
+            {"timings": {"firstPaint": 302}},
+            {"timings": {"firstPaint": 303}},
+        ],
+    }
+
+    yield {"data_1": data_1, "data_2": data_2, "data_3": data_3}
+
+
+@pytest.fixture(scope="session", autouse=True)
+def standarized_data():
+    return {
+        "browsertime": [
+            {
+                "data": [
+                    {"value": 1, "xaxis": 1, "file": "file_1"},
+                    {"value": 2, "xaxis": 2, "file": "file_2"},
+                ],
+                "name": "name",
+                "subtest": "subtest",
+            }
+        ]
+    }
+
+
+@pytest.fixture(scope="session", autouse=True)
+def files(data):
+    # Create a temporary directory.
+    with temp_dir() as td:
+        tmp_path = pathlib.Path(td)
+
+    dirs = {
+        "resources": tmp_path / "resources",
+        "output": tmp_path / "output",
+    }
+
+    for d in dirs.values():
+        d.mkdir(parents=True, exist_ok=True)
+
+    # Create temporary data files for tests.
+    def _create_temp_files(path, data):
+        path.touch(exist_ok=True)
+        path.write_text(data)
+        return path.resolve().as_posix()
+
+    resources = {}
+    json_1 = dirs["resources"] / "file_1.json"
+    resources["file_1"] = _create_temp_files(json_1, json.dumps(data["data_1"]))
+
+    json_2 = dirs["resources"] / "file_2.json"
+    resources["file_2"] = _create_temp_files(json_2, json.dumps(data["data_2"]))
+
+    txt_3 = dirs["resources"] / "file_3.txt"
+    resources["file_3"] = _create_temp_files(txt_3, str(data["data_3"]))
+
+    output = dirs["output"] / "output.json"
+
+    yield {
+        "resources": resources,
+        "dirs": dirs,
+        "output": output,
+    }
+
+
+@pytest.fixture(scope="session", autouse=True)
+def ptetls(files):
+    resources, dirs, output = files["resources"], files["dirs"], files["output"]
+    _, metadata, _ = get_running_env()
+    config = {"output": output}
+    file_group_list = {"group_1": list(resources.values())}
+    file_group_str = {"group_1": dirs["resources"].resolve().as_posix()}
+
+    yield {
+        "ptetl_list": PerftestETL(
+            file_groups=file_group_list,
+            config=config,
+            prefix="PerftestETL",
+            logger=metadata,
+            sort_files=True,
+        ),
+        "ptetl_str": PerftestETL(
+            file_groups=file_group_str,
+            config=config,
+            prefix="PerftestETL",
+            logger=metadata,
+            sort_files=True,
+        ),
+    }
+
+
+@pytest.fixture(scope="session", autouse=True)
+def ptnb(standarized_data):
+    _, metadata, _ = get_running_env()
+    return PerftestNotebook(standarized_data, metadata, "PerftestNotebook")
+
+
+@pytest.fixture(scope="function", autouse=True)
+def perftestetl_plugin():
+
+    ret = HERE / "data" / "perftestetl_plugin"
+
+    os.environ["PERFTESTETL_PLUGIN"] = ret.resolve().as_posix()
+
+    yield ret
+
+    del os.environ["PERFTESTETL_PLUGIN"]