|
|
@@ -0,0 +1,153 @@
|
|
|
+using System;
|
|
|
+using System.IO;
|
|
|
+using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
|
+using Moq;
|
|
|
+using Renci.SshNet.Common;
|
|
|
+using Renci.SshNet.Sftp;
|
|
|
+
|
|
|
+namespace Renci.SshNet.Tests.Classes.Sftp
|
|
|
+{
|
|
|
+ [TestClass]
|
|
|
+ public class SftpFileStreamTest_Read_ReadMode_NoDataInReaderBufferAndReadLessBytesFromServerThanCount : SftpFileStreamTestBase
|
|
|
+ {
|
|
|
+ private string _path;
|
|
|
+ private SftpFileStream _target;
|
|
|
+ private byte[] _handle;
|
|
|
+ private uint _bufferSize;
|
|
|
+ private uint _readBufferSize;
|
|
|
+ private uint _writeBufferSize;
|
|
|
+ private int _actual;
|
|
|
+ private byte[] _buffer;
|
|
|
+ private byte[] _serverData1;
|
|
|
+ private byte[] _serverData2;
|
|
|
+ private int _serverData1Length;
|
|
|
+ private int _serverData2Length;
|
|
|
+ private int _numberOfBytesToRead;
|
|
|
+
|
|
|
+ protected override void SetupData()
|
|
|
+ {
|
|
|
+ base.SetupData();
|
|
|
+
|
|
|
+ var random = new Random();
|
|
|
+ _path = random.Next().ToString();
|
|
|
+ _handle = GenerateRandom(5, random);
|
|
|
+ _bufferSize = (uint)random.Next(1, 1000);
|
|
|
+ _readBufferSize = 20;
|
|
|
+ _writeBufferSize = 500;
|
|
|
+
|
|
|
+ _numberOfBytesToRead = 20;
|
|
|
+ _buffer = new byte[_numberOfBytesToRead];
|
|
|
+ _serverData1Length = _numberOfBytesToRead - 5;
|
|
|
+ _serverData1 = GenerateRandom(_serverData1Length, random);
|
|
|
+ _serverData2Length = _numberOfBytesToRead - _serverData1Length + 3;
|
|
|
+ _serverData2 = GenerateRandom(_serverData2Length, random);
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void SetupMocks()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.RequestOpen(_path, Flags.Read, false))
|
|
|
+ .Returns(_handle);
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.CalculateOptimalReadLength(_bufferSize))
|
|
|
+ .Returns(_readBufferSize);
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.CalculateOptimalWriteLength(_bufferSize, _handle))
|
|
|
+ .Returns(_writeBufferSize);
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.IsOpen)
|
|
|
+ .Returns(true);
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.RequestRead(_handle, 0UL, _readBufferSize))
|
|
|
+ .Returns(_serverData1);
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.RequestRead(_handle, (ulong) _serverData1.Length, _readBufferSize))
|
|
|
+ .Returns(_serverData2);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestCleanup]
|
|
|
+ public void TearDown()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(MockSequence)
|
|
|
+ .Setup(p => p.RequestClose(_handle));
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void Arrange()
|
|
|
+ {
|
|
|
+ base.Arrange();
|
|
|
+
|
|
|
+ _target = new SftpFileStream(SftpSessionMock.Object,
|
|
|
+ _path,
|
|
|
+ FileMode.Open,
|
|
|
+ FileAccess.Read,
|
|
|
+ (int)_bufferSize);
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void Act()
|
|
|
+ {
|
|
|
+ _actual = _target.Read(_buffer, 0, _numberOfBytesToRead);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void ReadShouldHaveReturnedTheNumberOfBytesRequested()
|
|
|
+ {
|
|
|
+ Assert.AreEqual(_buffer.Length, _actual);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void ReadShouldHaveWrittenBytesToTheCallerSuppliedBuffer()
|
|
|
+ {
|
|
|
+ Assert.IsTrue(_serverData1.IsEqualTo(_buffer.Take(_serverData1Length)));
|
|
|
+
|
|
|
+ var bytesWrittenFromSecondRead = _numberOfBytesToRead - _serverData1Length;
|
|
|
+ Assert.IsTrue(_serverData2.Take(bytesWrittenFromSecondRead).IsEqualTo(_buffer.Take(_serverData1Length, bytesWrittenFromSecondRead)));
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void PositionShouldReturnNumberOfBytesWrittenToBuffer()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(MockSequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+
|
|
|
+ Assert.AreEqual(_buffer.Length, _target.Position);
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(2));
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void ReadShouldReturnAllRemaningBytesFromReadBufferWhenCountIsEqualToNumberOfRemainingBytes()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(MockSequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+
|
|
|
+ var numberOfBytesRemainingInReadBuffer = _serverData1Length + _serverData2Length - _numberOfBytesToRead;
|
|
|
+
|
|
|
+ _buffer = new byte[numberOfBytesRemainingInReadBuffer];
|
|
|
+
|
|
|
+ var actual = _target.Read(_buffer, 0, _buffer.Length);
|
|
|
+
|
|
|
+ Assert.AreEqual(_buffer.Length, actual);
|
|
|
+ Assert.IsTrue(_serverData2.Take(_numberOfBytesToRead - _serverData1Length, _buffer.Length).IsEqualTo(_buffer));
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(2));
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void ReadShouldReturnAllRemaningBytesFromReadBufferAndReadAgainWhenCountIsGreaterThanNumberOfRemainingBytesAndNewReadReturnsZeroBytes()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(MockSequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+ SftpSessionMock.InSequence(MockSequence).Setup(p => p.RequestRead(_handle, (ulong) (_serverData1Length + _serverData2Length), _readBufferSize)).Returns(Array<byte>.Empty);
|
|
|
+
|
|
|
+ var numberOfBytesRemainingInReadBuffer = _serverData1Length + _serverData2Length - _numberOfBytesToRead;
|
|
|
+
|
|
|
+ _buffer = new byte[numberOfBytesRemainingInReadBuffer + 1];
|
|
|
+
|
|
|
+ var actual = _target.Read(_buffer, 0, _buffer.Length);
|
|
|
+
|
|
|
+ Assert.AreEqual(numberOfBytesRemainingInReadBuffer, actual);
|
|
|
+ Assert.IsTrue(_serverData2.Take(_numberOfBytesToRead - _serverData1Length, numberOfBytesRemainingInReadBuffer).IsEqualTo(_buffer.Take(numberOfBytesRemainingInReadBuffer)));
|
|
|
+ Assert.AreEqual(0, _buffer[numberOfBytesRemainingInReadBuffer]);
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(2));
|
|
|
+ SftpSessionMock.Verify(p => p.RequestRead(_handle, (ulong) (_serverData1Length + _serverData2Length), _readBufferSize));
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|