1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
package org.traccar;
import io.netty.channel.Channel;
import org.mockito.ArgumentCaptor;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.traccar.config.Config;
import org.traccar.session.ConnectionManager;
import org.traccar.database.IdentityManager;
import org.traccar.database.MediaManager;
import org.traccar.database.StatisticsManager;
import org.traccar.model.Device;
import org.traccar.session.DeviceSession;
import java.net.SocketAddress;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.TimeZone;
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 {
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.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);
var connectionManager = mock(ConnectionManager.class);
var uniqueIdsProvided = new HashSet<Boolean>();
when(connectionManager.getDeviceSession(any(), any(), any(), any())).thenAnswer(invocation -> {
var mock = new DeviceSession(1L, "", mock(Protocol.class), mock(Channel.class), mock(SocketAddress.class));
if (uniqueIdsProvided.isEmpty()) {
if (invocation.getArguments().length > 3) {
uniqueIdsProvided.add(true);
return mock;
}
return null;
} else {
return mock;
}
});
decoder.setConnectionManager(connectionManager);
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 {
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;
}
}
|
