Skip to content

Commit 19e66af

Browse files
committed
Don't use star import
This is required for flake8 compliance.
1 parent c44e5ea commit 19e66af

File tree

1 file changed

+36
-23
lines changed

1 file changed

+36
-23
lines changed

tests/test_reportsizedeltas.py

Lines changed: 36 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,25 @@
11
import distutils.dir_util
2+
import json
3+
import os
4+
import tempfile
25
import unittest.mock
6+
import urllib
37

4-
from reportsizedeltas import *
8+
import reportsizedeltas
59

610

711
# noinspection PyUnresolvedReferences
812
class TestReportsizedeltas(unittest.TestCase):
913
# NOTE: the tests are run in order sorted by method name, not in the order below
1014

11-
set_verbosity(enable_verbosity=False)
15+
reportsizedeltas.set_verbosity(enable_verbosity=False)
1216

1317
# @unittest.skip("")
1418
def test_set_verbosity(self):
1519
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)
1923

2024
# @unittest.skip("")
2125
def test_report_size_deltas(self):
@@ -24,7 +28,8 @@ def test_report_size_deltas(self):
2428
artifact_folder_object = "test_artifact_folder_object"
2529
report = {"markdown": "test_markdown", "data": "test_data"}
2630

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")
2833

2934
json_data = [{"number": 1, "locked": True, "head": {"sha": "foo123", "ref": "asdf"}, "user": {"login": "1234"}},
3035
{"number": 2, "locked": True, "head": {"sha": "foo123", "ref": "asdf"},
@@ -90,7 +95,8 @@ def test_report_exists(self):
9095
pr_number = 42
9196
pr_head_sha = "foo123"
9297

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")
94100

95101
json_data = [{"body": "foo123"}, {"body": report_size_deltas.report_key_beginning + pr_head_sha + "foo"}]
96102
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):
114120
test_artifact_url = "test_artifact_url"
115121
run_id = "4567"
116122

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")
118125

119126
json_data = {"workflow_runs": [{"head_sha": "foo123", "id": "1234"}, {"head_sha": pr_head_sha, "id": run_id}]}
120127
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):
158165
archive_download_url = "archive_download_url"
159166
run_id = "1234"
160167

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")
162170

163171
json_data = {"artifacts": [{"name": artifact_name, "archive_download_url": archive_download_url},
164172
{"name": "bar123", "archive_download_url": "wrong_artifact_url"}]}
@@ -192,7 +200,8 @@ def test_generate_report(self):
192200
pr_number = 42
193201
repository_name = "test_user/test_repo"
194202

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")
196205

197206
artifact_folder_object = tempfile.TemporaryDirectory(prefix="test_reportsizedeltas-")
198207
try:
@@ -256,7 +265,8 @@ def test_comment_report(self):
256265
report_markdown = "test_report_markdown"
257266
repository_name = "test_user/test_repo"
258267

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")
260270

261271
report_size_deltas.http_request = unittest.mock.MagicMock()
262272

@@ -280,7 +290,7 @@ def test_api_request(self):
280290
request_parameters = "test_parameters"
281291
page_number = 1
282292

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")
284294

285295
report_size_deltas.get_json_response = unittest.mock.MagicMock(return_value=response_data)
286296

@@ -296,7 +306,7 @@ def test_get_json_response(self):
296306
response = {"headers": {"Link": None}, "body": "[]"}
297307
url = "test_url"
298308

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")
300310

301311
report_size_deltas.http_request = unittest.mock.MagicMock(return_value=response)
302312

@@ -332,7 +342,7 @@ def test_http_request(self):
332342
url = "test_url"
333343
data = "test_data"
334344

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")
336346

337347
report_size_deltas.raw_http_request = unittest.mock.MagicMock()
338348

@@ -349,8 +359,9 @@ def test_raw_http_request(self):
349359
data = "test_data"
350360
request = "test_request"
351361

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)
354365

355366
urllib.request.Request = unittest.mock.MagicMock(return_value=request)
356367
report_size_deltas.handle_rate_limiting = unittest.mock.MagicMock()
@@ -379,7 +390,7 @@ def test_raw_http_request(self):
379390

380391
# @unittest.skip("")
381392
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")
383394

384395
json_data = {"json_data": {"resources": {"core": {"remaining": 0, "reset": 1234, "limit": 42}}}}
385396
report_size_deltas.get_json_response = unittest.mock.MagicMock(return_value=json_data)
@@ -396,18 +407,20 @@ def test_handle_rate_limiting(self):
396407

397408
@unittest.skip("disabled because it causes a delay")
398409
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)))
400412

401413
# @unittest.skip("")
402414
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)))
404417

405418
# @unittest.skip("")
406419
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"))
411424

412425

413426
if __name__ == '__main__':

0 commit comments

Comments
 (0)