def create_response(cls, quota, prod, non_prod, response_code=ResponseCode.OK):
return Response(
responseCode=response_code,
details=[ResponseDetail(message='test')],
result=Result(getQuotaResult=GetQuotaResult(
quota=quota, prodSharedConsumption=prod, nonProdSharedConsumption=non_prod))
)
def create_response(cls, quota, prod, non_prod, response_code=None):
response_code = ResponseCode.OK if response_code is None else response_code
resp = Response(responseCode=response_code,
details=[ResponseDetail(message='test')])
resp.result = Result(getQuotaResult=GetQuotaResult(
quota=quota, prodConsumption=prod, nonProdConsumption=non_prod))
return resp
def mock_get_quota(self, allocated, consumed, response_code=None):
response_code = ResponseCode.OK if response_code is None else response_code
resp = Response(responseCode=response_code, messageDEPRECATED='test')
resp.result = Result(getQuotaResult=GetQuotaResult(
quota=deepcopy(allocated), prodConsumption=deepcopy(consumed)))
self._scheduler.getQuota.return_value = resp
def setup_mock_quota_call_with_consumption(cls, mock_context):
api = mock_context.get_api('west')
response = cls.create_simple_success_response()
response.result = Result(getQuotaResult=GetQuotaResult(
quota=ResourceAggregate(resources=frozenset([
Resource(numCpus=5),
Resource(ramMb=20480),
Resource(diskMb=40960)
])),
prodSharedConsumption=ResourceAggregate(resources=frozenset([
Resource(numCpus=1),
Resource(ramMb=512),
Resource(diskMb=1024)
])),
prodDedicatedConsumption=ResourceAggregate(resources=frozenset([
Resource(numCpus=2),
Resource(ramMb=1024),
Resource(diskMb=2048)
])),
nonProdSharedConsumption=ResourceAggregate(resources=frozenset([
Resource(numCpus=3),
Resource(ramMb=2048),
Resource(diskMb=4096)
])),
nonProdDedicatedConsumption=ResourceAggregate(resources=frozenset([
Resource(numCpus=4),
Resource(ramMb=4096),
Resource(diskMb=8192)
])),
))
api.get_quota.return_value = response
def setup_mock_quota_call_no_consumption(cls, mock_context):
api = mock_context.get_api('west')
response = cls.create_simple_success_response()
response.result = Result(getQuotaResult=GetQuotaResult(
quota=ResourceAggregate(numCpus=5, ramMb=20480, diskMb=40960),
prodConsumption=None,
nonProdConsumption=None))
api.get_quota.return_value = response