def testGetReportByName(self): """Ensure GetReportByName instantiates correct subclasses based on name.""" with report_plugins_test_mocks.MockedReportPlugins(): report_object = report_plugins.GetReportByName("BarReportPlugin") self.assertTrue( isinstance(report_object, report_plugins_test_mocks.BarReportPlugin))
def testGetAvailableReportPlugins(self): """Ensure GetAvailableReportPlugins lists ReportPluginBase's subclasses.""" with report_plugins_test_mocks.MockedReportPlugins(): self.assertTrue(report_plugins_test_mocks.FooReportPlugin in report_plugins.GetAvailableReportPlugins()) self.assertTrue(report_plugins_test_mocks.BarReportPlugin in report_plugins.GetAvailableReportPlugins())
def Run(self): with report_plugins_test_mocks.MockedReportPlugins(): self.Check("GetReport", args=stats_plugin.ApiGetReportArgs( name="BarReportPlugin", start_time=rdfvalue.RDFDatetime.FromHumanReadable( "2012/12/14").AsMicrosecondsSinceEpoch(), duration="4d"))
def Run(self): with report_plugins_test_mocks.MockedReportPlugins(): self.Check("ListReports")