test_server.py

import unittest
from unittest.mock import MagicMock, patch
from server import Server

class TestGuessingGameServer(unittest.TestCase):

    @patch("guessing_game_server.socket")
    def test_game_flow_correct_guess(self, mock_socket_module):
        #mock socket and connection
        mock_server_socket = MagicMock()
        mock_conn = MagicMock()

        #simulate correct guess directly
        mock_conn.recv.side_effect = [b'50', b'']  #one guess=disconnect
        mock_socket_module.socket.return_value = mock_server_socket
        mock_server_socket.accept.return_value = (mock_conn, ("127.0.0.1", 12345))

        #use fixed secret number for predictability
        server = Server(secret_number=50, socket_module=mock_socket_module)

        #run once only (simulate a single connection), then break loop
        with patch.object(server, 'server_socket', mock_server_socket):
            with patch("builtins.print"):  # suppress output
                server.start()
        
        #assert expected behavior
        mock_conn.sendall.assert_called_with(b'Correct! You win!: 1 guesses = very good')

    @patch("guessing_game_server.socket")
    def test_invalid_input(self, mock_socket_module):
        mock_server_socket = MagicMock()
        mock_conn = MagicMock()
        mock_conn.recv.side_effect = [b'hello', b'']  #invalid input= disconnect
        mock_socket_module.socket.return_value = mock_server_socket
        mock_server_socket.accept.return_value = (mock_conn, ("127.0.0.1", 12345))

        server = Server(secret_number=42, socket_module=mock_socket_module)

        with patch.object(server, 'server_socket', mock_server_socket):
            with patch("builtins.print"):
                server.start()

        mock_conn.sendall.assert_called_with(b'Invalid input! Please enter a number.')

if __name__ == "__main__":
    unittest.main()