common_test.go

package neugo

import (
	"flag"
	"github.com/stretchr/testify/assert"
	"net/http"
	"strings"
	"testing"
)

func TestUse(t *testing.T) {
	a := assert.New(t)
	session := NewSession()
	s1 := Use(session).WithAuth("test", "test")
	ctx, ok := s1.(*useCtx)
	a.True(ok)
	a.Equal("test", ctx.Username)
	a.Equal("test", ctx.Password)

	emptyClient := &http.Client{}
	s2 := Use(emptyClient).WithToken("test")
	ctx, ok = s2.(*useCtx)
	a.True(ok)
	a.NotNil(ctx.Client.Jar)
	a.Equal("test", ctx.Token)
	a.Equal(true, ctx.UseToken)

	s3 := Use(session)
	s3.WithToken("test")
	s3.WithAuth("test", "test")
	ctx, ok = s3.(*useCtx)
	a.True(ok)
	a.Equal(false, ctx.UseToken)
	a.Equal("test", ctx.Username)
	a.Equal("test", ctx.Password)
}

func TestAll(t *testing.T) {
	a := assert.New(t)
	session := NewSession()
	if !flag.Parsed() {
		flag.Parse()
	}

	argList := flag.Args()
	if len(argList) < 2 {
		a.Fail("should provide account to continue tests.")
		return
	}
	username := argList[0]
	password := argList[1]

	err := Use(session).WithAuth(username, password).Login(CAS)
	if err != nil && strings.Contains(err.Error(), "timeout") {
		a.Nil(err)
	}
	err = Use(session).WithAuth(username, password).Login(WebVPN)
	if err != nil && strings.Contains(err.Error(), "timeout") {
		a.Nil(err)
	}
	_, err = Use(session).WithAuth(username, password).DebugLogin(CAS)
	if err != nil && strings.Contains(err.Error(), "timeout") {
		a.Nil(err)
	}

	// about
	{
		token := About(session).Token(CAS)
		a.NotEmpty(token)
		session := NewSession()
		err = Use(session).WithToken(token).Login(CAS)
		if err != nil && strings.Contains(err.Error(), "timeout") {
			a.Nil(err)
		}
	}

	{
		token := About(session).Token(WebVPN)
		a.NotEmpty(token)
		session := NewSession()
		err = Use(session).WithToken(token).Login(WebVPN)
		if err != nil && strings.Contains(err.Error(), "timeout") {
			a.Nil(err)
		}
	}
}