def AASetup(self):
self.sev_db = severity.Severity('severity.db', 'unknown')
def test_no_arg_to_severity(self):
with self.assertRaises(AppArmorException):
severity.Severity()
def test_default_severity(self):
sev_db = severity.Severity('severity.db', 'unknown')
obj = BaseRule()
rank = obj.severity(sev_db)
self.assertEqual(rank, sev_db.NOT_IMPLEMENTED)
def _test_db(self, contents):
self.db_file = self.writeTmpfile('severity.db', contents)
self.sev_db = severity.Severity(self.db_file)
return self.sev_db
def _run_test(self, params, expected):
sev_db = severity.Severity('severity.db', 'unknown')
obj = CapabilityRule(params)
rank = obj.severity(sev_db)
self.assertEqual(rank, expected)