def test_none(self): """Test when security_flag is False.""" self.crash.security_flag = False result = data_handler._get_security_severity( # pylint: disable=protected-access self.crash, 'job', self.gestures) self.assertIsNone(result) self.assertEqual(0, self.mock.get_security_severity.call_count)
def test_get(self): """Test when security_flag is True.""" self.crash.security_flag = True result = data_handler._get_security_severity( # pylint: disable=protected-access self.crash, 'job', self.gestures) self.assertEqual('Low', result) self.mock.get_security_severity.assert_called_with( 'type', 'trace', 'job', False)