Scribd Logo
Upload

Welcome to Scribd!

  • 7 views

    Websocket Tests

    Uploaded by

    bohdan.shraier.ir.2023
    This document contains the code for a unit test that tests a friend request controller in a Spring Boot application. It sets up a PostgreSQL database container, creates sample student data, sends a friend request message between the students, and asserts that the recipient received the message by checking a future object. It connects to the STOMP websocket, subscribes to receive messages, and verifies the message content was received as expected.

    Copyright:

    © All Rights Reserved
    Download as TXT, PDF, TXT or read online from Scribd

    Websocket Tests

    Uploaded by

    bohdan.shraier.ir.2023
    7 views2 pages
    This document contains the code for a unit test that tests a friend request controller in a Spring Boot application. It sets up a PostgreSQL database container, creates sample student data, sends a friend request message between the students, and asserts that the recipient received the message by checking a future object. It connects to the STOMP websocket, subscribes to receive messages, and verifies the message content was received as expected.

    Original Title

    webSocket tests

    Copyright

    © © All Rights Reserved

    Available Formats

    TXT, PDF, TXT or read online from Scribd

    Did you find this document useful?

    Is this content inappropriate?

    This document contains the code for a unit test that tests a friend request controller in a Spring Boot application. It sets up a PostgreSQL database container, creates sample student data, sends a friend request message between the students, and asserts that the recipient received the message by checking a future object. It connects to the STOMP websocket, subscribes to receive messages, and verifies the message content was received as expected.

    Copyright:

    © All Rights Reserved
    Download as TXT, PDF, TXT or read online from Scribd
    Download as txt, pdf, or txt
    7 views2 pages

    Websocket Tests

    Uploaded by

    bohdan.shraier.ir.2023
    This document contains the code for a unit test that tests a friend request controller in a Spring Boot application. It sets up a PostgreSQL database container, creates sample student data, sends a friend request message between the students, and asserts that the recipient received the message by checking a future object. It connects to the STOMP websocket, subscribes to receive messages, and verifies the message content was received as expected.

    Copyright:

    © All Rights Reserved
    Download as TXT, PDF, TXT or read online from Scribd
    Download as txt, pdf, or txt
    You are on page 1of 2

    package dev.backend.tutor.

    controllers;

    import dev.backend.tutor.dtos.MessageDto;
    import dev.backend.tutor.entities.Student;
    import dev.backend.tutor.repositories.StudentRepository;
    import dev.backend.tutor.utils.StudentGenerator;
    import dev.backend.tutor.utils.WsTestUtils;
    import dev.backend.tutor.sevices.messages.MessageService;
    import org.junit.After;
    import org.junit.Before;
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.boot.test.context.SpringBootTest;
    import
    org.springframework.boot.testcontainers.service.connection.ServiceConnection;
    import org.springframework.messaging.simp.stomp.StompSession;
    import org.springframework.test.context.DynamicPropertyRegistry;
    import org.springframework.test.context.DynamicPropertySource;
    import org.springframework.test.context.junit4.SpringRunner;
    import org.springframework.web.socket.messaging.WebSocketStompClient;
    import org.testcontainers.containers.PostgreSQLContainer;
    import org.testcontainers.junit.jupiter.Container;
    import org.testcontainers.lifecycle.Startables;

    import java.util.concurrent.CompletableFuture;

    import static java.util.concurrent.TimeUnit.SECONDS;


    import static org.assertj.core.api.Assertions.assertThat;
    import static
    org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT;

    @RunWith(SpringRunner.class)
    @SpringBootTest(webEnvironment = RANDOM_PORT)
    public class FriendControllerTest {

    @Container
    @ServiceConnection
    static PostgreSQLContainer<?> postgreSQLContainer = new
    PostgreSQLContainer<>("postgres:alpine");

    @DynamicPropertySource
    public static void setUpThings(DynamicPropertyRegistry registry) {
    Startables.deepStart(postgreSQLContainer).join();
    }

    @Autowired
    private StudentRepository studentRepository;

    @Autowired
    private MessageService messageService;

    private WebSocketStompClient stompClient;


    private StompSession stompSession;
    private final WsTestUtils wsTestUtils = new WsTestUtils();

    @Before
    public void setUp() throws Exception {
    String wsUrl = "ws://localhost:8080" + "/add_friend";
    stompClient = wsTestUtils.createWebSocketClient();
    stompSession = stompClient.connectAsync(wsUrl, new
    WsTestUtils.MyStompSessionHandler()).get();
    }

    @After
    public void tearDown() {
    stompSession.disconnect();
    stompClient.stop();
    }

    @Test
    public void connectsToSocket() {
    assertThat(stompSession.isConnected()).isTrue();
    }
    record StudentAndResultKeeper(Student student, CompletableFuture<String>
    resultKeeper){}

    @Test
    public void receivesMessageFromSubscribedQueue() throws Exception {
    //given
    var senderStudentData = initStudent("senderLogin");
    var recipientStudentData = initStudent("recipientLogin");

    var senderStudent = senderStudentData.student;


    var recipientStudent = recipientStudentData.student;

    var resultKeeper = recipientStudentData.resultKeeper;

    MessageDto messageDto = messageService


    .messageDtoForFriendshipRequest(
    senderStudent.getUsername(),
    recipientStudent.getUsername());

    //when
    messageService.sendMessageToUser(recipientStudent.getUsername(),
    messageDto);

    //then
    var result = resultKeeper.get(2, SECONDS);
    System.out.println(result);
    assertThat(result).isEqualTo(messageDto.toString());
    }

    private StudentAndResultKeeper initStudent(String username) throws


    InterruptedException {
    CompletableFuture<String> resultKeeper = new CompletableFuture<>();
    Student student = StudentGenerator.generateStudent(username);
    studentRepository.save(student);
    stompSession.subscribe(
    "/user/"+student.getUsername()+"/queue/notifications",
    new WsTestUtils.MyStompFrameHandler(resultKeeper::complete));
    Thread.sleep(2000);
    return new StudentAndResultKeeper(student, resultKeeper);
    }
    }

    You might also like