代码拉取完成,页面将自动刷新
// Copyright 2018 The goftp Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package server_test
import (
"io/ioutil"
"net"
"os"
"strings"
"testing"
"time"
filedriver "github.com/goftp/file-driver"
"github.com/goftp/server"
"github.com/jlaffaye/ftp"
"github.com/stretchr/testify/assert"
)
func runServer(t *testing.T, execute func()) {
os.MkdirAll("./testdata", os.ModePerm)
var perm = server.NewSimplePerm("test", "test")
opt := &server.ServerOpts{
Name: "test ftpd",
Factory: &filedriver.FileDriverFactory{
RootPath: "./testdata",
Perm: perm,
},
Port: 2121,
Auth: &server.SimpleAuth{
Name: "admin",
Password: "admin",
},
Logger: new(server.DiscardLogger),
}
s := server.NewServer(opt)
go func() {
err := s.ListenAndServe()
assert.EqualError(t, err, server.ErrServerClosed.Error())
}()
execute()
assert.NoError(t, s.Shutdown())
}
func TestConnect(t *testing.T) {
runServer(t, func() {
// Give server 0.5 seconds to get to the listening state
timeout := time.NewTimer(time.Millisecond * 500)
for {
f, err := ftp.Connect("localhost:2121")
if err != nil && len(timeout.C) == 0 { // Retry errors
continue
}
assert.NoError(t, err)
assert.NoError(t, f.Login("admin", "admin"))
assert.Error(t, f.Login("admin", ""))
var content = `test`
assert.NoError(t, f.Stor("server_test.go", strings.NewReader(content)))
names, err := f.NameList("/")
assert.NoError(t, err)
assert.EqualValues(t, 1, len(names))
assert.EqualValues(t, "server_test.go", names[0])
bs, err := ioutil.ReadFile("./testdata/server_test.go")
assert.NoError(t, err)
assert.EqualValues(t, content, string(bs))
entries, err := f.List("/")
assert.NoError(t, err)
assert.EqualValues(t, 1, len(entries))
assert.EqualValues(t, "server_test.go", entries[0].Name)
assert.EqualValues(t, 4, entries[0].Size)
assert.EqualValues(t, ftp.EntryTypeFile, entries[0].Type)
curDir, err := f.CurrentDir()
assert.NoError(t, err)
assert.EqualValues(t, "/", curDir)
size, err := f.FileSize("/server_test.go")
assert.NoError(t, err)
assert.EqualValues(t, 4, size)
/*resp, err := f.RetrFrom("/server_test.go", 0)
assert.NoError(t, err)
var buf []byte
l, err := resp.Read(buf)
assert.NoError(t, err)
assert.EqualValues(t, 4, l)
assert.EqualValues(t, 4, len(buf))
assert.EqualValues(t, content, string(buf))*/
err = f.Rename("/server_test.go", "/server.test.go")
assert.NoError(t, err)
err = f.MakeDir("/src")
assert.NoError(t, err)
err = f.Delete("/server.test.go")
assert.NoError(t, err)
err = f.RemoveDir("/src")
assert.NoError(t, err)
err = f.Quit()
assert.NoError(t, err)
break
}
})
}
func TestServe(t *testing.T) {
os.MkdirAll("./testdata", os.ModePerm)
var perm = server.NewSimplePerm("test", "test")
// Server options without hostname or port
opt := &server.ServerOpts{
Name: "test ftpd",
Factory: &filedriver.FileDriverFactory{
RootPath: "./testdata",
Perm: perm,
},
Auth: &server.SimpleAuth{
Name: "admin",
Password: "admin",
},
Logger: new(server.DiscardLogger),
}
// Start the listener
l, err := net.Listen("tcp", ":2121")
assert.NoError(t, err)
// Start the server using the listener
s := server.NewServer(opt)
go func() {
err := s.Serve(l)
assert.EqualError(t, err, server.ErrServerClosed.Error())
}()
// Give server 0.5 seconds to get to the listening state
timeout := time.NewTimer(time.Millisecond * 500)
for {
f, err := ftp.Connect("localhost:2121")
if err != nil && len(timeout.C) == 0 { // Retry errors
continue
}
assert.NoError(t, err)
assert.NoError(t, f.Login("admin", "admin"))
assert.Error(t, f.Login("admin", ""))
err = f.Quit()
assert.NoError(t, err)
break
}
assert.NoError(t, s.Shutdown())
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。