using System; using System.Collections.Generic; using System.Globalization; using Microsoft.Extensions.Logging.Abstractions; using Microsoft.VisualStudio.TestTools.UnitTesting; using Moq; using Renci.SshNet.Channels; using Renci.SshNet.Common; namespace Renci.SshNet.Tests.Classes { [TestClass] public class SubsystemSession_OnChannelDataReceived_Connected { private Mock _sessionMock; private Mock _channelMock; private string _subsystemName; private SubsystemSessionStub _subsystemSession; private int _operationTimeout; private IList _disconnectedRegister; private IList _errorOccurredRegister; private ChannelDataEventArgs _channelDataEventArgs; private MockSequence _sequence; [TestInitialize] public void Setup() { Arrange(); Act(); } protected void Arrange() { var random = new Random(); _subsystemName = random.Next().ToString(CultureInfo.InvariantCulture); _operationTimeout = 30000; _disconnectedRegister = new List(); _errorOccurredRegister = new List(); _channelDataEventArgs = new ChannelDataEventArgs( (uint)random.Next(0, int.MaxValue), new[] { (byte)random.Next(byte.MinValue, byte.MaxValue) }); _sessionMock = new Mock(MockBehavior.Strict); _sessionMock.Setup(p => p.SessionLoggerFactory).Returns(NullLoggerFactory.Instance); _channelMock = new Mock(MockBehavior.Strict); _sequence = new MockSequence(); _sessionMock.InSequence(_sequence).Setup(p => p.CreateChannelSession()).Returns(_channelMock.Object); _channelMock.InSequence(_sequence).Setup(p => p.Open()); _channelMock.InSequence(_sequence).Setup(p => p.SendSubsystemRequest(_subsystemName)).Returns(true); _subsystemSession = new SubsystemSessionStub( _sessionMock.Object, _subsystemName, _operationTimeout); _subsystemSession.Disconnected += (sender, args) => _disconnectedRegister.Add(args); _subsystemSession.ErrorOccurred += (sender, args) => _errorOccurredRegister.Add(args); _subsystemSession.Connect(); } protected void Act() { _channelMock.Raise(s => s.DataReceived += null, _channelDataEventArgs); } [TestMethod] public void DisconnectHasNeverFired() { Assert.AreEqual(0, _disconnectedRegister.Count); } [TestMethod] public void ErrorOccurredHasNeverFired() { Assert.AreEqual(0, _errorOccurredRegister.Count); } [TestMethod] public void OnDataReceivedShouldBeInvokedOnce() { Assert.AreEqual(1, _subsystemSession.OnDataReceivedInvocations.Count); var received = _subsystemSession.OnDataReceivedInvocations[0]; Assert.AreEqual(_channelDataEventArgs.Data, received.Data); } [TestMethod] public void IsOpenShouldReturnTrueWhenChannelIsOpen() { _channelMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(true); Assert.IsTrue(_subsystemSession.IsOpen); _channelMock.Verify(p => p.IsOpen, Times.Once); } [TestMethod] public void IsOpenShouldReturnFalseWhenSessionIsNotConnected() { _channelMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(false); Assert.IsFalse(_subsystemSession.IsOpen); _channelMock.Verify(p => p.IsOpen, Times.Once); } [TestMethod] public void DisposeOnChannelShouldNeverBeInvoked() { _channelMock.Verify(p => p.Dispose(), Times.Never); } } }