def test_persist_data_stream(self):
ingestval = DotDict()
ingestval.queues = None
ingestval.exchange_point = "test"
queueval = DotDict()
queueval.name = "test"
queueval.datastore_name = "test"
testval = "dataset_id"
self.ingestion_management.read_ingestion_configuration = Mock()
self.ingestion_management.read_ingestion_configuration.return_value = ingestval
self.ingestion_management._determine_queue = Mock()
self.ingestion_management._determine_queue.return_value = queueval
self.pubsub_read.return_value = DotDict({"persisted": False})
self.ingestion_management._existing_dataset = Mock()
retval = self.ingestion_management.persist_data_stream("stream_id", "config_id", "dataset_id")
self.assertTrue(self.pubsub_act_sub.call_count)
self.assertTrue(self.pubsub_create_sub.call_count)
self.assertTrue(self.rr_create_assoc.call_count)
self.assertTrue(retval == testval)
