chore(e2e): add addUsers helper func

This commit is contained in:
hexxa 2021-09-30 17:12:49 +08:00 committed by Hexxa
parent 74aa28d874
commit e1ad4c203e
2 changed files with 26 additions and 18 deletions

View file

@ -15,6 +15,7 @@ import (
"github.com/ihexxa/gocfg"
"github.com/ihexxa/quickshare/src/client"
fspkg "github.com/ihexxa/quickshare/src/fs"
"github.com/ihexxa/quickshare/src/userstore"
)
func startTestServer(config string) *Server {
@ -51,6 +52,29 @@ func setUpEnv(t *testing.T, rootPath string, adminName, adminPwd string) {
}
}
func getUserName(id int) string {
return fmt.Sprintf("user_%d", id)
}
func addUsers(t *testing.T, addr, userPwd string, userCount int, adminToken *http.Cookie) map[string]string {
usersCl := client.NewSingleUserClient(addr)
users := map[string]string{}
for i := range make([]int, userCount) {
userName := getUserName(i)
resp, adResp, errs := usersCl.AddUser(userName, userPwd, userstore.UserRole, adminToken)
if len(errs) > 0 {
t.Fatal(errs)
} else if resp.StatusCode != 200 {
t.Fatal("failed to add user")
}
users[userName] = adResp.ID
}
return users
}
func isServerReady(addr string) bool {
retry := 20
setCl := client.NewSettingsClient(addr)