Coverage for ocp_resources/virtual_machine_migration_resource_quota.py: 0%

1# API reference: https://github.com/kubevirt/managed-tenant-quota#virtualmachinemigrationresourcequota 

2# TODO: update API reference when OCP doc is available 

3 

4from ocp_resources.resource import NamespacedResource 

5 

6 

7class VirtualMachineMigrationResourceQuota(NamespacedResource): 

8 api_group = NamespacedResource.ApiGroup.MTQ_KUBEVIRT_IO 

9 

10 def __init__( 

11 self, 

12 requests_cpu=None, 

13 requests_memory=None, 

14 limits_cpu=None, 

15 limits_memory=None, 

16 **kwargs, 

17 ): 

18 """ 

19 Create VirtualMachineMigrationResourceQuota object. 

20 

21 Args: 

22 requests_cpu (str, optional): Additional CPU requests 

23 limits_cpu (str, optional): Additional CPU limits 

24 requests_memory (str, optional): Additional Memory requests 

25 limits_memory (str, optional): Additional Memory limits 

26 

27 """ 

28 super().__init__(**kwargs) 

29 self.requests_cpu = requests_cpu 

30 self.requests_memory = requests_memory 

31 self.limits_cpu = limits_cpu 

32 self.limits_memory = limits_memory 

33 

34 def to_dict(self) -> None: 

35 super().to_dict() 

36 if not self.kind_dict and not self.yaml_file: 

37 additional_resources = self.res.setdefault("spec", {}).setdefault("additionalMigrationResources", {}) 

38 

39 if self.requests_cpu: 

40 additional_resources["requests.cpu"] = self.requests_cpu 

41 if self.requests_memory: 

42 additional_resources["requests.memory"] = self.requests_memory 

43 if self.limits_cpu: 

44 additional_resources["limits.cpu"] = self.limits_cpu 

45 if self.limits_memory: 

46 additional_resources["limits.memory"] = self.limits_memory