|
|
@@ -0,0 +1,193 @@
|
|
|
+using System;
|
|
|
+using System.Globalization;
|
|
|
+using System.IO;
|
|
|
+using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
|
+using Moq;
|
|
|
+using Renci.SshNet.Sftp;
|
|
|
+using Renci.SshNet.Tests.Common;
|
|
|
+using System.Threading;
|
|
|
+using Renci.SshNet.Sftp.Responses;
|
|
|
+
|
|
|
+namespace Renci.SshNet.Tests.Classes.Sftp
|
|
|
+{
|
|
|
+ /// <summary>
|
|
|
+ /// - In write mode
|
|
|
+ /// - Bytes in (write) buffer
|
|
|
+ /// - New length greater than client position and greater than server position
|
|
|
+ /// </summary>
|
|
|
+ [TestClass]
|
|
|
+ public class SftpFileStreamTest_SetLength_DataInWriteBuffer_NewLengthGreaterThanPosition : SftpFileStreamTestBase
|
|
|
+ {
|
|
|
+ private string _path;
|
|
|
+ private SftpFileStream _sftpFileStream;
|
|
|
+ private byte[] _handle;
|
|
|
+ private uint _bufferSize;
|
|
|
+ private uint _readBufferSize;
|
|
|
+ private uint _writeBufferSize;
|
|
|
+ private MockSequence _sequence;
|
|
|
+ private long _length;
|
|
|
+
|
|
|
+ private SftpFileAttributes _fileAttributes;
|
|
|
+ private SftpFileAttributes _originalFileAttributes;
|
|
|
+ private SftpFileAttributes _newFileAttributes;
|
|
|
+ private byte[] _readBytes;
|
|
|
+ private byte[] _actualReadBytes;
|
|
|
+ private byte[] _writeBytes;
|
|
|
+ private byte[] _actualWrittenBytes;
|
|
|
+
|
|
|
+ protected override void SetupData()
|
|
|
+ {
|
|
|
+ var random = new Random();
|
|
|
+
|
|
|
+ _path = random.Next().ToString(CultureInfo.InvariantCulture);
|
|
|
+ _handle = GenerateRandom(random.Next(2, 6), random);
|
|
|
+ _bufferSize = (uint) random.Next(1, 1000);
|
|
|
+ _readBufferSize = (uint) random.Next(1, 1000);
|
|
|
+ _writeBufferSize = (uint) random.Next(100, 1000);
|
|
|
+ _readBytes = new byte[5];
|
|
|
+ _actualReadBytes = GenerateRandom(_readBytes.Length, random);
|
|
|
+ _writeBytes = new byte[] { 0x01, 0x02, 0x03, 0x04 };
|
|
|
+ _length = _readBytes.Length + _writeBytes.Length + 2;
|
|
|
+ _actualWrittenBytes = null;
|
|
|
+
|
|
|
+ _fileAttributes = new SftpFileAttributesBuilder().WithExtension("X", "ABC")
|
|
|
+ .WithExtension("V", "VValue")
|
|
|
+ .WithGroupId(random.Next())
|
|
|
+ .WithLastAccessTime(DateTime.Now.AddSeconds(random.Next()))
|
|
|
+ .WithLastWriteTime(DateTime.Now.AddSeconds(random.Next()))
|
|
|
+ .WithPermissions((uint)random.Next())
|
|
|
+ .WithSize(_length + 100)
|
|
|
+ .WithUserId(random.Next())
|
|
|
+ .Build();
|
|
|
+ _originalFileAttributes = _fileAttributes.Clone();
|
|
|
+ _newFileAttributes = null;
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void SetupMocks()
|
|
|
+ {
|
|
|
+ _sequence = new MockSequence();
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestOpen(_path, Flags.Read | Flags.Write, false))
|
|
|
+ .Returns(_handle);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.CalculateOptimalReadLength(_bufferSize))
|
|
|
+ .Returns(_readBufferSize);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.CalculateOptimalWriteLength(_bufferSize, _handle))
|
|
|
+ .Returns(_writeBufferSize);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.IsOpen)
|
|
|
+ .Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestRead(_handle, 0, _readBufferSize))
|
|
|
+ .Returns(_actualReadBytes);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.IsOpen)
|
|
|
+ .Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.IsOpen)
|
|
|
+ .Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestWrite(_handle, (uint) _readBytes.Length, It.IsAny<byte[]>(), 0, _writeBytes.Length, It.IsAny<AutoResetEvent>(), null))
|
|
|
+ .Callback<byte[], ulong, byte[], int, int, AutoResetEvent, Action<SftpStatusResponse>>((handle, serverOffset, data, offset, length, wait, writeCompleted)
|
|
|
+ =>
|
|
|
+ {
|
|
|
+ _actualWrittenBytes = data.Take(0, _writeBytes.Length);
|
|
|
+ wait.Set();
|
|
|
+ });
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestFStat(_handle, false))
|
|
|
+ .Returns(_fileAttributes);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestFSetStat(_handle, _fileAttributes))
|
|
|
+ .Callback<byte[], SftpFileAttributes>((bytes, attributes) => _newFileAttributes = attributes.Clone());
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void Arrange()
|
|
|
+ {
|
|
|
+ base.Arrange();
|
|
|
+
|
|
|
+ _sftpFileStream = new SftpFileStream(SftpSessionMock.Object, _path, FileMode.Open, FileAccess.ReadWrite, (int)_bufferSize);
|
|
|
+ _sftpFileStream.Read(_readBytes, 0, _readBytes.Length);
|
|
|
+ _sftpFileStream.Write(new byte[] { 0x01, 0x02, 0x03, 0x04 }, 0, 4);
|
|
|
+ }
|
|
|
+
|
|
|
+ protected override void Act()
|
|
|
+ {
|
|
|
+ _sftpFileStream.SetLength(_length);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void PositionShouldReturnSamePositionAsBeforeSetLength()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+
|
|
|
+ Assert.AreEqual(_readBytes.Length + _writeBytes.Length, _sftpFileStream.Position);
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(4));
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void RequestFSetStatOnSftpSessionShouldBeInvokedOnce()
|
|
|
+ {
|
|
|
+ SftpSessionMock.Verify(p => p.RequestFSetStat(_handle, _fileAttributes), Times.Once);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void SizeOfSftpFileAttributesShouldBeModifiedToNewLengthBeforePassedToRequestFSetStat()
|
|
|
+ {
|
|
|
+ DictionaryAssert.AreEqual(_originalFileAttributes.Extensions, _newFileAttributes.Extensions);
|
|
|
+ Assert.AreEqual(_originalFileAttributes.GroupId, _newFileAttributes.GroupId);
|
|
|
+ Assert.AreEqual(_originalFileAttributes.LastAccessTime, _newFileAttributes.LastAccessTime);
|
|
|
+ Assert.AreEqual(_originalFileAttributes.LastWriteTime, _newFileAttributes.LastWriteTime);
|
|
|
+ Assert.AreEqual(_originalFileAttributes.Permissions, _newFileAttributes.Permissions);
|
|
|
+ Assert.AreEqual(_originalFileAttributes.UserId, _newFileAttributes.UserId);
|
|
|
+
|
|
|
+ Assert.AreEqual(_length, _newFileAttributes.Size);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void WrittenBytesShouldByFlushedToServer()
|
|
|
+ {
|
|
|
+ Assert.IsNotNull(_actualWrittenBytes);
|
|
|
+ CollectionAssert.AreEqual(_writeBytes, _actualWrittenBytes);
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void ReadShouldReadStartFromSamePositionAsBeforeSetLength()
|
|
|
+ {
|
|
|
+ SftpSessionMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestRead(_handle, (uint) (_readBytes.Length + _writeBytes.Length), _readBufferSize))
|
|
|
+ .Returns(new byte[] { 0x0f });
|
|
|
+
|
|
|
+ var byteRead = _sftpFileStream.ReadByte();
|
|
|
+
|
|
|
+ Assert.AreEqual(0x0f, byteRead);
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.RequestRead(_handle, (uint) (_readBytes.Length + _writeBytes.Length), _readBufferSize), Times.Once);
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(4));
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void WriteShouldStartFromSamePositionAsBeforeSetLength()
|
|
|
+ {
|
|
|
+ var bytesToWrite = GenerateRandom(5);
|
|
|
+
|
|
|
+ SftpSessionMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence).Setup(p => p.IsOpen).Returns(true);
|
|
|
+ SftpSessionMock.InSequence(_sequence)
|
|
|
+ .Setup(p => p.RequestWrite(_handle, (uint)(_readBytes.Length + _writeBytes.Length), It.IsAny<byte[]>(), 0, bytesToWrite.Length, It.IsAny<AutoResetEvent>(), null))
|
|
|
+ .Callback<byte[], ulong, byte[], int, int, AutoResetEvent, Action<SftpStatusResponse>>((handle, serverOffset, data, offset, length, wait, writeCompleted) =>
|
|
|
+ {
|
|
|
+ wait.Set();
|
|
|
+ });
|
|
|
+
|
|
|
+ _sftpFileStream.Write(bytesToWrite, 0, bytesToWrite.Length);
|
|
|
+ _sftpFileStream.Flush();
|
|
|
+
|
|
|
+ SftpSessionMock.Verify(p => p.RequestWrite(_handle, (uint) (_readBytes.Length + _writeBytes.Length), It.IsAny<byte[]>(), 0, bytesToWrite.Length, It.IsAny<AutoResetEvent>(), null), Times.Once);
|
|
|
+ SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(5));
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|