ソケット通信をJUnitでテストしたい時に、MockRestServiceServerのようなモックサーバーを立てることは可能か

Question

以下のようなソケット通信を行うメソッドをテストしたいと考えています。

public byte[] callSocketHogeServer(byte[] reqData){
  try(
        Socket socket = new Socket("hoge",1000);
        OutoutStream out = socket.getOutputStream();
        InputStream in = socket.getInputStream();){
    socket.setSoTimeout(3000);
    out.write(reqData,0,reqData.length);
    out.flush();

    // InputStreamからの読み込み

    return resData;

  } catch (IOException e){
    // 例外処理
  }
}

テスト観点としては、
・ソケット通信先から正常なデータ付きのレスポンスが返却されたときに正常処理かどうか確認する
・ソケット通信先が応答しない場合、タイムアウトにする
といったテストを行いたいです。
Rest通信にはMockRestServiceServerのようなモックサーバを作成するものが存在するのは知っているので、Socket通信についても同様にこのリクエストにはこう返すと設定できれば簡単にテストできると考えたのですが、その様なクラスはJunitにありますでしょうか?

Comments

[jimbe]

(回答が付いていないようなので)スレッドで簡単にサーバ機能を作っても良いのでは。

[makosankibu]

返信遅くなり申し訳ありません。
なるほどと思い、サーバー機能を実装しようとしたのですが、詰まってしまったので自己解決の方で対応しました。
提案ありがとうございます。

Answer 1

以下のようにInputStream に書き込んでからそれをモックに入れ込むことで解決しました。
※SocketFactoryはソケット生成用の自作クラスです。テストの為に作成しました。

@SpringBootTest
class HogeAppApplicationTests {

	@MockBean
	SocketFactory socketFactory

	// テスト対象
	@Autowired
	HogeAppApplication target;
	
	@Mock
	private Socket socket;

	@Mock
	private OutputStream out;

	@Mock
	private InputStream in;

	@BeforeEach
	public void setUp() throws IOException {
		when(socketFactory.createSocket("hoge", 9000)).thenReturn(socket);
		when(socket.getOutputStream()).thenReturn(out);
	}

	@Test
	void contextLoads() {
		byte[] reponse = "response".getBytes();
		in = new ByteArrayInputStream(response);

		when(socket.getInputStream()).thenReturn(in);
		try {
			byte[] reqData = "request".getBytes();
			target.callSocketHogeServer(reqData);
		} catch (Exception e) {
			fail();
		}
	}
}