def test_none_or_empty(self): ct = get_scanner_type_from_string(None) self.assertIsNone(ct) ct = get_scanner_type_from_string('') self.assertIsNone(ct)
def test_osp_scanner(self): ct = get_scanner_type_from_string('1') self.assertEqual(ct, ScannerType.OSP_SCANNER_TYPE) ct = get_scanner_type_from_string('osp') self.assertEqual(ct, ScannerType.OSP_SCANNER_TYPE)
def test_invalid(self): with self.assertRaises(InvalidArgument): get_scanner_type_from_string('foo')