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())
