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