use of com.tvd12.ezyfoxserver.socket.EzySimplePackage in project ezyfox-server by youngmonkeys.
the class EzyWsResponseApiTest method responseTest.
@Test
public void responseTest() throws Exception {
// given
EzyObjectToStringEncoder encoder = mock(EzyObjectToStringEncoder.class);
EzyWsResponseApi sut = new EzyWsResponseApi(encoder);
EzyArray data = EzyEntityFactory.EMPTY_ARRAY;
EzySimplePackage pack = new EzySimplePackage();
pack.setData(data);
pack.setEncrypted(true);
pack.setTransportType(EzyTransportType.TCP);
int sessionCount = RandomUtil.randomSmallInt() + 1;
List<EzySession> sessions = RandomUtil.randomList(sessionCount, () -> {
EzySession session = mock(EzySession.class);
when(session.getConnectionType()).thenReturn(EzyConnectionType.WEBSOCKET);
return session;
});
pack.addRecipients(sessions);
// when
sut.response(pack);
// then
verify(encoder, times(1)).encode(data, String.class);
verify(encoder, times(0)).toMessageContent(data);
verify(encoder, times(0)).encryptMessageContent(any(byte[].class), any(byte[].class));
}
use of com.tvd12.ezyfoxserver.socket.EzySimplePackage in project ezyfox-server by youngmonkeys.
the class EzySocketResponseApiTest method secureResponseImmediateTest.
@Test
public void secureResponseImmediateTest() throws Exception {
// given
EzyArray data = EzyEntityFactory.EMPTY_ARRAY;
byte[] bytes = RandomUtil.randomShortByteArray();
EzyObjectToByteEncoder encoder = mock(EzyObjectToByteEncoder.class);
when(encoder.toMessageContent(data)).thenReturn(bytes);
EzySocketResponseApi sut = new EzySocketResponseApi(encoder);
EzySimplePackage pack = new EzySimplePackage();
pack.setData(data);
pack.setEncrypted(true);
pack.setTransportType(EzyTransportType.TCP);
int sessionCount = RandomUtil.randomSmallInt() + 1;
List<EzySession> sessions = RandomUtil.randomList(sessionCount, () -> {
EzySession session = mock(EzySession.class);
when(session.getConnectionType()).thenReturn(EzyConnectionType.SOCKET);
return session;
});
pack.addRecipients(sessions);
// when
sut.response(pack, true);
// then
verify(encoder, times(1)).toMessageContent(data);
verify(encoder, times(sessionCount)).encryptMessageContent(any(byte[].class), any(byte[].class));
}
use of com.tvd12.ezyfoxserver.socket.EzySimplePackage in project ezyfox-server by youngmonkeys.
the class EzySocketResponseApiTest method normalResponseTest.
@Test
public void normalResponseTest() throws Exception {
// given
EzyArray data = EzyEntityFactory.EMPTY_ARRAY;
EzyObjectToByteEncoder encoder = mock(EzyObjectToByteEncoder.class);
byte[] bytes = RandomUtil.randomShortByteArray();
when(encoder.encode(data)).thenReturn(bytes);
EzySocketResponseApi sut = new EzySocketResponseApi(encoder);
EzySimplePackage pack = new EzySimplePackage();
pack.setData(data);
pack.setEncrypted(false);
pack.setTransportType(EzyTransportType.TCP);
int sessionCount = RandomUtil.randomSmallInt() + 1;
List<EzySession> sessions = RandomUtil.randomList(sessionCount, () -> {
EzySession session = mock(EzySession.class);
when(session.getConnectionType()).thenReturn(EzyConnectionType.SOCKET);
return session;
});
pack.addRecipients(sessions);
// when
sut.response(pack);
// then
verify(encoder, times(1)).encode(data);
verify(encoder, times(0)).toMessageContent(data);
verify(encoder, times(0)).encryptMessageContent(any(byte[].class), any(byte[].class));
}
use of com.tvd12.ezyfoxserver.socket.EzySimplePackage in project ezyfox-server by youngmonkeys.
the class EzySendResponseImpl method newPackage.
protected EzySimplePackage newPackage(EzyArray data, boolean encrypted, EzyTransportType transportType) {
EzySimplePackage pack = new EzySimplePackage();
pack.setData(data);
pack.setEncrypted(encrypted);
pack.setTransportType(transportType);
return pack;
}
use of com.tvd12.ezyfoxserver.socket.EzySimplePackage in project ezyfox-server by youngmonkeys.
the class EzySendResponseImpl method execute.
@Override
public void execute(EzyResponse response, Collection<EzySession> recipients, boolean encrypted, boolean immediate, EzyTransportType transportType) {
boolean success = false;
EzyResponseApi responseApi = server.getResponseApi();
EzyArray data = response.serialize();
EzySimplePackage pack = newPackage(data, encrypted, transportType);
pack.addRecipients(recipients);
try {
responseApi.response(pack, immediate);
success = true;
} catch (Exception e) {
logger.error("send data: {}, to client: {} error", pack.getData(), getRecipientsNames(recipients), e);
} finally {
pack.release();
}
boolean debug = server.getSettings().isDebug();
if (debug && success && !ignoredLogCommands.contains(response.getCommand())) {
logger.debug("send to: {} data: {}", getRecipientsNames(recipients), data);
}
}
Aggregations