def test_sets_reuseaddr(self): patch_socket(self) with udp_socket() as sock: pass self.assertEqual( [mock.call(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)], sock.setsockopt.mock_calls)
def test_yields_open_socket(self): patch_socket(self) with udp_socket() as sock: socket_calls = list(socket.socket.mock_calls) close_calls = list(sock.close.mock_calls) self.assertEqual([mock.call(socket.AF_INET, socket.SOCK_DGRAM)], socket_calls) self.assertEqual([], close_calls)
def test_closes_socket_on_exit(self): patch_socket(self) with udp_socket() as sock: pass self.assertEqual([mock.call()], sock.close.mock_calls)