@@ -25,14 +25,30 @@ def test_os_info(self):
25
25
"grm_host" : "10.2.123.43" ,
26
26
"grm_port" : "3334" ,
27
27
}
28
+
29
+ # Case that includes an IP address but no protocol
28
30
arg = [expected ["grm_module" ], expected ["grm_host" ], expected ["grm_port" ]]
29
31
result = DefaultTestSelector ._parse_grm (":" .join (arg ))
30
32
self .assertEqual (result , expected )
31
33
34
+ # Case that includes an IP address but no protocol nor a no port
35
+ expected ["grm_port" ] = None
36
+ arg = [expected ["grm_module" ], expected ["grm_host" ]]
37
+ result = DefaultTestSelector ._parse_grm (":" .join (arg ))
38
+ self .assertEqual (result , expected )
39
+
40
+ # Case that includes an IP address and a protocol
32
41
expected ["grm_host" ] = "https://10.2.123.43"
42
+ expected ["grm_port" ] = "443"
33
43
arg = [expected ["grm_module" ], expected ["grm_host" ], expected ["grm_port" ]]
34
44
result = DefaultTestSelector ._parse_grm (":" .join (arg ))
35
45
self .assertEqual (result , expected )
36
46
47
+ # Case that includes an IP address and a protocol, but no port
48
+ expected ["grm_port" ] = None
49
+ arg = [expected ["grm_module" ], expected ["grm_host" ]]
50
+ result = DefaultTestSelector ._parse_grm (":" .join (arg ))
51
+ self .assertEqual (result , expected )
52
+
37
53
if __name__ == '__main__' :
38
54
unittest .main ()
0 commit comments