172 lines
3.2 KiB
Go
172 lines
3.2 KiB
Go
package ftp
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/textproto"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const (
|
|
testData = "Just some text"
|
|
testDir = "mydir"
|
|
)
|
|
|
|
func TestConnPASV(t *testing.T) {
|
|
testConn(t, true)
|
|
}
|
|
|
|
func TestConnEPSV(t *testing.T) {
|
|
testConn(t, false)
|
|
}
|
|
|
|
func testConn(t *testing.T, disableEPSV bool) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
require := require.New(t)
|
|
|
|
c, err := DialTimeout("localhost:21", 5*time.Second)
|
|
require.NoError(err)
|
|
|
|
if disableEPSV {
|
|
delete(c.features, "EPSV")
|
|
c.disableEPSV = true
|
|
}
|
|
|
|
err = c.Login("anonymous", "anonymous")
|
|
require.NoError(err)
|
|
|
|
err = c.NoOp()
|
|
require.NoError(err)
|
|
|
|
err = c.ChangeDir("incoming")
|
|
require.NoError(err)
|
|
|
|
data := bytes.NewBufferString(testData)
|
|
err = c.Stor("test", data)
|
|
require.NoError(err)
|
|
|
|
_, err = c.List(".")
|
|
require.NoError(err)
|
|
|
|
err = c.Rename("test", "tset")
|
|
require.NoError(err)
|
|
|
|
r, err := c.Retr("tset")
|
|
require.NoError(err)
|
|
buf, err := ioutil.ReadAll(r)
|
|
require.NoError(err)
|
|
require.Equal(testData, string(buf))
|
|
err = r.Close()
|
|
require.NoError(err)
|
|
|
|
r, err = c.RetrFrom("tset", 5)
|
|
require.NoError(err)
|
|
buf, err = ioutil.ReadAll(r)
|
|
require.NoError(err)
|
|
require.Equal(testData[5:], string(buf))
|
|
r.Close()
|
|
|
|
err = c.Delete("tset")
|
|
require.NoError(err)
|
|
|
|
err = c.MakeDir(testDir)
|
|
require.NoError(err)
|
|
|
|
err = c.ChangeDir(testDir)
|
|
require.NoError(err)
|
|
|
|
dir, err := c.CurrentDir()
|
|
require.NoError(err)
|
|
require.Equal("/incoming/"+testDir, dir)
|
|
|
|
err = c.ChangeDirToParent()
|
|
require.NoError(err)
|
|
|
|
entries, err := c.NameList("/")
|
|
require.NoError(err)
|
|
require.EqualValues([]string{"/incoming"}, entries)
|
|
|
|
err = c.RemoveDir(testDir)
|
|
require.NoError(err)
|
|
|
|
err = c.Logout()
|
|
if err != nil {
|
|
if protoErr := err.(*textproto.Error); protoErr != nil {
|
|
if protoErr.Code != StatusNotImplemented {
|
|
t.Error(err)
|
|
}
|
|
} else {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
err = c.Quit()
|
|
require.NoError(err)
|
|
|
|
err = c.NoOp()
|
|
require.Error(err)
|
|
require.Regexp("write tcp .* use of closed network connection", err.Error())
|
|
}
|
|
|
|
func TestConnIPv6(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
require := require.New(t)
|
|
|
|
c, err := DialTimeout("[::1]:21", 5*time.Second)
|
|
require.NoError(err)
|
|
|
|
err = c.Login("anonymous", "anonymous")
|
|
require.NoError(err)
|
|
|
|
_, err = c.List(".")
|
|
require.NoError(err)
|
|
|
|
err = c.Quit()
|
|
require.NoError(err)
|
|
}
|
|
|
|
// TestConnect tests the legacy Connect function
|
|
func TestConnect(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
|
|
c, err := Connect("localhost:21")
|
|
require.NoError(t, err)
|
|
|
|
c.Quit()
|
|
}
|
|
|
|
func TestDialTimeout(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
require := require.New(t)
|
|
|
|
_, err := DialTimeout("localhost:2121", 1*time.Second)
|
|
require.Error(err)
|
|
require.Regexp("dial tcp .* connection refused", err.Error())
|
|
}
|
|
|
|
func TestWrongLogin(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
require := require.New(t)
|
|
|
|
c, err := DialTimeout("localhost:21", 5*time.Second)
|
|
require.NoError(err)
|
|
defer c.Quit()
|
|
|
|
err = c.Login("zoo2Shia", "fei5Yix9")
|
|
require.Error(err)
|
|
require.Regexp("(Login incorrect|anonymous only)", err.Error())
|
|
}
|