public async Task TestTearDownStreamBroken() { var source = new MultiStreamProvider <int>(_provider, 2); var query = await source.Select(x => x, GrainClient.GrainFactory); var queryOutputStreams = await query.GetStreamIdentities(); var resultConsumer = new TestTransactionalStreamConsumerAggregate <int>(_provider); await resultConsumer.SetInput(queryOutputStreams); Assert.AreEqual(2, queryOutputStreams.Count); Assert.AreEqual(0, resultConsumer.Items.Count); Assert.IsFalse(await resultConsumer.AllConsumersTearDownCalled()); await query.TearDown(); await resultConsumer.TearDown(); Assert.IsTrue(await resultConsumer.AllConsumersTearDownCalled()); await source.SendItems(new List <int>() { 2, 3 }); Assert.AreEqual(0, resultConsumer.Items.Count); Assert.IsTrue(await resultConsumer.AllConsumersTearDownCalled()); }
public async Task TestItemAggregation() { var aggregate = GrainClient.GrainFactory.GetGrain <IStreamProcessorSelectAggregate <int, int> >(Guid.NewGuid()); await aggregate.SetFunction(_ => _); var itemsToSend = new List <int> { 1, 5, 32, -12 }; var provider = GrainClient.GetStreamProvider(StreamProvider); var inputAggregate = new MultiStreamProvider <int>(provider, 2); await aggregate.SetInput(await inputAggregate.GetStreamIdentities()); Assert.AreEqual(2, (await aggregate.GetStreamIdentities()).Count); var consumerAggregate = new TestTransactionalStreamConsumerAggregate <int>(provider); await consumerAggregate.SetInput(await aggregate.GetStreamIdentities()); var tid = await inputAggregate.SendItems(itemsToSend); var waitForTransaction = consumerAggregate.TransactionComplete(tid); var resultItems = consumerAggregate.Items; Assert.AreEqual(4, resultItems.Count); CollectionAssert.AreEquivalent(itemsToSend, resultItems); Assert.IsFalse(await consumerAggregate.AllConsumersTearDownCalled()); await inputAggregate.TearDown(); Assert.IsTrue(await consumerAggregate.AllConsumersTearDownCalled()); }