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