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)