@@ -45,9 +45,10 @@ public void Cleanup()
45
45
}
46
46
}
47
47
48
- protected void Arrange ( )
48
+ private void SetupData ( )
49
49
{
50
50
var random = new Random ( ) ;
51
+
51
52
_bufferSize = random . Next ( 5 , 15 ) ;
52
53
_fileSize = _bufferSize + 2 ; //force uploading 2 chunks
53
54
_fileContent = CreateContent ( _fileSize ) ;
@@ -56,12 +57,18 @@ protected void Arrange()
56
57
_fileInfo = new FileInfo ( _fileName ) ;
57
58
_path = random . Next ( ) . ToString ( CultureInfo . InvariantCulture ) ;
58
59
_uploadingRegister = new List < ScpUploadEventArgs > ( ) ;
60
+ }
59
61
62
+ private void CreateMocks ( )
63
+ {
60
64
_serviceFactoryMock = new Mock < IServiceFactory > ( MockBehavior . Strict ) ;
61
65
_sessionMock = new Mock < ISession > ( MockBehavior . Strict ) ;
62
66
_channelSessionMock = new Mock < IChannelSession > ( MockBehavior . Strict ) ;
63
67
_pipeStreamMock = new Mock < PipeStream > ( MockBehavior . Strict ) ;
68
+ }
64
69
70
+ private void SetupMocks ( )
71
+ {
65
72
var sequence = new MockSequence ( ) ;
66
73
_serviceFactoryMock . InSequence ( sequence )
67
74
. Setup ( p => p . CreateSession ( _connectionInfo ) )
@@ -73,22 +80,16 @@ protected void Arrange()
73
80
_channelSessionMock . InSequence ( sequence )
74
81
. Setup (
75
82
p => p . SendExecRequest ( string . Format ( "scp -t \" {0}\" " , _path ) ) ) . Returns ( true ) ;
76
- for ( var i = 0 ; i < random . Next ( 1 , 3 ) ; i ++ )
77
- _pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( - 1 ) ;
78
83
_pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( 0 ) ;
79
84
_channelSessionMock . InSequence ( sequence ) . Setup ( p => p . SendData ( It . IsAny < byte [ ] > ( ) ) ) ;
80
- for ( var i = 0 ; i < random . Next ( 1 , 3 ) ; i ++ )
81
- _pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( - 1 ) ;
82
85
_pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( 0 ) ;
83
86
_channelSessionMock . InSequence ( sequence )
84
87
. Setup ( p => p . SendData ( It . Is < byte [ ] > ( b => b . SequenceEqual ( CreateData (
85
- string . Format ( "C0644 {0} {1}\n " ,
86
- _fileInfo . Length ,
87
- Path . GetFileName ( _fileName )
88
- )
89
- ) ) ) ) ) ;
90
- for ( var i = 0 ; i < random . Next ( 1 , 3 ) ; i ++ )
91
- _pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( - 1 ) ;
88
+ string . Format ( "C0644 {0} {1}\n " ,
89
+ _fileInfo . Length ,
90
+ Path . GetFileName ( _fileName )
91
+ )
92
+ ) ) ) ) ) ;
92
93
_pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( 0 ) ;
93
94
_channelSessionMock . InSequence ( sequence )
94
95
. Setup (
@@ -98,13 +99,18 @@ protected void Arrange()
98
99
p => p . SendData ( It . Is < byte [ ] > ( b => b . Take ( 0 , _fileContent . Length - _bufferSize ) . SequenceEqual ( _fileContent . Take ( _bufferSize , _fileContent . Length - _bufferSize ) ) ) , 0 , _fileContent . Length - _bufferSize ) ) ;
99
100
_channelSessionMock . InSequence ( sequence )
100
101
. Setup (
101
- p => p . SendData ( It . Is < byte [ ] > ( b => b . SequenceEqual ( new byte [ ] { 0 } ) ) ) ) ;
102
- for ( var i = 0 ; i < random . Next ( 1 , 3 ) ; i ++ )
103
- _pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( - 1 ) ;
102
+ p => p . SendData ( It . Is < byte [ ] > ( b => b . SequenceEqual ( new byte [ ] { 0 } ) ) ) ) ;
104
103
_pipeStreamMock . InSequence ( sequence ) . Setup ( p => p . ReadByte ( ) ) . Returns ( 0 ) ;
105
104
_channelSessionMock . InSequence ( sequence ) . Setup ( p => p . Close ( ) ) ;
106
105
_channelSessionMock . InSequence ( sequence ) . Setup ( p => p . Dispose ( ) ) ;
107
106
_pipeStreamMock . As < IDisposable > ( ) . InSequence ( sequence ) . Setup ( p => p . Dispose ( ) ) ;
107
+ }
108
+
109
+ protected void Arrange ( )
110
+ {
111
+ SetupData ( ) ;
112
+ CreateMocks ( ) ;
113
+ SetupMocks ( ) ;
108
114
109
115
_scpClient = new ScpClient ( _connectionInfo , false , _serviceFactoryMock . Object )
110
116
{
@@ -161,12 +167,12 @@ public void UploadingShouldHaveFiredTwice()
161
167
Assert . AreEqual ( _fileSize , uploading . Uploaded ) ;
162
168
}
163
169
164
- private IEnumerable < byte > CreateData ( string command )
170
+ private static IEnumerable < byte > CreateData ( string command )
165
171
{
166
172
return Encoding . Default . GetBytes ( command ) ;
167
173
}
168
174
169
- private byte [ ] CreateContent ( int length )
175
+ private static byte [ ] CreateContent ( int length )
170
176
{
171
177
var random = new Random ( ) ;
172
178
var content = new byte [ length ] ;
@@ -176,7 +182,7 @@ private byte[] CreateContent(int length)
176
182
return content ;
177
183
}
178
184
179
- private string CreateTemporaryFile ( byte [ ] content )
185
+ private static string CreateTemporaryFile ( byte [ ] content )
180
186
{
181
187
var tempFile = Path . GetTempFileName ( ) ;
182
188
using ( var fs = File . OpenWrite ( tempFile ) )
0 commit comments