def test_send(self, mock_cloud): conn = PiCloudConnection() conn.connect() requests = [request.Request('http://dummy') for n in range(3)] results = [Mock() for n in range(3)] results[0].request.id = requests[1].id results[1].request.id = requests[2].id results[2].request.id = requests[0].id conn._loop = Mock() conn._loop.return_value = results ret = conn.send(requests) conn._request_queue.push.assert_called_once_with(requests) conn._loop.assert_called_once_with() # results should be sorted in the order of requests eq_([results[2], results[0], results[1]], ret)
def test_loop(self, mock_cloud): conn = PiCloudConnection() conn.connect() mock_request_completed = Mock() mock_request_completed.side_effect = [False, True] conn._requests_completed = mock_request_completed mock_result_queue = Mock() result1 = Mock() result2 = Mock() mock_result_queue.pop.side_effect = [[result1], [result2]] + [None] * 2 conn._result_queue = mock_result_queue ret = conn._loop() eq_([result1, result2], ret) result1.run_callback.assert_called_once_with() result2.run_callback.assert_called_once_with()
def test_loop(self, mock_cloud): conn = PiCloudConnection() conn.connect() mock_request_completed = Mock() mock_request_completed.side_effect = [False, True] conn._requests_completed = mock_request_completed mock_result_queue = Mock() result1 = Mock() result2 = Mock() mock_result_queue.pop.side_effect = [[result1], [result2]] + [None] * 2 conn._result_queue = mock_result_queue ret = conn._loop() eq_([result1, result2], ret) result1.run_callback.assert_called_once_with() result2.run_callback.assert_called_once_with()
def test_send(self, mock_cloud): conn = PiCloudConnection() conn.connect() requests = [request.Request('http://dummy') for n in range(3)] results = [Mock() for n in range(3)] results[0].request.id = requests[1].id results[1].request.id = requests[2].id results[2].request.id = requests[0].id conn._loop = Mock() conn._loop.return_value = results ret = conn.send(requests) conn._request_queue.push.assert_called_once_with(requests) conn._loop.assert_called_once_with() # results should be sorted in the order of requests eq_([results[2], results[0], results[1]], ret)