1
1
import distutils .dir_util
2
+ import json
3
+ import os
4
+ import tempfile
2
5
import unittest .mock
6
+ import urllib
3
7
4
- from reportsizedeltas import *
8
+ import reportsizedeltas
5
9
6
10
7
11
# noinspection PyUnresolvedReferences
8
12
class TestReportsizedeltas (unittest .TestCase ):
9
13
# NOTE: the tests are run in order sorted by method name, not in the order below
10
14
11
- set_verbosity (enable_verbosity = False )
15
+ reportsizedeltas . set_verbosity (enable_verbosity = False )
12
16
13
17
# @unittest.skip("")
14
18
def test_set_verbosity (self ):
15
19
with self .assertRaises (TypeError ):
16
- set_verbosity (enable_verbosity = 2 )
17
- set_verbosity (enable_verbosity = True )
18
- set_verbosity (enable_verbosity = False )
20
+ reportsizedeltas . set_verbosity (enable_verbosity = 2 )
21
+ reportsizedeltas . set_verbosity (enable_verbosity = True )
22
+ reportsizedeltas . set_verbosity (enable_verbosity = False )
19
23
20
24
# @unittest.skip("")
21
25
def test_report_size_deltas (self ):
@@ -24,7 +28,8 @@ def test_report_size_deltas(self):
24
28
artifact_folder_object = "test_artifact_folder_object"
25
29
report = {"markdown" : "test_markdown" , "data" : "test_data" }
26
30
27
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" , token = "foo" )
31
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" ,
32
+ token = "foo" )
28
33
29
34
json_data = [{"number" : 1 , "locked" : True , "head" : {"sha" : "foo123" , "ref" : "asdf" }, "user" : {"login" : "1234" }},
30
35
{"number" : 2 , "locked" : True , "head" : {"sha" : "foo123" , "ref" : "asdf" },
@@ -90,7 +95,8 @@ def test_report_exists(self):
90
95
pr_number = 42
91
96
pr_head_sha = "foo123"
92
97
93
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = artifact_name , token = "foo" )
98
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name ,
99
+ artifact_name = artifact_name , token = "foo" )
94
100
95
101
json_data = [{"body" : "foo123" }, {"body" : report_size_deltas .report_key_beginning + pr_head_sha + "foo" }]
96
102
report_size_deltas .api_request = unittest .mock .MagicMock (return_value = {"json_data" : json_data ,
@@ -114,7 +120,8 @@ def test_get_artifact_download_url_for_sha(self):
114
120
test_artifact_url = "test_artifact_url"
115
121
run_id = "4567"
116
122
117
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" , token = "foo" )
123
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" ,
124
+ token = "foo" )
118
125
119
126
json_data = {"workflow_runs" : [{"head_sha" : "foo123" , "id" : "1234" }, {"head_sha" : pr_head_sha , "id" : run_id }]}
120
127
report_size_deltas .api_request = unittest .mock .MagicMock (return_value = {"json_data" : json_data ,
@@ -158,7 +165,8 @@ def test_get_artifact_download_url_for_run(self):
158
165
archive_download_url = "archive_download_url"
159
166
run_id = "1234"
160
167
161
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = artifact_name , token = "foo" )
168
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name ,
169
+ artifact_name = artifact_name , token = "foo" )
162
170
163
171
json_data = {"artifacts" : [{"name" : artifact_name , "archive_download_url" : archive_download_url },
164
172
{"name" : "bar123" , "archive_download_url" : "wrong_artifact_url" }]}
@@ -192,7 +200,8 @@ def test_generate_report(self):
192
200
pr_number = 42
193
201
repository_name = "test_user/test_repo"
194
202
195
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" , token = "foo" )
203
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" ,
204
+ token = "foo" )
196
205
197
206
artifact_folder_object = tempfile .TemporaryDirectory (prefix = "test_reportsizedeltas-" )
198
207
try :
@@ -256,7 +265,8 @@ def test_comment_report(self):
256
265
report_markdown = "test_report_markdown"
257
266
repository_name = "test_user/test_repo"
258
267
259
- report_size_deltas = ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" , token = "foo" )
268
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = repository_name , artifact_name = "foo" ,
269
+ token = "foo" )
260
270
261
271
report_size_deltas .http_request = unittest .mock .MagicMock ()
262
272
@@ -280,7 +290,7 @@ def test_api_request(self):
280
290
request_parameters = "test_parameters"
281
291
page_number = 1
282
292
283
- report_size_deltas = ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
293
+ report_size_deltas = reportsizedeltas . ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
284
294
285
295
report_size_deltas .get_json_response = unittest .mock .MagicMock (return_value = response_data )
286
296
@@ -296,7 +306,7 @@ def test_get_json_response(self):
296
306
response = {"headers" : {"Link" : None }, "body" : "[]" }
297
307
url = "test_url"
298
308
299
- report_size_deltas = ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
309
+ report_size_deltas = reportsizedeltas . ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
300
310
301
311
report_size_deltas .http_request = unittest .mock .MagicMock (return_value = response )
302
312
@@ -332,7 +342,7 @@ def test_http_request(self):
332
342
url = "test_url"
333
343
data = "test_data"
334
344
335
- report_size_deltas = ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
345
+ report_size_deltas = reportsizedeltas . ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
336
346
337
347
report_size_deltas .raw_http_request = unittest .mock .MagicMock ()
338
348
@@ -349,8 +359,9 @@ def test_raw_http_request(self):
349
359
data = "test_data"
350
360
request = "test_request"
351
361
352
- report_size_deltas = ReportSizeDeltas (repository_name = user_name + "/" + repo_name , artifact_name = "foo" ,
353
- token = token )
362
+ report_size_deltas = reportsizedeltas .ReportSizeDeltas (repository_name = user_name + "/" + repo_name ,
363
+ artifact_name = "foo" ,
364
+ token = token )
354
365
355
366
urllib .request .Request = unittest .mock .MagicMock (return_value = request )
356
367
report_size_deltas .handle_rate_limiting = unittest .mock .MagicMock ()
@@ -379,7 +390,7 @@ def test_raw_http_request(self):
379
390
380
391
# @unittest.skip("")
381
392
def test_handle_rate_limiting (self ):
382
- report_size_deltas = ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
393
+ report_size_deltas = reportsizedeltas . ReportSizeDeltas (repository_name = "foo" , artifact_name = "foo" , token = "foo" )
383
394
384
395
json_data = {"json_data" : {"resources" : {"core" : {"remaining" : 0 , "reset" : 1234 , "limit" : 42 }}}}
385
396
report_size_deltas .get_json_response = unittest .mock .MagicMock (return_value = json_data )
@@ -396,18 +407,20 @@ def test_handle_rate_limiting(self):
396
407
397
408
@unittest .skip ("disabled because it causes a delay" )
398
409
def test_determine_urlopen_retry_true (self ):
399
- self .assertTrue (determine_urlopen_retry (exception = urllib .error .HTTPError (None , 502 , "Bad Gateway" , None , None )))
410
+ self .assertTrue (reportsizedeltas .determine_urlopen_retry (
411
+ exception = urllib .error .HTTPError (None , 502 , "Bad Gateway" , None , None )))
400
412
401
413
# @unittest.skip("")
402
414
def test_determine_urlopen_retry_false (self ):
403
- self .assertFalse (determine_urlopen_retry (exception = urllib .error .HTTPError (None , 404 , "Not Found" , None , None )))
415
+ self .assertFalse (reportsizedeltas .determine_urlopen_retry (
416
+ exception = urllib .error .HTTPError (None , 404 , "Not Found" , None , None )))
404
417
405
418
# @unittest.skip("")
406
419
def test_generate_value_cell (self ):
407
- self .assertEqual (" | :small_red_triangle: +42" , generate_value_cell (42 ))
408
- self .assertEqual (" | 0" , generate_value_cell (0 ))
409
- self .assertEqual (" | :green_heart: -42" , generate_value_cell (- 42 ))
410
- self .assertEqual (" | N/A" , generate_value_cell ("N/A" ))
420
+ self .assertEqual (" | :small_red_triangle: +42" , reportsizedeltas . generate_value_cell (42 ))
421
+ self .assertEqual (" | 0" , reportsizedeltas . generate_value_cell (0 ))
422
+ self .assertEqual (" | :green_heart: -42" , reportsizedeltas . generate_value_cell (- 42 ))
423
+ self .assertEqual (" | N/A" , reportsizedeltas . generate_value_cell ("N/A" ))
411
424
412
425
413
426
if __name__ == '__main__' :
0 commit comments