fusenapi/utils/tests/basic.go

package fstests

import (
	"fmt"
	"testing"

	"github.com/474420502/requests"
)

func GetSesssion() *requests.Session {
	ses := requests.NewSession()
	return ses
}

func GetSessionWithUserToken(t *testing.T, server requests.ITestServer, Host string, Port int) *requests.Session {
	ses := requests.NewSession()
	tp := ses.Post(fmt.Sprintf("http://%s:%d/user/login", Host, Port))
	tp.SetBodyJson(map[string]interface{}{
		"name": "9107058@qq.com",
		"pwd":  "$2y$13$2y4O4OIz/zcK5C0vlSc9LuSpjWySjInLBSe49yDkE.iURb.R1hDsy",
	})
	resp, err := tp.TestExecute(server)
	if err != nil {
		t.Error(err)
	}
	result := resp.Json()
	code := result.Get("code").Int()
	if code != 200 {
		t.Error("code is not 200")
	}

	token := result.Get("data.token")
	if !token.Exists() {
		t.Error("data.token is not exists")
	}
	ses.Header.Add("Authorization", token.String())

	return ses
}

func GetBackendSessionWithUserToken(t *testing.T, server requests.ITestServer, Host string, Port int) *requests.Session {
	ses := requests.NewSession()
	tp := ses.Post(fmt.Sprintf("http://%s:%d/backend-user/login", Host, Port))
	tp.SetBodyJson(map[string]interface{}{
		"name": "admin@admin.com",
		"pwd":  "ZnVzZW5fYmFja2VuZF8yMDIz47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU=",
	})
	resp, err := tp.TestExecute(server)
	if err != nil {
		t.Error(err)
	}
	result := resp.Json()
	code := result.Get("code").Int()
	if code != 200 {
		t.Error("code is not 200")
	}

	token := result.Get("data.token")
	if !token.Exists() {
		t.Error("data.token is not exists")
	}
	ses.Header.Add("Authorization", token.String())

	return ses
}

func GetSesssionWithGuestToken(t *testing.T, server requests.ITestServer, Host string, Port int) *requests.Session {
	ses := requests.NewSession()
	tp := ses.Post(fmt.Sprintf("http://%s:%d/accept/cookie", Host, Port))

	resp, err := tp.TestExecute(server)
	if err != nil {
		t.Error(err)
	}
	result := resp.Json()
	token := result.Get("data.token")
	if !token.Exists() {
		t.Error("data.token is not exists")
	}
	ses.Header.Add("Authorization", token.String())

	return ses
}