def setUp(self): super(ApplyPluginToMultiTypeCollectionTest, self).setUp() self.plugin = test_plugins.TestInstantOutputPlugin( source_urn=rdfvalue.RDFURN("aff4:/foo/bar"), token=self.token) self.client_id = self.SetupClients(1)[0] self.pool = data_store.DB.GetMutationPool() self.collection = multi_type_collection.MultiTypeCollection( rdfvalue.RDFURN("aff4:/mt_collection/testAddScan"))
def TypedResultCollectionForFID(cls, flow_id, token=None): return multi_type_collection.MultiTypeCollection( flow_id.Add(RESULTS_PER_TYPE_SUFFIX), token=token)
def setUp(self): super(MultiTypeCollectionTest, self).setUp() self.pool = data_store.DB.GetMutationPool(token=self.token) self.collection = multi_type_collection.MultiTypeCollection( rdfvalue.RDFURN("aff4:/mt_collection/testAddScan"), token=self.token)