src/test/java/com/organization/service/MessageBroadcastServiceTest.java - master - itokar98/java-diplom-networkchat

src/test/java/com/organization/service/MessageBroadcastServiceTest.java

81 строка2 KB

Ilya Tokar

First commit

21 дек 2025, 15:29

21 дек 2025, 15:293f8814c

package com.organization.service;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import java.io.PrintWriter;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.*;

class MessageBroadcastServiceTest {

    private MessageBroadcastService broadcastService;

    @Mock
    private PrintWriter writer1;

    @Mock
    private PrintWriter writer2;

    @Mock
    private PrintWriter writer3;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        broadcastService = new MessageBroadcastService();
    }

    @Test
    void broadcastMessage_MultipleClients_SendsToAllExceptSender() {
        // Arrange
        String message = "Hello World";
        broadcastService.addClient(writer1);
        broadcastService.addClient(writer2);
        broadcastService.addClient(writer3);

        // Act
        broadcastService.broadcastMessage(message, writer1);

        // Assert
        verify(writer2, times(1)).println(message);
        verify(writer3, times(1)).println(message);
        verify(writer1, never()).println(message);
    }

    @Test
    void addClient_NewClient_IncreasesCount() {
        // Act
        broadcastService.addClient(writer1);
        broadcastService.addClient(writer2);

        // Assert
        assertEquals(2, broadcastService.getClientCount());
    }

    @Test
    void removeClient_ExistingClient_DecreasesCount() {
        // Arrange
        broadcastService.addClient(writer1);
        broadcastService.addClient(writer2);

        // Act
        broadcastService.removeClient(writer1);

        // Assert
        assertEquals(1, broadcastService.getClientCount());
    }

    @Test
    void broadcastMessage_NoClients_DoesNothing() {
        // Act
        broadcastService.broadcastMessage("test", writer1);

        // Assert - No exception should be thrown
        assertTrue(true);
    }
}