gigtest_test.go

package gig

import (
	"testing"
	"time"

	"github.com/matryer/is"
)

func TestNewFakeContext(t *testing.T) {
	is := is.New(t)
	g := New()

	c, conn := g.NewFakeContext("/login", nil)

	is.NoErr(c.Response().WriteHeader(StatusGone, "oops"))
	is.Equal("52 oops\r\n", conn.Written)

	n, err := conn.Read(make([]byte, 1))
	is.Equal(1, n)
	is.NoErr(err)

	n, err = conn.Write([]byte("test"))
	is.Equal(4, n)
	is.NoErr(err)

	is.Equal(nil, conn.Close())
	is.Equal(conn.LocalAddr().String(), "192.0.2.1:25")
	is.Equal(conn.RemoteAddr().String(), "192.0.2.1:25")
	is.Equal(nil, conn.SetDeadline(time.Now()))
	is.Equal(nil, conn.SetReadDeadline(time.Now()))
	is.Equal(nil, conn.SetWriteDeadline(time.Now()))
}

func TestNewFakeContext_panic(t *testing.T) {
	var (
		is = is.New(t)
		g  = New()
	)

	defer func() {
		r := recover()
		is.True(r != nil)
	}()

	_, _ = g.NewFakeContext(":", nil)

	is.Fail()
}

func TestFakeAddr(t *testing.T) {
	is := is.New(t)
	addr := &FakeAddr{}

	is.Equal("tcp", addr.Network())
	is.Equal("192.0.2.1:25", addr.String())
}

func TestFakeConn(t *testing.T) {
	is := is.New(t)
	conn := &FakeConn{FailAfter: 5}

	n, err := conn.Write([]byte("test"))
	is.Equal(4, n)
	is.NoErr(err)

	n, err = conn.Write([]byte("more"))
	is.Equal(1, n)

	if err == nil {
		is.Fail()
	}
}