99 lines
5.1 KiB
Java
99 lines
5.1 KiB
Java
package com.ruoyi.Dzp;
|
|
|
|
import com.ruoyi.Dzp.domain.basLed;
|
|
import com.ruoyi.Dzp.mapper.BasLedMapper;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.io.IOException;
|
|
import java.util.List;
|
|
import java.util.concurrent.*;
|
|
import static org.assertj.core.api.Assertions.*;
|
|
import static org.mockito.Mockito.*;
|
|
import static org.mockito.ArgumentMatchers.*;
|
|
|
|
class DzpSendTest {
|
|
@Test
|
|
void failedExchangeClosesItsSocketAndDoesNotAcknowledgeTheDisplay() throws Exception {
|
|
var socket = mock(MyUdpSocket.class);
|
|
var sender = new DzpSend() { @Override MyUdpSocket openSocket() { return socket; } };
|
|
var mapper = mock(BasLedMapper.class);
|
|
ReflectionTestUtils.setField(sender, "mapper", mapper);
|
|
var record = new basLed(); record.setModel("3"); record.setSendFlag("0"); record.setIpAddress("127.0.0.1"); record.setErrorMsg("alarm");
|
|
when(mapper.selectBasLedList(any())).thenAnswer(call -> call.getArgument(0, basLed.class).getId() == 1 ? List.of(record) : List.of());
|
|
try (var protocol = mockStatic(ProtocolDemo.class)) {
|
|
protocol.when(() -> ProtocolDemo.demo_play_chapterschedule2(socket, "127.0.0.1", "报警信息", "alarm"))
|
|
.thenThrow(new IllegalStateException("test send failure")).thenReturn(true);
|
|
assertThatThrownBy(sender::ErrorLed).hasMessage("test send failure");
|
|
verify(socket).close();
|
|
verify(mapper, never()).updateBasLed(any());
|
|
assertThat(record.getSendFlag()).isEqualTo("0");
|
|
sender.ErrorLed();
|
|
verify(socket, times(2)).close();
|
|
verify(mapper).updateBasLed(record);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
void busyLocalPortDefersOtherTimerWithoutBlockingOrChangingRows() throws Exception {
|
|
var entered = new CountDownLatch(1);
|
|
var release = new CountDownLatch(1);
|
|
var sender = new DzpSend() {
|
|
@Override MyUdpSocket openSocket() throws IOException {
|
|
entered.countDown();
|
|
try { if (!release.await(3, TimeUnit.SECONDS)) throw new IOException("test timeout"); }
|
|
catch (InterruptedException e) { Thread.currentThread().interrupt(); throw new IOException(e); }
|
|
throw new IOException("fixture port failure");
|
|
}
|
|
};
|
|
var mapper = mock(BasLedMapper.class);
|
|
ReflectionTestUtils.setField(sender, "mapper", mapper);
|
|
var record = new basLed(); record.setModel("1"); record.setSendFlag("0");
|
|
when(mapper.selectBasLedList(any())).thenReturn(List.of(record));
|
|
var executor = Executors.newFixedThreadPool(2);
|
|
try {
|
|
var first = executor.submit(() -> { assertThatThrownBy(sender::InLed).isInstanceOf(IOException.class); });
|
|
assertThat(entered.await(2, TimeUnit.SECONDS)).isTrue();
|
|
executor.submit(() -> { sender.ErrorLed(); return null; }).get(1, TimeUnit.SECONDS);
|
|
verify(mapper, times(1)).selectBasLedList(any());
|
|
verify(mapper, never()).updateBasLed(any());
|
|
release.countDown();
|
|
first.get(2, TimeUnit.SECONDS);
|
|
} finally { release.countDown(); executor.shutdownNow(); }
|
|
}
|
|
|
|
@Test
|
|
void eachScheduledMessageBuildsItsOwnPayload() throws Exception {
|
|
try (var payloads = mockConstruction(LEDSender2010.class,
|
|
(sender, context) -> when(sender.GetDataSize()).thenReturn((1 << 21) + 1))) {
|
|
var socket = mock(MyUdpSocket.class);
|
|
assertThat(ProtocolDemo.demo_play_chapterschedule2(socket, "127.0.0.1", "title1", "one")).isFalse();
|
|
assertThat(ProtocolDemo.demo_play_chapterschedule2(socket, "127.0.0.1", "title2", "two")).isFalse();
|
|
assertThat(payloads.constructed()).hasSize(2);
|
|
verify(payloads.constructed().get(0)).GetDataSize();
|
|
verify(payloads.constructed().get(1)).GetDataSize();
|
|
verifyNoInteractions(socket);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
void transientSocketErrorsStayInsideTheBoundedProtocolRetry() throws Exception {
|
|
var socket = mock(MyUdpSocket.class);
|
|
var payload = mock(LEDSender2010.class);
|
|
doThrow(new IOException("transient send")).doNothing().when(socket).send(anyString(), anyInt(), any(), anyInt());
|
|
when(socket.receive()).thenThrow(new java.io.UncheckedIOException(new IOException("transient receive"))).thenReturn(1);
|
|
when(payload.parse_trans_respond(any(), anyInt(), anyInt(), anyInt())).thenReturn(1);
|
|
assertThat(ProtocolDemo.send_data1(socket, payload, "127.0.0.1")).isTrue();
|
|
verify(socket, times(3)).send(anyString(), anyInt(), any(), anyInt());
|
|
verify(socket, times(3)).receive();
|
|
}
|
|
|
|
@Test
|
|
void persistentSendFailureStopsAfterFiveAttempts() throws Exception {
|
|
var socket = mock(MyUdpSocket.class);
|
|
doThrow(new IOException("unreachable")).when(socket).send(anyString(), anyInt(), any(), anyInt());
|
|
assertThat(ProtocolDemo.send_data1(socket, mock(LEDSender2010.class), "127.0.0.1")).isFalse();
|
|
verify(socket, times(5)).send(anyString(), anyInt(), any(), anyInt());
|
|
verify(socket, never()).receive();
|
|
}
|
|
}
|