wsy182/Cloudreve
1
0
Fork 0
mirror of https://github.com/halejohn/Cloudreve.git synced 2026-01-26 17:41:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: test failed due to missing context

Browse Source
This commit is contained in:
HFO4
2019-11-19 18:10:18 +08:00
parent 20ea86eaf6
commit f9b37a3359
4 changed files with 40 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/filesystem/hooks_test.go
View File

@@ -10,11 +10,11 @@ import (
func TestGenericBeforeUpload(t *testing.T) {
asserts := assert.New(t)
ctx := context.Background()
file := local.FileData{
Size: 5,
Name: "1.txt",
}
ctx := context.WithValue(context.Background(), FileHeaderCtx, file)
fs := FileSystem{
User: &model.User{
Storage: 0,
@@ -30,12 +30,18 @@ func TestGenericBeforeUpload(t *testing.T) {
},
}
asserts.Error(GenericBeforeUpload(ctx, &fs, file))
asserts.Error(GenericBeforeUpload(ctx, &fs))
file.Size = 1
file.Name = "1"
asserts.Error(GenericBeforeUpload(ctx, &fs, file))
ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
asserts.Error(GenericBeforeUpload(ctx, &fs))
file.Name = "1.txt"
asserts.NoError(GenericBeforeUpload(ctx, &fs, file))
ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
asserts.NoError(GenericBeforeUpload(ctx, &fs))
file.Name = "1.t/xt"
asserts.Error(GenericBeforeUpload(ctx, &fs, file))
ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
asserts.Error(GenericBeforeUpload(ctx, &fs))
}