Coverage for ocp_resources/node_maintenance.py: 0%

1from ocp_resources.utils.constants import TIMEOUT_4MINUTES 

2from ocp_resources.resource import Resource 

3 

4 

5class NodeMaintenance(Resource): 

6 """ 

7 Node Maintenance object, inherited from Resource. 

8 """ 

9 

10 api_group = Resource.ApiGroup.NODEMAINTENANCE_KUBEVIRT_IO 

11 

12 def __init__( 

13 self, 

14 name=None, 

15 client=None, 

16 node=None, 

17 reason="TEST Reason", 

18 teardown=True, 

19 timeout=TIMEOUT_4MINUTES, 

20 yaml_file=None, 

21 delete_timeout=TIMEOUT_4MINUTES, 

22 **kwargs, 

23 ): 

24 super().__init__( 

25 name=name, 

26 client=client, 

27 teardown=teardown, 

28 timeout=timeout, 

29 yaml_file=yaml_file, 

30 delete_timeout=delete_timeout, 

31 **kwargs, 

32 ) 

33 self.node = node 

34 self.reason = reason 

35 

36 def to_dict(self) -> None: 

37 super().to_dict() 

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

39 assert self.node, "node is mandatory for create" 

40 self.res["spec"] = {"nodeName": self.node.name, "reason": self.reason}