SSH.NET/test/Renci.SshNet.Tests/Classes/BaseClientTest_Disconnected_Connect.cs

using Microsoft.VisualStudio.TestTools.UnitTesting;
using Moq;
using Renci.SshNet.Connection;

namespace Renci.SshNet.Tests.Classes
{
    [TestClass]
    public class BaseClientTest_Disconnected_Connect : BaseClientTestBase
    {
        private Mock<ISocketFactory> _socketFactory2Mock;
        private Mock<ISession> _session2Mock;
        private BaseClient _client;
        private ConnectionInfo _connectionInfo;

        protected override void SetupData()
        {
            _connectionInfo = new ConnectionInfo("host", "user", new PasswordAuthenticationMethod("user", "pwd"));
        }

        protected override void CreateMocks()
        {
            base.CreateMocks();

            _socketFactory2Mock = new Mock<ISocketFactory>(MockBehavior.Strict);
            _session2Mock = new Mock<ISession>(MockBehavior.Strict);
        }

        protected override void SetupMocks()
        {
            var sequence = new MockSequence();

            ServiceFactoryMock.InSequence(sequence)
                               .Setup(p => p.CreateSocketFactory())
                               .Returns(SocketFactoryMock.Object);
            ServiceFactoryMock.InSequence(sequence)
                               .Setup(p => p.CreateSession(_connectionInfo, SocketFactoryMock.Object))
                               .Returns(SessionMock.Object);
            SessionMock.InSequence(sequence)
                        .Setup(p => p.Connect());
            SessionMock.InSequence(sequence)
                        .Setup(p => p.OnDisconnecting());
            SessionMock.InSequence(sequence)
                        .Setup(p => p.Dispose());
            ServiceFactoryMock.InSequence(sequence)
                               .Setup(p => p.CreateSocketFactory())
                               .Returns(_socketFactory2Mock.Object);
            ServiceFactoryMock.InSequence(sequence)
                               .Setup(p => p.CreateSession(_connectionInfo, _socketFactory2Mock.Object))
                               .Returns(_session2Mock.Object);
            _session2Mock.InSequence(sequence)
                        .Setup(p => p.Connect());
        }

        protected override void Arrange()
        {
            base.Arrange();

            _client = new MyClient(_connectionInfo, false, ServiceFactoryMock.Object);
            _client.Connect();
            _client.Disconnect();
        }

        protected override void TearDown()
        {
            if (_client != null)
            {
                _session2Mock.Setup(p => p.OnDisconnecting());
                _session2Mock.Setup(p => p.Dispose());
                _client.Dispose();
            }
        }

        protected override void Act()
        {
            _client.Connect();
        }

        [TestMethod]
        public void CreateSocketFactoryOnServiceFactoryShouldBeInvokedTwic()
        {
            ServiceFactoryMock.Verify(p => p.CreateSocketFactory(), Times.Exactly(2));
        }

        [TestMethod]
        public void CreateSessionOnServiceFactoryShouldBeInvokedTwice()
        {
            ServiceFactoryMock.Verify(p => p.CreateSession(_connectionInfo, SocketFactoryMock.Object),
                                       Times.Once);
            ServiceFactoryMock.Verify(p => p.CreateSession(_connectionInfo, _socketFactory2Mock.Object),
                                       Times.Once);
        }

        [TestMethod]
        public void ConnectOnSessionShouldBeInvokedTwice()
        {
            SessionMock.Verify(p => p.Connect(), Times.Once);
            _session2Mock.Verify(p => p.Connect(), Times.Once);
        }

        private class MyClient : BaseClient
        {
            public MyClient(ConnectionInfo connectionInfo, bool ownsConnectionInfo, IServiceFactory serviceFactory) : base(connectionInfo, ownsConnectionInfo, serviceFactory)
            {
            }
        }
    }
}