def test_requests_completed(self, mock_cloud):
conn = PiCloudConnection()
conn.connect()
mock_request_queue = Mock()
conn._request_queue = mock_request_queue
mock_result_queue = Mock()
conn._result_queue = mock_result_queue
def set_queue_info(request_count=0,
request_processing_jobs=0,
request_queued_jobs=0,
result_count=0):
mock_request_queue.info.return_value = dict(
count=request_count,
processing_jobs=request_processing_jobs,
queued_jobs=request_queued_jobs
)
mock_result_queue.count.return_value = result_count
set_queue_info()
ok_(conn._requests_completed())
set_queue_info(request_count=1)
ok_(not conn._requests_completed())
set_queue_info(request_processing_jobs=1)
ok_(not conn._requests_completed())
set_queue_info(request_queued_jobs=1)
ok_(not conn._requests_completed())
set_queue_info(result_count=1)
ok_(not conn._requests_completed())
def test_requests_completed(self, mock_cloud):
conn = PiCloudConnection()
conn.connect()
mock_request_queue = Mock()
conn._request_queue = mock_request_queue
mock_result_queue = Mock()
conn._result_queue = mock_result_queue
def set_queue_info(request_count=0,
request_processing_jobs=0,
request_queued_jobs=0,
result_count=0):
mock_request_queue.info.return_value = dict(
count=request_count,
processing_jobs=request_processing_jobs,
queued_jobs=request_queued_jobs)
mock_result_queue.count.return_value = result_count
set_queue_info()
ok_(conn._requests_completed())
set_queue_info(request_count=1)
ok_(not conn._requests_completed())
set_queue_info(request_processing_jobs=1)
ok_(not conn._requests_completed())
set_queue_info(request_queued_jobs=1)
ok_(not conn._requests_completed())
set_queue_info(result_count=1)
ok_(not conn._requests_completed())
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()
