1 files changed, 33 insertions, 0 deletions
diff --git a/testing/mozbase/mozproxy/tests/test_recordings.py b/testing/mozbase/mozproxy/tests/test_recordings.py
new file mode 100644
index 0000000000..537a0eb7c2
--- /dev/null
+++ b/testing/mozbase/mozproxy/tests/test_recordings.py
@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+import os
+
+import mozunit
+from mozproxy.recordings import RecordingFile
+
+here = os.path.dirname(__file__)
+os.environ["MOZPROXY_DIR"] = os.path.join(here, "files")
+
+
+def test_recording_generation(*args):
+    test_file = os.path.join(here, "files", "new_file.zip")
+    file = RecordingFile(test_file)
+    with open(file.recording_path, "w") as recording:
+        recording.write("This is a recording")
+
+    file.set_metadata("test_file", True)
+    file.generate_zip_file()
+
+    assert os.path.exists(test_file)
+    os.remove(test_file)
+
+
+def test_recording_content(*args):
+    test_file = os.path.join(here, "files", "recording.zip")
+    file = RecordingFile(test_file)
+
+    assert file.metadata("test_file") is True
+    assert os.path.exists(file.recording_path)
+
+
+if __name__ == "__main__":
+    mozunit.main(runwith="pytest")