def testRaisesWhenInitializedFromInvalidValues(self):
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("blah")
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("H:")
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("1234")
with self.assertRaises(ValueError):
hunt_plugin.ApiHuntId("H:1234/foo")
def testConvertsToHunttURN(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)