def testRaisesWhenInitializedFromInvalidValues(self):
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("bl%ah")
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("H:")
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("H:1234/foo")
def testConvertsToHuntURN(self):
hunt_id = hunt_plugin.ApiHuntId("H:1234")
hunt_urn = hunt_id.ToURN()
self.assertEqual(hunt_urn.Basename(), hunt_id)
self.assertEqual(hunt_urn, "aff4:/hunts/H:1234")
def testRaisesWhenToURNCalledOnUninitializedValue(self):
hunt_id = hunt_plugin.ApiHuntId()
with self.assertRaises(ValueError):
hunt_id.ToURN()
def GenerateSample(self, number=0):
return hunt_plugin.ApiHuntId("H:%d" % number)
def testConvertsToString(self):
hunt_id = hunt_plugin.ApiHuntId("1234")
hunt_id_str = hunt_id.ToString()
self.assertEqual(hunt_id_str, "1234")
