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

test(e2e): refine space limit tests

Browse source
This commit is contained in:
hexxa 2021-09-29 18:49:38 +08:00 • committed by Hexxa
parent b8fdb2bbee
commit 5605839411
2 changed files with 32 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

38
src/server/server_space_limit_test.go
View file

@ -13,7 +13,13 @@ import (
func TestSpaceLimit(t *testing.T) {
addr := "http://127.0.0.1:8686"
root := "testData"
config := `{
spaceLimit := 1000000
fileSize := 100000
if spaceLimit%fileSize != 0 {
t.Fatal("spaceLimit % fileSize must be zero")
}
config := fmt.Sprintf(`{
"users": {
"enableAuth": true,
"minUserNameLen": 2,
@ -21,7 +27,7 @@ func TestSpaceLimit(t *testing.T) {
"captchaEnabled": false,
"uploadSpeedLimit": 409600,
"downloadSpeedLimit": 409600,
"spaceLimit": 100,
"spaceLimit": %d,
"limiterCapacity": 1000,
"limiterCyc": 1000
},
@ -32,7 +38,7 @@ func TestSpaceLimit(t *testing.T) {
"fs": {
"root": "testData"
}
}`
}`, spaceLimit)
adminName := "qs"
adminPwd := "quicksh@re"
@ -100,11 +106,11 @@ func TestSpaceLimit(t *testing.T) {
token := client.GetCookie(resp.Cookies(), q.TokenCookie)
fileContent := ""
for i := 0; i < 10; i++ {
for i := 0; i < fileSize; i++ {
fileContent += "0"
}
for i := 0; i < 10; i++ {
for i := 0; i < spaceLimit/fileSize; i++ {
ok := assertUploadOK(t, fmt.Sprintf("%s/files/spacelimit/f_%d", getUserName(0), i), fileContent, addr, token)
if !ok {
t.Fatalf("space limit failed at %d", i)
@ -114,8 +120,8 @@ func TestSpaceLimit(t *testing.T) {
if len(errs) > 0 {
t.Fatal(errs)
} else if resp.StatusCode != 200 {
t.Fatal("failed to add user")
} else if selfResp.UsedSpace != int64((i+1)*10) {
t.Fatal("failed to get self")
} else if selfResp.UsedSpace != int64((i+1)*fileSize) {
t.Fatal("incorrect used space")
}
}
@ -128,6 +134,24 @@ func TestSpaceLimit(t *testing.T) {
} else if res.StatusCode != 429 {
t.Fatal("(space limit): this request should be rejected")
}
for i := 0; i < spaceLimit/fileSize; i++ {
resp, _, errs := cl.Delete(fmt.Sprintf("%s/files/spacelimit/f_%d", getUserName(0), i))
if len(errs) > 0 {
t.Fatalf("failed to delete %d", i)
} else if resp.StatusCode != 200 {
t.Fatalf("failed to delete status %d", resp.StatusCode)
}
resp, selfResp, errs := usersCl.Self(token)
if len(errs) > 0 {
t.Fatal(errs)
} else if resp.StatusCode != 200 {
t.Fatal("failed to get self")
} else if selfResp.UsedSpace != int64(spaceLimit)-int64((i+1)*fileSize) {
t.Fatal("incorrect used space")
}
}
})
resp, _, errs = usersCl.Logout(token)