fix: format all code with black
and from now on should not deviate from that...
This commit is contained in:
parent
49f9a0228b
commit
a6bb538ce9
59 changed files with 2762 additions and 2131 deletions
|
|
@ -14,15 +14,15 @@ class TestProblemCheck(ConfigTestCase):
|
|||
|
||||
def test_system_key(self):
|
||||
check = self.make_check()
|
||||
self.assertRaises(AttributeError, getattr, check, 'system_key')
|
||||
self.assertRaises(AttributeError, getattr, check, "system_key")
|
||||
|
||||
def test_problem_key(self):
|
||||
check = self.make_check()
|
||||
self.assertRaises(AttributeError, getattr, check, 'problem_key')
|
||||
self.assertRaises(AttributeError, getattr, check, "problem_key")
|
||||
|
||||
def test_title(self):
|
||||
check = self.make_check()
|
||||
self.assertRaises(AttributeError, getattr, check, 'title')
|
||||
self.assertRaises(AttributeError, getattr, check, "title")
|
||||
|
||||
def test_find_problems(self):
|
||||
check = self.make_check()
|
||||
|
|
@ -44,8 +44,8 @@ class TestProblemCheck(ConfigTestCase):
|
|||
|
||||
|
||||
class FakeProblemCheck(mod.ProblemCheck):
|
||||
system_key = 'wuttatest'
|
||||
problem_key = 'fake_check'
|
||||
system_key = "wuttatest"
|
||||
problem_key = "fake_check"
|
||||
title = "Fake problem check"
|
||||
|
||||
# def find_problems(self):
|
||||
|
|
@ -69,7 +69,7 @@ class TestProblemHandler(ConfigTestCase):
|
|||
self.assertEqual(len(checks), 0)
|
||||
|
||||
# but let's configure our fake check
|
||||
self.config.setdefault('wutta.problems.modules', 'tests.test_problems')
|
||||
self.config.setdefault("wutta.problems.modules", "tests.test_problems")
|
||||
checks = self.handler.get_all_problem_checks()
|
||||
self.assertIsInstance(checks, list)
|
||||
self.assertEqual(len(checks), 1)
|
||||
|
|
@ -82,27 +82,31 @@ class TestProblemHandler(ConfigTestCase):
|
|||
self.assertEqual(len(checks), 0)
|
||||
|
||||
# but let's configure our fake check
|
||||
self.config.setdefault('wutta.problems.modules', 'tests.test_problems')
|
||||
self.config.setdefault("wutta.problems.modules", "tests.test_problems")
|
||||
checks = self.handler.filter_problem_checks()
|
||||
self.assertIsInstance(checks, list)
|
||||
self.assertEqual(len(checks), 1)
|
||||
|
||||
# filter by system_key
|
||||
checks = self.handler.filter_problem_checks(systems=['wuttatest'])
|
||||
checks = self.handler.filter_problem_checks(systems=["wuttatest"])
|
||||
self.assertEqual(len(checks), 1)
|
||||
checks = self.handler.filter_problem_checks(systems=['something_else'])
|
||||
checks = self.handler.filter_problem_checks(systems=["something_else"])
|
||||
self.assertEqual(len(checks), 0)
|
||||
|
||||
# filter by problem_key
|
||||
checks = self.handler.filter_problem_checks(problems=['fake_check'])
|
||||
checks = self.handler.filter_problem_checks(problems=["fake_check"])
|
||||
self.assertEqual(len(checks), 1)
|
||||
checks = self.handler.filter_problem_checks(problems=['something_else'])
|
||||
checks = self.handler.filter_problem_checks(problems=["something_else"])
|
||||
self.assertEqual(len(checks), 0)
|
||||
|
||||
# filter by both
|
||||
checks = self.handler.filter_problem_checks(systems=['wuttatest'], problems=['fake_check'])
|
||||
checks = self.handler.filter_problem_checks(
|
||||
systems=["wuttatest"], problems=["fake_check"]
|
||||
)
|
||||
self.assertEqual(len(checks), 1)
|
||||
checks = self.handler.filter_problem_checks(systems=['wuttatest'], problems=['bad_check'])
|
||||
checks = self.handler.filter_problem_checks(
|
||||
systems=["wuttatest"], problems=["bad_check"]
|
||||
)
|
||||
self.assertEqual(len(checks), 0)
|
||||
|
||||
def test_get_supported_systems(self):
|
||||
|
|
@ -113,14 +117,14 @@ class TestProblemHandler(ConfigTestCase):
|
|||
self.assertEqual(len(systems), 0)
|
||||
|
||||
# but let's configure our fake check
|
||||
self.config.setdefault('wutta.problems.modules', 'tests.test_problems')
|
||||
self.config.setdefault("wutta.problems.modules", "tests.test_problems")
|
||||
systems = self.handler.get_supported_systems()
|
||||
self.assertIsInstance(systems, list)
|
||||
self.assertEqual(systems, ['wuttatest'])
|
||||
self.assertEqual(systems, ["wuttatest"])
|
||||
|
||||
def test_get_system_title(self):
|
||||
title = self.handler.get_system_title('wutta')
|
||||
self.assertEqual(title, 'wutta')
|
||||
title = self.handler.get_system_title("wutta")
|
||||
self.assertEqual(title, "wutta")
|
||||
|
||||
def test_is_enabled(self):
|
||||
check = FakeProblemCheck(self.config)
|
||||
|
|
@ -129,7 +133,7 @@ class TestProblemHandler(ConfigTestCase):
|
|||
self.assertTrue(self.handler.is_enabled(check))
|
||||
|
||||
# config can disable
|
||||
self.config.setdefault('wutta.problems.wuttatest.fake_check.enabled', 'false')
|
||||
self.config.setdefault("wutta.problems.wuttatest.fake_check.enabled", "false")
|
||||
self.assertFalse(self.handler.is_enabled(check))
|
||||
|
||||
def test_should_run_for_weekday(self):
|
||||
|
|
@ -140,8 +144,8 @@ class TestProblemHandler(ConfigTestCase):
|
|||
self.assertTrue(self.handler.should_run_for_weekday(check, weekday))
|
||||
|
||||
# config can disable, e.g. for weekends
|
||||
self.config.setdefault('wutta.problems.wuttatest.fake_check.day5', 'false')
|
||||
self.config.setdefault('wutta.problems.wuttatest.fake_check.day6', 'false')
|
||||
self.config.setdefault("wutta.problems.wuttatest.fake_check.day5", "false")
|
||||
self.config.setdefault("wutta.problems.wuttatest.fake_check.day6", "false")
|
||||
for weekday in range(5):
|
||||
self.assertTrue(self.handler.should_run_for_weekday(check, weekday))
|
||||
for weekday in (5, 6):
|
||||
|
|
@ -152,10 +156,10 @@ class TestProblemHandler(ConfigTestCase):
|
|||
|
||||
organized = self.handler.organize_problem_checks(checks)
|
||||
self.assertIsInstance(organized, dict)
|
||||
self.assertEqual(list(organized), ['wuttatest'])
|
||||
self.assertIsInstance(organized['wuttatest'], dict)
|
||||
self.assertEqual(list(organized['wuttatest']), ['fake_check'])
|
||||
self.assertIs(organized['wuttatest']['fake_check'], FakeProblemCheck)
|
||||
self.assertEqual(list(organized), ["wuttatest"])
|
||||
self.assertIsInstance(organized["wuttatest"], dict)
|
||||
self.assertEqual(list(organized["wuttatest"]), ["fake_check"])
|
||||
self.assertIs(organized["wuttatest"]["fake_check"], FakeProblemCheck)
|
||||
|
||||
def test_find_problems(self):
|
||||
check = FakeProblemCheck(self.config)
|
||||
|
|
@ -165,7 +169,7 @@ class TestProblemHandler(ConfigTestCase):
|
|||
def test_get_email_key(self):
|
||||
check = FakeProblemCheck(self.config)
|
||||
key = self.handler.get_email_key(check)
|
||||
self.assertEqual(key, 'wuttatest_problems_fake_check')
|
||||
self.assertEqual(key, "wuttatest_problems_fake_check")
|
||||
|
||||
def test_get_global_email_context(self):
|
||||
context = self.handler.get_global_email_context()
|
||||
|
|
@ -175,44 +179,53 @@ class TestProblemHandler(ConfigTestCase):
|
|||
check = FakeProblemCheck(self.config)
|
||||
problems = []
|
||||
context = self.handler.get_check_email_context(check, problems)
|
||||
self.assertEqual(context, {'system_title': 'wuttatest'})
|
||||
self.assertEqual(context, {"system_title": "wuttatest"})
|
||||
|
||||
def test_send_problem_report(self):
|
||||
check = FakeProblemCheck(self.config)
|
||||
problems = []
|
||||
with patch.object(self.app, 'send_email') as send_email:
|
||||
with patch.object(self.app, "send_email") as send_email:
|
||||
self.handler.send_problem_report(check, problems)
|
||||
send_email.assert_called_once_with('wuttatest_problems_fake_check', {
|
||||
'system_title': 'wuttatest',
|
||||
'config': self.config,
|
||||
'app': self.app,
|
||||
'check': check,
|
||||
'problems': problems,
|
||||
}, default_subject="Fake problem check", attachments=None)
|
||||
send_email.assert_called_once_with(
|
||||
"wuttatest_problems_fake_check",
|
||||
{
|
||||
"system_title": "wuttatest",
|
||||
"config": self.config,
|
||||
"app": self.app,
|
||||
"check": check,
|
||||
"problems": problems,
|
||||
},
|
||||
default_subject="Fake problem check",
|
||||
attachments=None,
|
||||
)
|
||||
|
||||
def test_run_problem_check(self):
|
||||
with patch.object(FakeProblemCheck, 'find_problems') as find_problems:
|
||||
with patch.object(self.handler, 'send_problem_report') as send_problem_report:
|
||||
with patch.object(FakeProblemCheck, "find_problems") as find_problems:
|
||||
with patch.object(
|
||||
self.handler, "send_problem_report"
|
||||
) as send_problem_report:
|
||||
|
||||
# check runs by default
|
||||
find_problems.return_value = [{'foo': 'bar'}]
|
||||
find_problems.return_value = [{"foo": "bar"}]
|
||||
problems = self.handler.run_problem_check(FakeProblemCheck)
|
||||
self.assertEqual(problems, [{'foo': 'bar'}])
|
||||
self.assertEqual(problems, [{"foo": "bar"}])
|
||||
find_problems.assert_called_once_with()
|
||||
send_problem_report.assert_called_once()
|
||||
|
||||
# does not run if generally disabled
|
||||
find_problems.reset_mock()
|
||||
send_problem_report.reset_mock()
|
||||
with patch.object(self.handler, 'is_enabled', return_value=False):
|
||||
with patch.object(self.handler, "is_enabled", return_value=False):
|
||||
problems = self.handler.run_problem_check(FakeProblemCheck)
|
||||
self.assertIsNone(problems)
|
||||
find_problems.assert_not_called()
|
||||
send_problem_report.assert_not_called()
|
||||
|
||||
# unless caller gives force flag
|
||||
problems = self.handler.run_problem_check(FakeProblemCheck, force=True)
|
||||
self.assertEqual(problems, [{'foo': 'bar'}])
|
||||
problems = self.handler.run_problem_check(
|
||||
FakeProblemCheck, force=True
|
||||
)
|
||||
self.assertEqual(problems, [{"foo": "bar"}])
|
||||
find_problems.assert_called_once_with()
|
||||
send_problem_report.assert_called_once()
|
||||
|
||||
|
|
@ -220,7 +233,9 @@ class TestProblemHandler(ConfigTestCase):
|
|||
find_problems.reset_mock()
|
||||
send_problem_report.reset_mock()
|
||||
weekday = datetime.date.today().weekday()
|
||||
self.config.setdefault(f'wutta.problems.wuttatest.fake_check.day{weekday}', 'false')
|
||||
self.config.setdefault(
|
||||
f"wutta.problems.wuttatest.fake_check.day{weekday}", "false"
|
||||
)
|
||||
problems = self.handler.run_problem_check(FakeProblemCheck)
|
||||
self.assertIsNone(problems)
|
||||
find_problems.assert_not_called()
|
||||
|
|
@ -228,16 +243,18 @@ class TestProblemHandler(ConfigTestCase):
|
|||
|
||||
# unless caller gives force flag
|
||||
problems = self.handler.run_problem_check(FakeProblemCheck, force=True)
|
||||
self.assertEqual(problems, [{'foo': 'bar'}])
|
||||
self.assertEqual(problems, [{"foo": "bar"}])
|
||||
find_problems.assert_called_once_with()
|
||||
send_problem_report.assert_called_once()
|
||||
|
||||
def test_run_problem_checks(self):
|
||||
with patch.object(FakeProblemCheck, 'find_problems') as find_problems:
|
||||
with patch.object(self.handler, 'send_problem_report') as send_problem_report:
|
||||
with patch.object(FakeProblemCheck, "find_problems") as find_problems:
|
||||
with patch.object(
|
||||
self.handler, "send_problem_report"
|
||||
) as send_problem_report:
|
||||
|
||||
# check runs by default
|
||||
find_problems.return_value = [{'foo': 'bar'}]
|
||||
find_problems.return_value = [{"foo": "bar"}]
|
||||
self.handler.run_problem_checks([FakeProblemCheck])
|
||||
find_problems.assert_called_once_with()
|
||||
send_problem_report.assert_called_once()
|
||||
|
|
@ -245,7 +262,7 @@ class TestProblemHandler(ConfigTestCase):
|
|||
# does not run if generally disabled
|
||||
find_problems.reset_mock()
|
||||
send_problem_report.reset_mock()
|
||||
with patch.object(self.handler, 'is_enabled', return_value=False):
|
||||
with patch.object(self.handler, "is_enabled", return_value=False):
|
||||
self.handler.run_problem_checks([FakeProblemCheck])
|
||||
find_problems.assert_not_called()
|
||||
send_problem_report.assert_not_called()
|
||||
|
|
@ -259,7 +276,9 @@ class TestProblemHandler(ConfigTestCase):
|
|||
find_problems.reset_mock()
|
||||
send_problem_report.reset_mock()
|
||||
weekday = datetime.date.today().weekday()
|
||||
self.config.setdefault(f'wutta.problems.wuttatest.fake_check.day{weekday}', 'false')
|
||||
self.config.setdefault(
|
||||
f"wutta.problems.wuttatest.fake_check.day{weekday}", "false"
|
||||
)
|
||||
self.handler.run_problem_checks([FakeProblemCheck])
|
||||
find_problems.assert_not_called()
|
||||
send_problem_report.assert_not_called()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue