1 files changed, 34 insertions, 0 deletions
diff --git a/src/test/java/org/traccar/BaseTest.java b/src/test/java/org/traccar/BaseTest.java
new file mode 100644
index 000000000..0b2c616ce
--- /dev/null
+++ b/src/test/java/org/traccar/BaseTest.java
@@ -0,0 +1,34 @@
+package org.traccar;
+
+import io.netty.buffer.ByteBuf;
+import org.traccar.database.MediaManager;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class BaseTest {
+
+    public static class MockMediaManager extends MediaManager {
+        Map<String, ByteBuf> files = new HashMap<>();
+
+        MockMediaManager() {
+            super("");
+        }
+
+        @Override
+        public String writeFile(String uniqueId, ByteBuf buf, String extension) {
+            String fileName = uniqueId + "/mock." + extension;
+            files.put(fileName, buf);
+            return fileName;
+        }
+
+        public ByteBuf readFile(String fileName) {
+            return files.get(fileName);
+        }
+    }
+
+    static {
+        Context.init(new TestIdentityManager(), new MockMediaManager());
+    }
+
+}