brouzouf/ftp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Actually store file content in mock

Browse Source
This commit is contained in:
Florian Goetghebeur 2020-03-10 11:41:46 +01:00
parent e1b8fdd0dc
commit adfb8ed9f3
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
conn_test.go
View File

@ -1,9 +1,9 @@
package ftp package ftp
import ( import (
"bytes"
"errors" "errors"
"io" "io"
"io/ioutil"
"net" "net"
"net/textproto" "net/textproto"
"reflect" "reflect"
@ -19,6 +19,7 @@ type ftpMock struct {
proto *textproto.Conn proto *textproto.Conn
commands []string // list of received commands commands []string // list of received commands
rest int rest int
fileCont *bytes.Buffer
dataConn *mockDataConn dataConn *mockDataConn
sync.WaitGroup sync.WaitGroup
} }
@ -166,7 +167,7 @@ func (mock *ftpMock) listen(t *testing.T) {
mock.dataConn.Wait() mock.dataConn.Wait()
mock.proto.Writer.PrintfLine("150 Opening ASCII mode data connection for file list") mock.proto.Writer.PrintfLine("150 Opening ASCII mode data connection for file list")
mock.dataConn.conn.Write([]byte(testData[mock.rest:])) mock.dataConn.conn.Write(mock.fileCont.Bytes()[mock.rest:])
mock.rest = 0 mock.rest = 0
mock.proto.Writer.PrintfLine("226 Transfer complete") mock.proto.Writer.PrintfLine("226 Transfer complete")
mock.closeDataConn() mock.closeDataConn()
@ -270,7 +271,8 @@ func (mock *ftpMock) listenDataConn() (int64, error) {
func (mock *ftpMock) recvDataConn() { func (mock *ftpMock) recvDataConn() {
mock.dataConn.Wait() mock.dataConn.Wait()
io.Copy(ioutil.Discard, mock.dataConn.conn) mock.fileCont = new(bytes.Buffer)
io.Copy(mock.fileCont, mock.dataConn.conn)
mock.proto.Writer.PrintfLine("226 Transfer Complete") mock.proto.Writer.PrintfLine("226 Transfer Complete")
mock.closeDataConn() mock.closeDataConn()
} }