def setUp(self):
super(ApiGetExportedHuntResultsHandlerTest, self).setUp()
self.handler = hunt_plugin.ApiGetExportedHuntResultsHandler()
self.hunt_id = self.StartHunt(
flow_runner_args=rdf_flow_runner.FlowRunnerArgs(
flow_name=flow_test_lib.DummyFlowWithSingleReply.__name__),
client_rate=0)
self.client_ids = self.SetupClients(5)
# Ensure that clients are processed sequentially - this way the test won't
# depend on the order of results in the collection (which is normally
# random).
for cid in self.client_ids:
self.RunHunt(client_ids=[cid], failrate=-1)
def setUp(self):
super(ApiGetExportedHuntResultsHandlerTest, self).setUp()
self.handler = hunt_plugin.ApiGetExportedHuntResultsHandler()
self.hunt = implementation.StartHunt(
hunt_name=standard.GenericHunt.__name__,
flow_runner_args=rdf_flow_runner.FlowRunnerArgs(
flow_name=flow_test_lib.DummyFlowWithSingleReply.__name__),
client_rate=0,
token=self.token)
self.hunt.Run()
self.client_ids = self.SetupClients(5)
# Ensure that clients are processed sequentially - this way the test won't
# depend on the order of results in the collection (which is normally
# random).
for cid in self.client_ids:
self.AssignTasksToClients(client_ids=[cid])
client_mock = hunt_test_lib.SampleHuntMock()
hunt_test_lib.TestHuntHelper(client_mock, [cid], token=self.token)
def GetExportedHuntResults(self, args, context=None):
return api_hunt.ApiGetExportedHuntResultsHandler()
class TestHuntResultsView(gui_test_lib.GRRSeleniumHuntTest):
def testHuntResultsView(self):
self.CreateGenericHuntWithCollection()
self.Open("/")
self.WaitUntil(self.IsElementPresent, "client_query")
self.Click("css=a[grrtarget=hunts]")
self.WaitUntil(self.IsTextPresent, "GenericHunt")
self.Click("css=td:contains('GenericHunt')")
# Click the Results tab.
self.Click("css=li[heading=Results]")
self.WaitUntil(self.IsTextPresent, "aff4:/sample/1")
self.WaitUntil(
self.IsTextPresent,
"aff4:/%s/fs/os/c/bin/bash" % self.client_ids[0].Basename())
self.WaitUntil(self.IsTextPresent, "aff4:/sample/3")
self.RequestAndGrantClientApproval(self.client_ids[0].Basename())
self.Click("link=aff4:/%s/fs/os/c/bin/bash" %
self.client_ids[0].Basename())
self.WaitUntil(
self.IsElementPresent,
"css=li.active a:contains('Browse Virtual Filesystem')")
def testClientSummaryModalIsShownWhenClientInfoButtonClicked(self):
hunt_urn, client_id = self.CreateGenericHuntWithCollection(
[rdfvalue.RDFString("foo-result")])
self.Open("/#/hunts/%s/results" % hunt_urn.Basename())
self.Click("css=td:contains('%s') button:has(.glyphicon-info-sign)" %
self.client_ids[0].Basename())
self.WaitUntil(
self.IsElementPresent,
"css=.modal-dialog:contains('Client %s')" % client_id.Basename())
def testResultsViewGetsAutoRefreshed(self):
hunt_urn, client_id = self.CreateGenericHuntWithCollection(
[rdfvalue.RDFString("foo-result")])
self.Open("/")
# Ensure auto-refresh updates happen every second.
self.GetJavaScriptValue(
"grrUi.core.resultsCollectionDirective.setAutoRefreshInterval(1000);"
)
self.Click("css=a[grrtarget=hunts]")
self.Click("css=td:contains('GenericHunt')")
self.Click("css=li[heading=Results]")
self.WaitUntil(self.IsElementPresent,
"css=grr-results-collection td:contains('foo-result')")
self.WaitUntilNot(
self.IsElementPresent,
"css=grr-results-collection td:contains('bar-result')")
self.AddResultsToHunt(hunt_urn, client_id,
[rdfvalue.RDFString("bar-result")])
self.WaitUntil(self.IsElementPresent,
"css=grr-results-collection td:contains('bar-result')")
def testDownloadAsPanelNotShownForEmptyHuntResults(self):
hunt_urn, _ = self.CreateGenericHuntWithCollection([])
self.Open("/#/hunts/%s/results" % hunt_urn.Basename())
self.WaitUntil(self.IsTextPresent, "Value")
self.WaitUntilNot(self.IsElementPresent,
"css=grr-download-collection-as")
@mock.patch.object(
api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks,
"GetExportedHuntResults",
return_value=api_hunt.ApiGetExportedHuntResultsHandler())
def testHuntResultsCanBeDownloadedAsCsv(self, mock_method):
self.checkHuntResultsCanBeDownloadedAsType(
mock_method, csv_plugin.CSVInstantOutputPlugin.plugin_name,
csv_plugin.CSVInstantOutputPlugin.friendly_name)
@mock.patch.object(
api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks,
"GetExportedHuntResults",
return_value=api_hunt.ApiGetExportedHuntResultsHandler())
def testHuntResultsCanBeDownloadedAsYaml(self, mock_method):
self.checkHuntResultsCanBeDownloadedAsType(
mock_method, yaml_plugin.
YamlInstantOutputPluginWithExportConversion.plugin_name,
yaml_plugin.YamlInstantOutputPluginWithExportConversion.
friendly_name)
@mock.patch.object(
api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks,
"GetExportedHuntResults",
return_value=api_hunt.ApiGetExportedHuntResultsHandler())
def testHuntResultsCanBeDownloadedAsSqlite(self, mock_method):
self.checkHuntResultsCanBeDownloadedAsType(
mock_method, sqlite_plugin.SqliteInstantOutputPlugin.plugin_name,
sqlite_plugin.SqliteInstantOutputPlugin.friendly_name)
def checkHuntResultsCanBeDownloadedAsType(self, mock_method, plugin,
plugin_display_name):
hunt_urn, _ = self.CreateGenericHuntWithCollection()
self.Open("/#/hunts/%s/results" % hunt_urn.Basename())
self.Select("id=plugin-select", plugin_display_name)
self.Click("css=grr-download-collection-as button[name='download-as']")
def MockMethodIsCalled():
try:
# Mock should be called twice: once for HEAD (to check permissions)
# and once for GET methods.
mock_method.assert_called_with(
api_hunt.ApiGetExportedHuntResultsArgs(
hunt_id=hunt_urn.Basename(), plugin_name=plugin),
token=mock.ANY)
return True
except AssertionError:
return False
self.WaitUntil(MockMethodIsCalled)