raw_conn_test.go

// Copyright 2015 Muir Manders.  All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package goftp

import (
	"io/ioutil"
	"strings"
	"testing"
)

func TestRawConn(t *testing.T) {
	for _, addr := range ftpdAddrs {
		c, err := DialConfig(goftpConfig, addr)

		if err != nil {
			t.Fatal(err)
		}

		rawConn, err := c.OpenRawConn()
		if err != nil {
			t.Fatal(err)
		}

		code, msg, err := rawConn.SendCommand("FEAT")
		if err != nil {
			t.Fatal(err)
		}

		if code != 211 {
			t.Errorf("got %d", code)
		}

		if !strings.Contains(msg, "REST") {
			t.Errorf("got %s", msg)
		}

		dcGetter, err := rawConn.PrepareDataConn()
		if err != nil {
			t.Fatal(err)
		}

		_, _, err = rawConn.SendCommand("LIST")
		if err != nil {
			t.Fatal(err)
		}

		dc, err := dcGetter()
		if err != nil {
			t.Fatal(err)
		}

		got, err := ioutil.ReadAll(dc)
		if err != nil {
			t.Fatal(err)
		}

		if !strings.Contains(string(got), "lorem.txt") {
			t.Errorf("got %s", got)
		}

		dc.Close()

		code, _, err = rawConn.ReadResponse()
		if err != nil {
			t.Fatal(err)
		}
		if code != 226 {
			t.Errorf("got: %d", code)
		}

		if err := rawConn.Close(); err != nil {
			t.Error(err)
		}
	}
}