You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.2 KiB
35 lines
1.2 KiB
2 years ago
|
import socket
|
||
|
from unittest.mock import MagicMock, patch
|
||
|
|
||
|
import pytest
|
||
|
|
||
|
from announce_server import get_ip_address
|
||
|
|
||
|
|
||
|
def test_get_ip_address():
|
||
|
with patch("socket.socket") as mock_socket:
|
||
|
# Create a MagicMock object for the socket object
|
||
|
mock_socket_instance = MagicMock()
|
||
|
mock_socket.return_value = mock_socket_instance
|
||
|
|
||
|
# Define the expected IP address
|
||
|
expected_ip = "192.168.1.100"
|
||
|
|
||
|
# Configure the mock socket instance to return the expected IP address
|
||
|
mock_socket_instance.getsockname.return_value = (expected_ip, 0)
|
||
|
|
||
|
# Test the get_ip_address function
|
||
|
result_ip = get_ip_address()
|
||
|
|
||
|
# Check if the result matches the expected IP address
|
||
|
assert result_ip == expected_ip
|
||
|
|
||
|
# Check if the socket object was created with the correct arguments
|
||
|
mock_socket.assert_called_once_with(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
||
|
# Check if the socket.connect method was called with the correct arguments
|
||
|
mock_socket_instance.connect.assert_called_once_with(("10.255.255.255", 1))
|
||
|
|
||
|
# Check if the socket.close method was called
|
||
|
mock_socket_instance.close.assert_called_once()
|