def setUp(self):
super(ApiModifyHuntHandlerTest, self).setUp()
self.handler = hunt_plugin.ApiModifyHuntHandler()
self.hunt_id = self.CreateHunt(description="the hunt")
self.args = hunt_plugin.ApiModifyHuntArgs(hunt_id=self.hunt_id)
def Run(self):
# Check client_limit update.
with test_lib.FakeTime(42):
hunt = self.CreateHunt(description="the hunt")
# Create replace dictionary.
replace = {hunt.urn.Basename(): "H:123456"}
with test_lib.FakeTime(43):
self.Check("ModifyHunt",
args=hunt_plugin.ApiModifyHuntArgs(
hunt_id=hunt.urn.Basename(), client_limit=142),
replace=replace)
self.Check("ModifyHunt",
args=hunt_plugin.ApiModifyHuntArgs(
hunt_id=hunt.urn.Basename(), state="STOPPED"),
replace=replace)
def Run(self):
# Check client_limit update.
with test_lib.FakeTime(42):
if data_store.RelationalDBEnabled():
hunt_id = self.CreateHunt(description="the hunt")
else:
hunt_obj = self.CreateHunt(description="the hunt")
hunt_id = hunt_obj.urn.Basename()
# Create replace dictionary.
replace = {hunt_id: "H:123456"}
with test_lib.FakeTime(43):
self.Check("ModifyHunt",
args=hunt_plugin.ApiModifyHuntArgs(hunt_id=hunt_id,
client_limit=142),
replace=replace)
self.Check("ModifyHunt",
args=hunt_plugin.ApiModifyHuntArgs(hunt_id=hunt_id,
state="STOPPED"),
replace=replace)
def setUp(self):
super(ApiModifyHuntHandlerTest, self).setUp()
self.handler = hunt_plugin.ApiModifyHuntHandler()
if data_store.RelationalDBEnabled():
self.hunt_id = self.CreateHunt(description="the hunt")
else:
self.hunt_obj = self.CreateHunt(description="the hunt")
self.hunt_urn = self.hunt_obj.urn
self.hunt_id = self.hunt_urn.Basename()
self.args = hunt_plugin.ApiModifyHuntArgs(hunt_id=self.hunt_id)
def testModifyHuntIsAccessChecked(self):
args = api_hunt.ApiModifyHuntArgs(hunt_id="H:123456")
self.CheckMethodIsAccessChecked(self.router.ModifyHunt,
"CheckHuntAccess",
args=args)
