def test_wait_for_data_when_timeout_occurred(self, select_function_mock): select_function_mock.return_value = [False] wait_time = 3 status = Network.wait_for_data(self.socket_mock, wait_time) select_function_mock.assert_called_once_with([self.socket_mock], [], [], wait_time) assert status == Network.WaitStatus.TIMEOUT
def test_wait_for_data_when_new_data_is_available(self, select_function_mock): select_function_mock.return_value = [True] wait_time = 3 status = Network.wait_for_data(self.socket_mock, wait_time) select_function_mock.assert_called_once_with([self.socket_mock], [], [], wait_time) assert status == Network.WaitStatus.NEW_DATA