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

test(files/quota): add e2e tests

Browse source
This commit is contained in:
hexxa 2021-08-17 14:41:15 +08:00 • committed by Hexxa
parent 41654e36d0
commit 839d7f43bc
4 changed files with 163 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/handlers/fileshdr/handlers.go
View file

@ -147,7 +147,11 @@ func (h *FileHandlers) Create(c *gin.Context) {
locker.Exec(func() {
err = h.deps.Users().SetUsed(userIDInt, true, req.FileSize)
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
if userstore.IsReachedLimitErr(err) {
c.JSON(q.ErrResp(c, 429, err))
} else {
c.JSON(q.ErrResp(c, 500, err))
}
return
}
@ -206,7 +210,11 @@ func (h *FileHandlers) Delete(c *gin.Context) {
err = h.deps.Users().SetUsed(userIDInt, false, info.Size())
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
if userstore.IsReachedLimitErr(err) {
c.JSON(q.ErrResp(c, 429, err))
} else {
c.JSON(q.ErrResp(c, 500, err))
}
return
}
@ -702,7 +710,11 @@ func (h *FileHandlers) DelUploading(c *gin.Context) {
err = h.deps.Users().SetUsed(userIDInt, false, size)
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
if userstore.IsReachedLimitErr(err) {
c.JSON(q.ErrResp(c, 429, err))
} else {
c.JSON(q.ErrResp(c, 500, err))
}
return
}