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)
