Skip to content

Commit d3635d1

Browse files
committed
Added test for recently introduced regression that is fixed by PR #186.
1 parent 08b2d05 commit d3635d1

5 files changed

+124
-1
lines changed

src/Renci.SshNet.Tests/Classes/Sftp/SftpFileStreamTestBase.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ public abstract class SftpFileStreamTestBase
1010
internal Mock<ISftpSession> SftpSessionMock;
1111
protected MockSequence MockSequence;
1212

13-
private void Arrange()
13+
protected virtual void Arrange()
1414
{
1515
SetupData();
1616
CreateMocks();

src/Renci.SshNet.Tests/Classes/Sftp/SftpFileStreamTest_ReadByte_ReadMode_NoDataInWriteBufferAndNoDataInReadBuffer_Eof.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ public void Setup()
2626
Act();
2727
}
2828

29+
[TestCleanup]
2930
public void TearDown()
3031
{
3132
_sftpSessionMock.InSequence(_sequence)

src/Renci.SshNet.Tests/Classes/Sftp/SftpFileStreamTest_ReadByte_ReadMode_NoDataInWriteBufferAndNoDataInReadBuffer_LessDataThanReadBufferSizeAvailable.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ public void Setup()
3232
Act();
3333
}
3434

35+
[TestCleanup]
3536
public void TearDown()
3637
{
3738
_sftpSessionMock.InSequence(_sequence)
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,120 @@
1+
using System;
2+
using System.IO;
3+
using Microsoft.VisualStudio.TestTools.UnitTesting;
4+
using Moq;
5+
using Renci.SshNet.Sftp;
6+
7+
namespace Renci.SshNet.Tests.Classes.Sftp
8+
{
9+
[TestClass]
10+
public class SftpFileStreamTest_Read_ReadMode_NoDataInReaderBufferAndReadMoreBytesThanCount : SftpFileStreamTestBase
11+
{
12+
private string _path;
13+
private SftpFileStream _target;
14+
private byte[] _handle;
15+
private uint _bufferSize;
16+
private uint _readBufferSize;
17+
private uint _writeBufferSize;
18+
private int _actual;
19+
private byte[] _buffer;
20+
private byte[] _serverData;
21+
private int _numberOfBytesInReadBuffer;
22+
private int _numberOfBytesToRead;
23+
24+
protected override void SetupData()
25+
{
26+
base.SetupData();
27+
28+
var random = new Random();
29+
_path = random.Next().ToString();
30+
_handle = GenerateRandom(5, random);
31+
_bufferSize = (uint) random.Next(1, 1000);
32+
_readBufferSize = 20;
33+
_writeBufferSize = 500;
34+
35+
_numberOfBytesToRead = 20;
36+
_buffer = new byte[_numberOfBytesToRead];
37+
_numberOfBytesInReadBuffer = 10;
38+
_serverData = GenerateRandom(_buffer.Length + _numberOfBytesInReadBuffer, random);
39+
}
40+
41+
protected override void SetupMocks()
42+
{
43+
SftpSessionMock.InSequence(MockSequence)
44+
.Setup(p => p.RequestOpen(_path, Flags.Read, false))
45+
.Returns(_handle);
46+
SftpSessionMock.InSequence(MockSequence)
47+
.Setup(p => p.CalculateOptimalReadLength(_bufferSize))
48+
.Returns(_readBufferSize);
49+
SftpSessionMock.InSequence(MockSequence)
50+
.Setup(p => p.CalculateOptimalWriteLength(_bufferSize, _handle))
51+
.Returns(_writeBufferSize);
52+
SftpSessionMock.InSequence(MockSequence)
53+
.Setup(p => p.IsOpen)
54+
.Returns(true);
55+
SftpSessionMock.InSequence(MockSequence)
56+
.Setup(p => p.RequestRead(_handle, 0UL, _readBufferSize))
57+
.Returns(_serverData);
58+
}
59+
60+
[TestCleanup]
61+
public void TearDown()
62+
{
63+
SftpSessionMock.InSequence(MockSequence)
64+
.Setup(p => p.RequestClose(_handle));
65+
}
66+
67+
protected override void Arrange()
68+
{
69+
base.Arrange();
70+
71+
_target = new SftpFileStream(SftpSessionMock.Object,
72+
_path,
73+
FileMode.Open,
74+
FileAccess.Read,
75+
(int) _bufferSize);
76+
}
77+
78+
protected override void Act()
79+
{
80+
_actual = _target.Read(_buffer, 0, _numberOfBytesToRead);
81+
}
82+
83+
[TestMethod]
84+
public void ReadShouldHaveReturnedTheNumberOfBytesWrittenToBuffer()
85+
{
86+
Assert.AreEqual(_buffer.Length, _actual);
87+
}
88+
89+
[TestMethod]
90+
public void ReadShouldHaveWrittenBytesToTheCallerSuppliedBuffer()
91+
{
92+
Assert.IsTrue(_serverData.Take(_buffer.Length).IsEqualTo(_buffer));
93+
}
94+
95+
[TestMethod]
96+
public void PositionShouldReturnNumberOfBytesWrittenToBuffer()
97+
{
98+
SftpSessionMock.InSequence(MockSequence).Setup(p => p.IsOpen).Returns(true);
99+
100+
Assert.AreEqual(_buffer.Length, _target.Position);
101+
102+
SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(2));
103+
}
104+
105+
[TestMethod]
106+
public void ReadShouldReturnAllRemaningBytesFromReadBufferWhenCountIsEqualToNumberOfRemainingBytes()
107+
{
108+
SftpSessionMock.InSequence(MockSequence).Setup(p => p.IsOpen).Returns(true);
109+
110+
_buffer = new byte[_numberOfBytesInReadBuffer];
111+
112+
var actual = _target.Read(_buffer, 0, _numberOfBytesInReadBuffer);
113+
114+
Assert.AreEqual(_numberOfBytesInReadBuffer, actual);
115+
Assert.IsTrue(_serverData.Take(_numberOfBytesToRead, _numberOfBytesInReadBuffer).IsEqualTo(_buffer));
116+
117+
SftpSessionMock.Verify(p => p.IsOpen, Times.Exactly(2));
118+
}
119+
}
120+
}

src/Renci.SshNet.Tests/Renci.SshNet.Tests.csproj

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,6 +439,7 @@
439439
<Compile Include="Classes\Sftp\SftpFileStreamTest_Dispose_Disposed.cs" />
440440
<Compile Include="Classes\Sftp\SftpFileStreamTest_Finalize_SessionOpen.cs" />
441441
<Compile Include="Classes\Sftp\SftpFileStreamTest_ReadByte_ReadMode_NoDataInWriteBufferAndNoDataInReadBuffer_LessDataThanReadBufferSizeAvailable.cs" />
442+
<Compile Include="Classes\Sftp\SftpFileStreamTest_Read_ReadMode_NoDataInReaderBufferAndReadMoreBytesThanCount.cs" />
442443
<Compile Include="Classes\Sftp\SftpFileStreamTest_SetLength_Closed.cs" />
443444
<Compile Include="Classes\Sftp\SftpFileStreamTest_SetLength_Disposed.cs" />
444445
<Compile Include="Classes\Sftp\SftpFileStreamTest_SetLength_SessionNotOpen.cs" />

0 commit comments

Comments
 (0)