def trigger_service_fault_async(self, service_fault, **kwargs):
fault = service_fault.fault_id.fault_name
if fault in self.service_fault_counts:
self.service_fault_counts[fault] += 1
else:
self.service_fault_counts[fault] = 1
return service_fault_pb2.TriggerServiceFaultResponse()
def test_trigger_service_fault_error():
# Test unset header error
response = service_fault_pb2.TriggerServiceFaultResponse()
assert isinstance(_trigger_service_fault_error(response), UnsetStatusError)
# Test header error
response.header.error.code = response.header.error.CODE_INTERNAL_SERVER_ERROR
assert isinstance(_trigger_service_fault_error(response), InternalServerError)
# Test unset status error
response.header.error.code = response.header.error.CODE_OK
assert isinstance(_trigger_service_fault_error(response), UnsetStatusError)
# Test status error
response.status = response.STATUS_FAULT_ALREADY_ACTIVE
assert isinstance(_trigger_service_fault_error(response), ServiceFaultAlreadyExistsError)
# Test OK
response.status = response.STATUS_OK
assert not _trigger_service_fault_error(response)