1 files changed, 24 insertions, 1 deletions
diff --git a/src/test/java/org/traccar/BaseTest.java b/src/test/java/org/traccar/BaseTest.java
index 35374a363..2eedd4bd8 100644
--- a/src/test/java/org/traccar/BaseTest.java
+++ b/src/test/java/org/traccar/BaseTest.java
@@ -1,5 +1,7 @@
 package org.traccar;
 
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
 import org.traccar.config.Config;
 import org.traccar.database.ConnectionManager;
 import org.traccar.database.IdentityManager;
@@ -7,7 +9,12 @@ import org.traccar.database.MediaManager;
 import org.traccar.database.StatisticsManager;
 import org.traccar.model.Device;
 
-import static org.mockito.Mockito.*;
+import static org.mockito.ArgumentMatchers.anyBoolean;
+import static org.mockito.ArgumentMatchers.anyInt;
+import static org.mockito.ArgumentMatchers.anyLong;
+import static org.mockito.Mockito.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 public class BaseTest {
 
@@ -20,7 +27,14 @@ public class BaseTest {
         var device = mock(Device.class);
         when(device.getId()).thenReturn(1L);
         var identityManager = mock(IdentityManager.class);
+        when(identityManager.getById(anyLong())).thenReturn(device);
         when(identityManager.getByUniqueId(any())).thenReturn(device);
+        when(identityManager.lookupAttributeBoolean(anyLong(), any(), anyBoolean(), anyBoolean(), anyBoolean()))
+                .thenAnswer(invocation -> invocation.getArguments()[2]);
+        when(identityManager.lookupAttributeString(anyLong(), any(), any(), anyBoolean(), anyBoolean()))
+                .thenAnswer(invocation -> invocation.getArguments()[2]);
+        when(identityManager.lookupAttributeInteger(anyLong(), any(), anyInt(), anyBoolean(), anyBoolean()))
+                .thenAnswer(invocation -> invocation.getArguments()[2]);
         decoder.setIdentityManager(identityManager);
         decoder.setConnectionManager(mock(ConnectionManager.class));
         decoder.setStatisticsManager(mock(StatisticsManager.class));
@@ -33,6 +47,15 @@ public class BaseTest {
     }
 
     protected <T extends BaseProtocolEncoder> T inject(T encoder) throws Exception {
+        var device = mock(Device.class);
+        when(device.getId()).thenReturn(1L);
+        when(device.getUniqueId()).thenReturn("123456789012345");
+        var identityManager = mock(IdentityManager.class);
+        when(identityManager.getDevicePassword(anyLong(), any(), any()))
+                .thenAnswer(invocation -> invocation.getArguments()[2]);
+        when(identityManager.getById(anyLong())).thenReturn(device);
+        when(identityManager.getByUniqueId(any())).thenReturn(device);
+        encoder.setIdentityManager(identityManager);
         return encoder;
     }