@@ -38,12 +38,10 @@ def test_post_update_no_options_registered(insights_config, insights_client, _is
38
38
Client run with no options.
39
39
If registered, exit with 0 exit code (don't kill parent)
40
40
"""
41
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
42
41
with raises (SystemExit ) as exc_info :
43
42
post_update ()
44
43
assert exc_info .value .code == 0
45
44
insights_client .return_value .get_machine_id .assert_called_once ()
46
- insights_client .return_value .get_registration_status .assert_called_once ()
47
45
insights_client .return_value .clear_local_registration .assert_not_called ()
48
46
insights_client .return_value .set_display_name .assert_not_called ()
49
47
@@ -56,27 +54,24 @@ def test_post_update_no_options_unregistered(insights_config, insights_client, _
56
54
Client run with no options.
57
55
If unregistered, exit with 101 exit code (kill parent)
58
56
"""
59
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
60
57
with raises (SystemExit ) as exc_info :
61
58
post_update ()
62
59
assert exc_info .value .code == 101
63
60
insights_client .return_value .clear_local_registration .assert_not_called ()
64
61
insights_client .return_value .set_display_name .assert_not_called ()
65
62
66
63
67
- @patch ("insights.client.phase.v1.isfile" , side_effect = [True ])
64
+ @patch ("insights.client.phase.v1.isfile" , side_effect = [False ])
68
65
@patch ("insights.client.phase.v1.InsightsClient" )
69
66
@patch_insights_config
70
67
def test_post_update_no_options_err_reg_check (insights_config , insights_client , _isfile ):
71
68
"""
72
69
Client run with no options.
73
70
If registration check fails, exit with 101 exit code
74
71
"""
75
- insights_client .return_value .get_registration_status = MagicMock (return_value = None )
76
72
with raises (SystemExit ) as exc_info :
77
73
post_update ()
78
74
assert exc_info .value .code == 101
79
- insights_client .return_value .get_registration_status .assert_called_once ()
80
75
insights_client .return_value .clear_local_registration .assert_not_called ()
81
76
insights_client .return_value .set_display_name .assert_not_called ()
82
77
@@ -90,11 +85,9 @@ def test_post_update_check_status_registered(insights_config, insights_client, _
90
85
If registered, exit with 100 exit code (kill parent)
91
86
"""
92
87
insights_config .return_value .load_all .return_value .status = True
93
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
94
88
with raises (SystemExit ) as exc_info :
95
89
post_update ()
96
90
assert exc_info .value .code == 100
97
- insights_client .return_value .get_registration_status .assert_called_once ()
98
91
insights_client .return_value .clear_local_registration .assert_not_called ()
99
92
insights_client .return_value .set_display_name .assert_not_called ()
100
93
@@ -108,7 +101,6 @@ def test_post_update_check_status_unregistered(insights_config, insights_client,
108
101
If unregistered, exit with 101 exit code (kill parent)
109
102
"""
110
103
insights_config .return_value .load_all .return_value .status = True
111
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
112
104
with raises (SystemExit ) as exc_info :
113
105
post_update ()
114
106
assert exc_info .value .code == 101
@@ -126,12 +118,10 @@ def test_post_update_register_registered(insights_config, insights_client, get_s
126
118
If registered, exit with 0 exit code
127
119
"""
128
120
insights_config .return_value .load_all .return_value .register = True
129
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
130
121
with raises (SystemExit ) as exc_info :
131
122
post_update ()
132
123
assert exc_info .value .code == 0
133
124
insights_client .return_value .get_machine_id .assert_called_once ()
134
- insights_client .return_value .get_registration_status .assert_called_once ()
135
125
insights_client .return_value .clear_local_registration .assert_not_called ()
136
126
insights_client .return_value .set_display_name .assert_not_called ()
137
127
get_scheduler .return_value .schedule .assert_called_once ()
@@ -166,15 +156,13 @@ def test_post_update_register_machineid(insights_config, insights_client, get_sc
166
156
Also enable scheduling.
167
157
"""
168
158
insights_config .return_value .load_all .return_value .register = True
169
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
170
159
with _mock_no_register_files_machineid_present ():
171
160
with raises (SystemExit ) as exc_info :
172
161
post_update ()
173
- assert exc_info .value .code == 101
174
- insights_client .return_value .get_registration_status .assert_called_once ()
162
+ assert exc_info .value .code == 0
175
163
insights_client .return_value .clear_local_registration .assert_not_called ()
176
164
insights_client .return_value .set_display_name .assert_not_called ()
177
- get_scheduler .return_value .schedule .assert_not_called ()
165
+ get_scheduler .return_value .schedule .assert_called_once ()
178
166
179
167
180
168
@contextmanager
@@ -200,13 +188,11 @@ def test_post_update_register_unregistered(insights_config, insights_client, get
200
188
Also enable scheduling.
201
189
"""
202
190
insights_config .return_value .load_all .return_value .register = True
203
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
204
191
with _mock_no_machineid_present ():
205
192
with raises (SystemExit ) as exc_info :
206
193
post_update ()
207
194
assert exc_info .value .code == 0
208
195
insights_client .return_value .get_machine_id .assert_called_once ()
209
- insights_client .return_value .get_registration_status .assert_not_called ()
210
196
insights_client .return_value .clear_local_registration .assert_not_called ()
211
197
insights_client .return_value .set_display_name .assert_not_called ()
212
198
get_scheduler .return_value .schedule .assert_called_once ()
@@ -223,12 +209,10 @@ def test_post_update_unregister_registered(insights_config, insights_client, get
223
209
Also disable scheduling.
224
210
"""
225
211
insights_config .return_value .load_all .return_value .unregister = True
226
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
227
212
insights_client .return_value .unregister = MagicMock (return_value = True )
228
213
with raises (SystemExit ) as exc_info :
229
214
post_update ()
230
215
assert exc_info .value .code == 100
231
- insights_client .return_value .get_registration_status .assert_called_once ()
232
216
insights_client .return_value .unregister .assert_called_once ()
233
217
insights_client .return_value .clear_local_registration .assert_not_called ()
234
218
insights_client .return_value .set_display_name .assert_not_called ()
@@ -245,7 +229,6 @@ def test_post_update_unregister_unregistered(insights_config, insights_client, g
245
229
If unregistered, exit with 101 exit code
246
230
"""
247
231
insights_config .return_value .load_all .return_value .unregister = True
248
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
249
232
insights_client .return_value .unregister = MagicMock (return_value = False )
250
233
with raises (SystemExit ) as exc_info :
251
234
post_update ()
@@ -267,7 +250,6 @@ def test_post_update_set_display_name_cli_no_register_unreg(insights_config, ins
267
250
"""
268
251
insights_config .return_value .load_all .return_value .display_name = True
269
252
insights_config .return_value .load_all .return_value ._cli_opts = ['display_name' ]
270
- insights_client .return_value .get_registration_status = MagicMock (return_value = False )
271
253
with raises (SystemExit ) as exc_info :
272
254
post_update ()
273
255
assert exc_info .value .code == 101
@@ -286,11 +268,9 @@ def test_post_update_set_display_name_cli_no_register_reg(insights_config, insig
286
268
"""
287
269
insights_config .return_value .load_all .return_value .display_name = True
288
270
insights_config .return_value .load_all .return_value ._cli_opts = ['display_name' ]
289
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
290
271
with raises (SystemExit ) as exc_info :
291
272
post_update ()
292
273
assert exc_info .value .code == 100
293
- insights_client .return_value .get_registration_status .assert_called_once ()
294
274
insights_client .return_value .set_display_name .assert_called_once ()
295
275
get_scheduler .return_value .schedule .assert_not_called ()
296
276
@@ -306,7 +286,6 @@ def test_post_update_set_display_name_cli_register(insights_config, insights_cli
306
286
Display name is not explicitly set here
307
287
"""
308
288
insights_config .return_value .load_all .return_value .register = True
309
- insights_client .return_value .get_registration_status = MagicMock (return_value = True )
310
289
with raises (SystemExit ) as exc_info :
311
290
post_update ()
312
291
assert exc_info .value .code == 0
@@ -327,7 +306,6 @@ def test_post_update_offline(insights_config, insights_client):
327
306
except SystemExit :
328
307
pass
329
308
insights_client .return_value .get_machine_id .assert_called_once ()
330
- insights_client .return_value .get_registration_status .assert_not_called ()
331
309
insights_client .return_value .clear_local_registration .assert_not_called ()
332
310
insights_client .return_value .set_display_name .assert_not_called ()
333
311
@@ -344,7 +322,6 @@ def test_post_update_no_upload(insights_config, insights_client):
344
322
except SystemExit :
345
323
pass
346
324
insights_client .return_value .get_machine_id .assert_called_once ()
347
- insights_client .return_value .get_registration_status .assert_not_called ()
348
325
insights_client .return_value .clear_local_registration .assert_not_called ()
349
326
insights_client .return_value .set_display_name .assert_not_called ()
350
327
0 commit comments