def setUp(self): super(ApiGetHuntContextHandlerTest, self).setUp() self.handler = hunt_plugin.ApiGetHuntContextHandler() # Set up mock hunt and context. with self.CreateHunt(description="the hunt") as hunt: # Assign for easier access. self.hunt = hunt self.context = self.hunt.state.context # Add custom properties for the tests. self.context.Register("string_prop", "some test value") self.context.Register("time_prop", rdfvalue.RDFDatetime().Now()) self.context.Register("int_prop", 42) self.context.Register("valid_float_prop", 42.0) self.context.Register("invalid_float_prop", 42.5) self.context.Register("bool_prop", True) self.context.Register("none_prop", None) self.context.Register("non_proto_prop", object()) self.context.Register( "proto_prop", hunt_plugin.ApiGetHuntContextArgs(hunt_id=hunt.urn.Basename()))
def GetHuntContext(self, args, token=None): return api_hunt.ApiGetHuntContextHandler()