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 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))
def test_should_fail_if_operation_is_not_set(self):
testf = lambda: check_request.sign(messages.CheckRequest())
expect(testf).to(raise_error(ValueError))
def test_should_fail_if_operation_is_missing(self):
req = messages.ServicecontrolServicesCheckRequest(
serviceName=self.SERVICE_NAME,
checkRequest=messages.CheckRequest())
testf = lambda: self.agg.check(req)
expect(testf).to(raise_error(ValueError))
