doesnm/quickshare
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(userstore): add test for checking visitor

Browse source
This commit is contained in:
hexxa 2021-09-25 22:58:38 +08:00 • committed by Hexxa
parent a46d98d416
commit 23f16bb3c2
1 changed files with 31 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/userstore/user_store_test.go
View file

@ -36,7 +36,30 @@ func TestUserStores(t *testing.T) {
t.Fatalf("incorrect root DownloadSpeedLimit") t.Fatalf("incorrect root DownloadSpeedLimit")
} }
id, name1 := uint64(1), "test_user1" visitor, err := store.GetUser(1)
if err != nil {
t.Fatal(err)
}
if visitor.Name != VisitorName {
t.Fatal("visitor not found")
}
if visitor.Pwd != rootPwd {
t.Fatalf("passwords not match %s", err)
}
if visitor.Role != VisitorRole {
t.Fatalf("incorrect visitor role")
}
if visitor.Quota.SpaceLimit != 0 {
t.Fatalf("incorrect visitor SpaceLimit")
}
if visitor.Quota.UploadSpeedLimit != visitorUploadSpeedLimit {
t.Fatalf("incorrect visitor UploadSpeedLimit")
}
if visitor.Quota.DownloadSpeedLimit != visitorDownloadSpeedLimit {
t.Fatalf("incorrect visitor DownloadSpeedLimit")
}
id, name1 := uint64(2), "test_user1"
pwd1, pwd2 := "666", "888" pwd1, pwd2 := "666", "888"
role1, role2 := UserRole, AdminRole role1, role2 := UserRole, AdminRole
spaceLimit1, upLimit1, downLimit1 := int64(17), 5, 7 spaceLimit1, upLimit1, downLimit1 := int64(17), 5, 7
@ -53,6 +76,9 @@ func TestUserStores(t *testing.T) {
DownloadSpeedLimit: downLimit1, DownloadSpeedLimit: downLimit1,
}, },
}) })
if err != nil {
t.Fatal("there should be no error")
}
user, err := store.GetUser(id) user, err := store.GetUser(id)
if err != nil { if err != nil {
@ -81,8 +107,8 @@ func TestUserStores(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if len(users) != 2 { if len(users) != 3 {
t.Fatalf("users size should be 2 (%d)", len(users)) t.Fatalf("users size should be 3 (%d)", len(users))
} }
for _, user := range users { for _, user := range users {
if user.ID == 0 { if user.ID == 0 {
@ -90,7 +116,7 @@ func TestUserStores(t *testing.T) {
t.Fatalf("incorrect root info %v", user) t.Fatalf("incorrect root info %v", user)
} }
} }
if user.ID == 1 { if user.ID == id {
if user.Name != name1 || user.Role != role1 { if user.Name != name1 || user.Role != role1 {
t.Fatalf("incorrect user info %v", user) t.Fatalf("incorrect user info %v", user)
} }
@ -181,7 +207,7 @@ func TestUserStores(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if len(users) != 1 { if len(users) != 2 {
t.Fatalf("users size should be 2 (%d)", len(users)) t.Fatalf("users size should be 2 (%d)", len(users))
} }
if users[0].ID != 0 || users[0].Name != rootName || users[0].Role != AdminRole { if users[0].ID != 0 || users[0].Name != rootName || users[0].Role != AdminRole {