|  | @@ -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));
 | 
	
		
			
				|  |  | +        }
 | 
	
		
			
				|  |  | +    }
 | 
	
		
			
				|  |  | +}
 |