reactive/Rx.NET/Source/tests/Tests.System.Reactive/Tests/Linq/Observable/FinallyTest.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT License.
// See the LICENSE file in the project root for more information. 

using System;
using System.Reactive;
using System.Reactive.Linq;
using Microsoft.Reactive.Testing;
using Microsoft.VisualStudio.TestTools.UnitTesting;

using Assert = Xunit.Assert;

namespace ReactiveTests.Tests
{
    [TestClass]
    public class FinallyTest : ReactiveTest
    {

        [TestMethod]
        public void Finally_ArgumentChecking()
        {
            var someObservable = Observable.Empty<int>();

            ReactiveAssert.Throws<ArgumentNullException>(() => Observable.Finally<int>(null, () => { }));
            ReactiveAssert.Throws<ArgumentNullException>(() => Observable.Finally(someObservable, null));
        }

        [TestMethod]
        public void Finally_Never()
        {
            var scheduler = new TestScheduler();

            var invoked = false;
            var res = scheduler.Start(() =>
                Observable.Never<int>().Finally(() => { invoked = true; })
            );

            res.Messages.AssertEqual(
            );

            Assert.True(invoked); // due to unsubscribe; see 1356
        }

        [TestMethod]
        public void Finally_OnlyCalledOnce_Never()
        {
            var invokeCount = 0;
            var someObservable = Observable.Never<int>().Finally(() => { invokeCount++; });
            var d = someObservable.Subscribe();
            d.Dispose();
            d.Dispose();

            Assert.Equal(1, invokeCount);
        }

        [TestMethod]
        public void Finally_OnlyCalledOnce_Empty()
        {
            var invokeCount = 0;
            var someObservable = Observable.Empty<int>().Finally(() => { invokeCount++; });
            var d = someObservable.Subscribe();
            d.Dispose();
            d.Dispose();

            Assert.Equal(1, invokeCount);
        }

        [TestMethod]
        public void Finally_Empty()
        {
            var scheduler = new TestScheduler();

            var xs = scheduler.CreateHotObservable(
                OnNext(150, 1),
                OnCompleted<int>(250)
            );

            var invoked = false;
            var res = scheduler.Start(() =>
                xs.Finally(() => { invoked = true; })
            );

            Assert.True(invoked);

            res.Messages.AssertEqual(
                OnCompleted<int>(250)
            );

            xs.Subscriptions.AssertEqual(
                Subscribe(200, 250)
            );
        }

        [TestMethod]
        public void Finally_Return()
        {
            var scheduler = new TestScheduler();

            var xs = scheduler.CreateHotObservable(
                OnNext(150, 1),
                OnNext(210, 2),
                OnCompleted<int>(250)
            );

            var invoked = false;
            var res = scheduler.Start(() =>
                xs.Finally(() => { invoked = true; })
            );

            Assert.True(invoked);

            res.Messages.AssertEqual(
                OnNext(210, 2),
                OnCompleted<int>(250)
            );

            xs.Subscriptions.AssertEqual(
                Subscribe(200, 250)
            );
        }

        [TestMethod]
        public void Finally_Throw()
        {
            var scheduler = new TestScheduler();

            var ex = new Exception();
            var xs = scheduler.CreateHotObservable(
                OnNext(150, 1),
                OnError<int>(250, ex)
            );

            var invoked = false;
            var res = scheduler.Start(() =>
                xs.Finally(() => { invoked = true; })
            );

            Assert.True(invoked);

            res.Messages.AssertEqual(
                OnError<int>(250, ex)
            );

            xs.Subscriptions.AssertEqual(
                Subscribe(200, 250)
            );
        }

        [TestMethod]
        public void Finally_DisposeOrder_Empty()
        {
            var order = "";
            Observable
                .Empty<Unit>()
                .Finally(() => order += "1")
                .Finally(() => order += "2")
                .Finally(() => order += "3")
                .Subscribe();

            Assert.Equal("123", order);
        }

        [TestMethod]
        public void Finally_DisposeOrder_Return()
        {
            var order = "";
            Observable
                .Return(Unit.Default)
                .Finally(() => order += "1")
                .Finally(() => order += "2")
                .Finally(() => order += "3")
                .Subscribe();

            Assert.Equal("123", order);
        }

        [TestMethod]
        public void Finally_DisposeOrder_Never()
        {
            var order = "";
            var d = Observable
                .Never<Unit>()
                .Finally(() => order += "1")
                .Finally(() => order += "2")
                .Finally(() => order += "3")
                .Subscribe();

            d.Dispose();

            Assert.Equal("123", order);
        }
    }
}