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)