From 014cf82a9e63a36e944e293932f9edf27e452919 Mon Sep 17 00:00:00 2001
From: Anton Tananaev <anton@traccar.org>
Date: Mon, 30 May 2022 14:04:24 -0700
Subject: No context dependency for decoder

---
 src/test/java/org/traccar/BaseTest.java | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'src/test/java/org/traccar/BaseTest.java')

diff --git a/src/test/java/org/traccar/BaseTest.java b/src/test/java/org/traccar/BaseTest.java
index 1dddbb03b..35374a363 100644
--- a/src/test/java/org/traccar/BaseTest.java
+++ b/src/test/java/org/traccar/BaseTest.java
@@ -1,9 +1,39 @@
 package org.traccar;
 
+import org.traccar.config.Config;
+import org.traccar.database.ConnectionManager;
+import org.traccar.database.IdentityManager;
+import org.traccar.database.MediaManager;
+import org.traccar.database.StatisticsManager;
+import org.traccar.model.Device;
+
+import static org.mockito.Mockito.*;
+
 public class BaseTest {
 
     static {
         Context.init(new TestIdentityManager());
     }
 
+    protected <T extends BaseProtocolDecoder> T inject(T decoder) throws Exception {
+        decoder.setConfig(new Config());
+        var device = mock(Device.class);
+        when(device.getId()).thenReturn(1L);
+        var identityManager = mock(IdentityManager.class);
+        when(identityManager.getByUniqueId(any())).thenReturn(device);
+        decoder.setIdentityManager(identityManager);
+        decoder.setConnectionManager(mock(ConnectionManager.class));
+        decoder.setStatisticsManager(mock(StatisticsManager.class));
+        decoder.setMediaManager(mock(MediaManager.class));
+        return decoder;
+    }
+
+    protected <T extends BaseFrameDecoder> T inject(T decoder) throws Exception {
+        return decoder;
+    }
+
+    protected <T extends BaseProtocolEncoder> T inject(T encoder) throws Exception {
+        return encoder;
+    }
+
 }
-- 
cgit v1.2.3