public async void Close_calls_the_handlers_HandleQueryCloseState() { var handlerMock = new Mock <IHandler>(); handlerMock.Setup(m => m.HandleQueryCloseState("Id", "ChannelId", "TxId")) .ReturnsAsync(new QueryResponse() { HasMore = false, Id = "Id" }); var response = new QueryResponse() { Id = "Id" }; var iterator = new StateQueryIterator(handlerMock.Object, "ChannelId", "TxId", response); var result = await iterator.Close(); handlerMock.VerifyAll(); result.Id.Should().Be("Id"); }