1
+ import pytest
2
+
3
+ from leapp .libraries .common .config import version
1
4
from leapp .models import (
2
5
CopyFile ,
3
6
CryptoPolicyInfo ,
8
11
)
9
12
10
13
11
- def test_actor_execution_default (current_actor_context ):
14
+ @pytest .mark .parametrize (('target_version' , 'should_run' ), [
15
+ ('8' , False ),
16
+ ('9' , True ),
17
+ ('10' , True ),
18
+ ])
19
+ def test_actor_execution_default (monkeypatch , current_actor_context , target_version , should_run ):
20
+ monkeypatch .setattr (version , 'get_target_major_version' , lambda : target_version )
12
21
current_actor_context .feed (
13
22
CryptoPolicyInfo (
14
23
current_policy = "DEFAULT" ,
@@ -17,10 +26,17 @@ def test_actor_execution_default(current_actor_context):
17
26
)
18
27
)
19
28
current_actor_context .run ()
20
- assert not current_actor_context .consume (TargetUserSpacePreupgradeTasks )
29
+ if should_run :
30
+ assert not current_actor_context .consume (TargetUserSpacePreupgradeTasks )
21
31
22
32
23
- def test_actor_execution_legacy (current_actor_context ):
33
+ @pytest .mark .parametrize (('target_version' , 'should_run' ), [
34
+ ('8' , False ),
35
+ ('9' , True ),
36
+ ('10' , True ),
37
+ ])
38
+ def test_actor_execution_legacy (monkeypatch , current_actor_context , target_version , should_run ):
39
+ monkeypatch .setattr (version , 'get_target_major_version' , lambda : target_version )
24
40
current_actor_context .feed (
25
41
CryptoPolicyInfo (
26
42
current_policy = "LEGACY" ,
@@ -30,15 +46,22 @@ def test_actor_execution_legacy(current_actor_context):
30
46
)
31
47
current_actor_context .run ()
32
48
33
- assert current_actor_context .consume (TargetUserSpacePreupgradeTasks )
34
- u = current_actor_context .consume (TargetUserSpacePreupgradeTasks )[0 ]
35
- assert u .install_rpms == ['crypto-policies-scripts' ]
36
- assert u .copy_files == []
49
+ if should_run :
50
+ assert current_actor_context .consume (TargetUserSpacePreupgradeTasks )
51
+ u = current_actor_context .consume (TargetUserSpacePreupgradeTasks )[0 ]
52
+ assert u .install_rpms == ['crypto-policies-scripts' ]
53
+ assert u .copy_files == []
37
54
38
- assert current_actor_context .consume (Report )
55
+ assert current_actor_context .consume (Report )
39
56
40
57
41
- def test_actor_execution_custom (current_actor_context ):
58
+ @pytest .mark .parametrize (('target_version' , 'should_run' ), [
59
+ ('8' , False ),
60
+ ('9' , True ),
61
+ ('10' , True ),
62
+ ])
63
+ def test_actor_execution_custom (monkeypatch , current_actor_context , target_version , should_run ):
64
+ monkeypatch .setattr (version , 'get_target_major_version' , lambda : target_version )
42
65
current_actor_context .feed (
43
66
CryptoPolicyInfo (
44
67
current_policy = "CUSTOM:SHA2" ,
@@ -52,12 +75,13 @@ def test_actor_execution_custom(current_actor_context):
52
75
)
53
76
current_actor_context .run ()
54
77
55
- assert current_actor_context .consume (TargetUserSpacePreupgradeTasks )
56
- u = current_actor_context .consume (TargetUserSpacePreupgradeTasks )[0 ]
57
- assert u .install_rpms == ['crypto-policies-scripts' ]
58
- assert u .copy_files == [
59
- CopyFile (src = '/etc/crypto-policies/policies/CUSTOM.pol' ),
60
- CopyFile (src = '/etc/crypto-policies/policies/modules/SHA2.pmod' ),
61
- ]
78
+ if should_run :
79
+ assert current_actor_context .consume (TargetUserSpacePreupgradeTasks )
80
+ u = current_actor_context .consume (TargetUserSpacePreupgradeTasks )[0 ]
81
+ assert u .install_rpms == ['crypto-policies-scripts' ]
82
+ assert u .copy_files == [
83
+ CopyFile (src = '/etc/crypto-policies/policies/CUSTOM.pol' ),
84
+ CopyFile (src = '/etc/crypto-policies/policies/modules/SHA2.pmod' ),
85
+ ]
62
86
63
- assert current_actor_context .consume (Report )
87
+ assert current_actor_context .consume (Report )
0 commit comments