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