def setUp(self):
op = messages.Operation(
consumerId=_TEST_CONSUMER_ID,
operationName=_TEST_OP_NAME
)
self.test_check_request = messages.CheckRequest(operation=op)
self.test_op = op
def _make_test_request(service_name, importance=None):
if importance is None:
importance = messages.Operation.ImportanceValueValuesEnum.LOW
op = messages.Operation(consumerId=_TEST_CONSUMER_ID,
operationName=_TEST_OP_NAME,
importance=importance)
check_request = messages.CheckRequest(operation=op)
return messages.ServicecontrolServicesCheckRequest(
serviceName=service_name, checkRequest=check_request)
def _make_test_request(service_name, importance=None, n=3, start=0):
if importance is None:
importance = messages.Operation.ImportanceValueValuesEnum.LOW
op_names = _make_op_names(n, start=start)
ops = [
messages.Operation(consumerId=_TEST_CONSUMER_ID,
operationName=op_name,
importance=importance) for op_name in op_names
]
if ops:
ops[0].labels = encoding.PyValueToMessage(
messages.Operation.LabelsValue,
{'key1': 'always add a label to the first op'})
report_request = messages.ReportRequest(operations=ops)
return messages.ServicecontrolServicesReportRequest(
serviceName=service_name, reportRequest=report_request)
def test_should_fail_if_operation_has_no_consumer_id(self):
op = messages.Operation(operationName=_TEST_OP_NAME)
testf = lambda: check_request.sign(messages.CheckRequest(operation=op))
expect(testf).to(raise_error(ValueError))
def test_should_fail_if_operation_has_no_operation_name(self):
op = messages.Operation(consumerId=_TEST_CONSUMER_ID)
testf = lambda: check_request.sign(messages.CheckRequest(operation=op))
expect(testf).to(raise_error(ValueError))
_WANTED_USER_AGENT = label_descriptor.USER_AGENT
_START_OF_EPOCH = timestamp.to_rfc3339(datetime.datetime(1970, 1, 1, 0, 0, 0))
_TEST_SERVICE_NAME = 'a_service_name'
_INFO_TESTS = [
(check_request.Info(operation_id='an_op_id',
operation_name='an_op_name',
referer='a_referer',
service_name=_TEST_SERVICE_NAME),
messages.Operation(
importance=messages.Operation.ImportanceValueValuesEnum.LOW,
labels=encoding.PyValueToMessage(
messages.Operation.LabelsValue, {
'servicecontrol.googleapis.com/user_agent':
_WANTED_USER_AGENT,
'servicecontrol.googleapis.com/referer': 'a_referer'
}),
operationId='an_op_id',
operationName='an_op_name',
startTime=_START_OF_EPOCH,
endTime=_START_OF_EPOCH)),
(check_request.Info(api_key='an_api_key',
api_key_valid=True,
operation_id='an_op_id',
operation_name='an_op_name',
referer='a_referer',
service_name=_TEST_SERVICE_NAME),
messages.Operation(
importance=messages.Operation.ImportanceValueValuesEnum.LOW,
consumerId='api_key:an_api_key',
labels=encoding.PyValueToMessage(
_EXPECTED_NOK_METRIC = metric_descriptor.KnownMetrics.CONSUMER_ERROR_COUNT
_ADD_LOG_TESTS = [
(report_request.Info(operation_id='an_op_id',
operation_name='an_op_name',
method='GET',
referer='a_referer',
backend_time=_TEST_LATENCY,
overhead_time=_TEST_LATENCY,
request_time=_TEST_LATENCY,
request_size=_TEST_SIZE,
response_size=_TEST_SIZE,
service_name=_TEST_SERVICE_NAME),
messages.Operation(
importance=messages.Operation.ImportanceValueValuesEnum.LOW,
logEntries=[_EXPECTED_OK_LOG_ENTRY],
operationId='an_op_id',
operationName='an_op_name',
startTime=_START_OF_EPOCH_TIMESTAMP,
endTime=_START_OF_EPOCH_TIMESTAMP)),
(report_request.Info(response_code=404,
operation_id='an_op_id',
operation_name='an_op_name',
method='GET',
referer='a_referer',
backend_time=_TEST_LATENCY,
overhead_time=_TEST_LATENCY,
request_time=_TEST_LATENCY,
request_size=_TEST_SIZE,
response_size=_TEST_SIZE,
service_name=_TEST_SERVICE_NAME),
messages.Operation(
_EXPECTED_NOK_METRIC = metric_descriptor.KnownMetrics.CONSUMER_ERROR_COUNT
_ADD_LOG_TESTS = [
(report_request.Info(operation_id='an_op_id',
operation_name='an_op_name',
method='GET',
referer='a_referer',
backend_time=_TEST_LATENCY,
overhead_time=_TEST_LATENCY,
request_time=_TEST_LATENCY,
request_size=_TEST_SIZE,
response_size=_TEST_SIZE,
service_name=_TEST_SERVICE_NAME),
messages.Operation(
importance=messages.Operation.ImportanceValueValuesEnum.LOW,
logEntries=[_EXPECTED_OK_LOG_ENTRY],
operationId='an_op_id',
operationName='an_op_name',
startTime=_START_OF_EPOCH_TIMESTAMP,
endTime=_START_OF_EPOCH_TIMESTAMP)),
(report_request.Info(response_code=404,
operation_id='an_op_id',
operation_name='an_op_name',
method='GET',
referer='a_referer',
backend_time=_TEST_LATENCY,
overhead_time=_TEST_LATENCY,
request_time=_TEST_LATENCY,
request_size=_TEST_SIZE,
response_size=_TEST_SIZE,
service_name=_TEST_SERVICE_NAME),
messages.Operation(
_LATER = timestamp.to_rfc3339(datetime.datetime(1980, 2, 2, 10, 0, 0))
_LATER_STILL = timestamp.to_rfc3339(datetime.datetime(1981, 2, 2, 10, 0, 0))
_TEST_LABELS = {
'key1': 'value1',
'key2': 'value2',
}
# in tests, the description field is not currently used, but should be filled
_TESTS = [{
'description':
'update the start time to that of the earliest',
'kinds':
None,
'initial':
messages.Operation(startTime=_EARLY, endTime=_LATER),
'ops': [
messages.Operation(startTime=_REALLY_EARLY, endTime=_LATER),
messages.Operation(startTime=_LATER, endTime=_LATER),
],
'want':
messages.Operation(startTime=_REALLY_EARLY, endTime=_LATER)
}, {
'description':
'update the end time to that of the latest',
'kinds':
None,
'initial':
messages.Operation(startTime=_EARLY, endTime=_LATER),
'ops': [
messages.Operation(startTime=_EARLY, endTime=_LATER),
