Coverage for melissa/launcher/action.py: 93%

1#!/usr/bin/python3 

2 

3# Copyright (c) 2021-2022, Institut National de Recherche en Informatique et en Automatique (Inria) 

4# All rights reserved. 

5# 

6# Redistribution and use in source and binary forms, with or without 

7# modification, are permitted provided that the following conditions are met: 

8# 

9# * Redistributions of source code must retain the above copyright notice, this 

10# list of conditions and the following disclaimer. 

11# 

12# * Redistributions in binary form must reproduce the above copyright notice, 

13# this list of conditions and the following disclaimer in the documentation 

14# and/or other materials provided with the distribution. 

15# 

16# * Neither the name of the copyright holder nor the names of its 

17# contributors may be used to endorse or promote products derived from 

18# this software without specific prior written permission. 

19# 

20# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 

21# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 

22# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 

23# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE 

24# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 

25# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 

26# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 

27# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 

28# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 

29# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

30 

31from typing import Generic, Dict, List, Optional, TypeVar 

32 

33from melissa.utility.networking import ConnectionId 

34 

35from melissa.utility.message import Message 

36 

37JobT = TypeVar("JobT") 

38 

39 

40class Action: 

41 pass 

42 

43 

44class ConnectionClosure(Action): 

45 def __init__(self, cid: ConnectionId) -> None: 

46 self.cid = cid 

47 

48 

49class Exit(Action): 

50 def __init__(self, status: int) -> None: 

51 assert status >= 0 

52 self.status = status 

53 

54 

55class JobCancellation(Generic[JobT], Action): 

56 def __init__(self, jobs: List[JobT]) -> None: 

57 self.jobs = jobs 

58 

59 

60class JobSubmission(Action): 

61 def __init__( 

62 self, 

63 working_directory: str, 

64 commands: List[List[str]], 

65 environment: Dict[str, Optional[str]], 

66 job_name: str, 

67 ) -> None: 

68 self.working_directory = working_directory 

69 self.commands = commands 

70 self.environment = environment 

71 self.job_name = job_name 

72 self.client_id: int = 0 

73 

74 

75class ServerJobSubmission(JobSubmission): 

76 def __init__(self, *args, **kwargs) -> None: 

77 super().__init__(*args, **kwargs) 

78 

79 

80class ClientJobSubmission(JobSubmission): 

81 def __init__(self, client_id: int, *args, **kwargs) -> None: 

82 super().__init__(*args, **kwargs) 

83 self.client_id = client_id 

84 

85 

86class JobUpdate(Generic[JobT], Action): 

87 def __init__(self, jobs: List[JobT]) -> None: 

88 self.jobs = jobs 

89 

90 

91class MessageSending(Action): 

92 def __init__( 

93 self, 

94 cid: List[Optional[ConnectionId]], 

95 message: Message 

96 ) -> None: 

97 self.cid = cid 

98 self.message = message 

99 

100 

101class PostponeSubmission(Action): 

102 def __init__(self, *args, **kwargs) -> None: 

103 super().__init__(*args, **kwargs) 

104 

105 

106class ConnectionServer(Action): 

107 def __init__(self, cid: ConnectionId) -> None: 

108 self.cid = cid