PolicyCheckerTest: Fix mock check names.

Signed-off-by: Chris PeBenito <pebenito@ieee.org>
This commit is contained in:
Chris PeBenito 2020-07-05 13:31:59 -04:00
parent d07cb1dbee
commit 58475daba2

View File

@ -65,31 +65,30 @@ class PolicyCheckerTest(unittest.TestCase):
def test_run_pass(self): def test_run_pass(self):
"""Test run with passing config.""" """Test run with passing config."""
with open("/dev/null", "w") as fd: with open(os.devnull, "w") as fd:
checker = PolicyChecker(self.p, "tests/checker/checker-valid.ini") checker = PolicyChecker(self.p, "tests/checker/checker-valid.ini")
# create additional disabled mock test # create additional disabled mock test
newcheck = Mock() newcheck = Mock()
newcheck.testname = "disabled" newcheck.checkname = "disabled"
newcheck.disable = True newcheck.disable = True
newcheck.validate_config.return_value = None newcheck.validate_config.return_value = None
newcheck.run.return_value = [] newcheck.run.return_value = []
checker.checks.append(newcheck) checker.checks.append(newcheck)
self.assertEqual(4, len(checker.checks)) self.assertEqual(4, len(checker.checks))
result = checker.run(output=fd) result = checker.run(output=fd)
self.assertEqual(0, result) self.assertEqual(0, result)
newcheck.run.assert_not_called() newcheck.run.assert_not_called()
def test_run_fail(self): def test_run_fail(self):
"""Test run with failing config.""" """Test run with failing config."""
with open("/dev/null", "w") as fd: with open(os.devnull, "w") as fd:
checker = PolicyChecker(self.p, "tests/checker/checker-valid.ini") checker = PolicyChecker(self.p, "tests/checker/checker-valid.ini")
# create additional failing mock test # create additional failing mock test
newcheck = Mock() newcheck = Mock()
newcheck.testname = "failing test" newcheck.checkname = "failing test"
newcheck.disable = False newcheck.disable = False
newcheck.validate_config.return_value = None newcheck.validate_config.return_value = None
newcheck.run.return_value = list(range(13)) newcheck.run.return_value = list(range(13))