add more tests

This commit is contained in:
Zack Scholl 2019-09-08 05:53:30 -07:00
parent 6287a7b7f7
commit f89dd01e87
2 changed files with 40 additions and 1 deletions

View File

@ -300,7 +300,7 @@ func (c *Client) Send(options TransferOptions) (err error) {
go func() {
log.Debugf("establishing connection to %s", c.Options.RelayAddress)
var banner string
conn, banner, ipaddr, err := tcp.ConnectToTCPServer(c.Options.RelayAddress, c.Options.SharedSecret)
conn, banner, ipaddr, err := tcp.ConnectToTCPServer(c.Options.RelayAddress, c.Options.SharedSecret, 5*time.Second)
log.Debugf("banner: %s", banner)
if err != nil {
err = errors.Wrap(err, fmt.Sprintf("could not connect to %s", c.Options.RelayAddress))

View File

@ -1,6 +1,7 @@
package croc
import (
"io/ioutil"
"os"
"sync"
"testing"
@ -8,6 +9,7 @@ import (
"github.com/schollz/croc/v6/src/tcp"
log "github.com/schollz/logger"
"github.com/stretchr/testify/assert"
)
func TestCroc(t *testing.T) {
@ -120,3 +122,40 @@ func TestCrocLocal(t *testing.T) {
wg.Wait()
}
func TestCrocError(t *testing.T) {
content := []byte("temporary file's content")
tmpfile, err := ioutil.TempFile("", "example")
if err != nil {
panic(err)
}
defer os.Remove(tmpfile.Name()) // clean up
if _, err := tmpfile.Write(content); err != nil {
panic(err)
}
if err := tmpfile.Close(); err != nil {
panic(err)
}
Debug(false)
log.SetLevel("warn")
sender, _ := New(Options{
IsSender: true,
SharedSecret: "test",
Debug: true,
RelayAddress: "doesntexistok.com:8181",
RelayPorts: []string{"8181", "8182"},
Stdout: true,
NoPrompt: true,
DisableLocal: true,
})
err = sender.Send(TransferOptions{
PathToFiles: []string{tmpfile.Name()},
KeepPathInRemote: true,
})
log.Debug(err)
assert.NotNil(t, err)
}