def test__refresh_http(): api_request = mock.Mock( return_value={'name': TEST_OPERATION_NAME, 'done': True}) result = operation._refresh_http(api_request, TEST_OPERATION_NAME) assert result.name == TEST_OPERATION_NAME assert result.done is True api_request.assert_called_once_with( method='GET', path='operations/{}'.format(TEST_OPERATION_NAME))
def test__refresh_http(): api_request = mock.Mock(return_value={"name": TEST_OPERATION_NAME, "done": True}) result = operation._refresh_http(api_request, TEST_OPERATION_NAME) assert result.name == TEST_OPERATION_NAME assert result.done is True api_request.assert_called_once_with( method="GET", path="operations/{}".format(TEST_OPERATION_NAME) )
def test__refresh_http_w_retry(): json_response = {"name": TEST_OPERATION_NAME, "done": True} api_request = mock.Mock() retry = mock.Mock() retry.return_value.return_value = json_response result = operation._refresh_http(api_request, TEST_OPERATION_NAME, retry=retry) assert isinstance(result, operations_pb2.Operation) assert result.name == TEST_OPERATION_NAME assert result.done is True api_request.assert_not_called() retry.assert_called_once_with(api_request) retry.return_value.assert_called_once_with( method="GET", path="operations/{}".format(TEST_OPERATION_NAME) )