lionwebsite-backend/src/test/java/com/lion/lionwebsite/Service/RemoteServiceTest.java

package com.lion.lionwebsite.Service;

import com.lion.lionwebsite.Dao.normal.GalleryMapper;
import com.lion.lionwebsite.Message.*;
import io.netty.channel.*;
import io.netty.channel.embedded.EmbeddedChannel;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

class RemoteServiceTest {
    private static ResponseMessage response(int messageId, byte result) {
        ResponseMessage message = new ResponseMessage();
        message.setMessageId(messageId);
        message.setResult(result);
        return message;
    }

    private RemoteService service() {
        return new RemoteService(mock(GalleryMapper.class), mock(PushService.class),
                mock(WebSocketService.class), mock(SubscriptionStandbySnapshotService.class));
    }

    @Test void immediateResponseHasRegisteredWaiter() {
        RemoteService service = service();
        EmbeddedChannel channel = new EmbeddedChannel(new ChannelOutboundHandlerAdapter() {
            @Override public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) {
                AbstractMessage request = (AbstractMessage) msg;
                ctx.fireChannelRead(response(request.messageId, (byte) 0));
                promise.setSuccess();
            }
        }, service.new MyChannelInboundHandlerAdapter());
        service.channel = channel;
        try {
            assertEquals(0, service.checkAvailability());
            assertTrue(service.promiseHashMap.isEmpty());
        } finally { service.shutdownResources(); channel.finishAndReleaseAll(); }
    }

    @Test void timeoutAndWriteFailureRemoveWaiters() {
        RemoteService service = service();
        EmbeddedChannel channel = new EmbeddedChannel(new ChannelOutboundHandlerAdapter() {
            @Override public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) {
                promise.setSuccess(); // no reply
            }
        });
        service.channel = channel;
        try {
            assertEquals(-1, service.sendRequest(new AvailableCheckMessage(), 1, TimeUnit.MILLISECONDS));
            assertTrue(service.promiseHashMap.isEmpty());
            channel.pipeline().addLast(new ChannelOutboundHandlerAdapter() {
                @Override public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) {
                    promise.setFailure(new IOException("test failure"));
                }
            });
            assertEquals(-1, service.sendRequest(new AvailableCheckMessage(), 1, TimeUnit.SECONDS));
            assertTrue(service.promiseHashMap.isEmpty());
        } finally { service.shutdownResources(); channel.finishAndReleaseAll(); }
    }
}