84 lines
2.0 KiB
Go
84 lines
2.0 KiB
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
|
|
}
|