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

fix(db): db interfaces should be updated in tests

Browse source
This commit is contained in:
hexxa 2022-09-05 21:37:42 +08:00 • committed by Hexxa
parent ce77eb7534
commit b226bc6f9f
1 changed files with 14 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/db/tests/files_test.go
View file

@ -54,12 +54,14 @@ func TestFileStore(t *testing.T) {
if _, ok := dirToID[sharingDir]; !ok { if _, ok := dirToID[sharingDir]; !ok {
t.Fatalf("sharing(%s) not found", sharingDir) t.Fatalf("sharing(%s) not found", sharingDir)
} }
mustTrue := store.IsSharing(ctx, adminId, sharingDir) mustTrue, err := store.IsSharing(ctx, adminId, sharingDir)
if !mustTrue { if err != nil {
t.Fatal(err)
} else if !mustTrue {
t.Fatalf("get sharing(%t) should exist", mustTrue) t.Fatalf("get sharing(%t) should exist", mustTrue)
} }
info, err := store.GetFileInfo(ctx, adminId, sharingDir) info, err := store.GetFileInfo(ctx, sharingDir)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} else if len(info.ShareID) != 7 { } else if len(info.ShareID) != 7 {
@ -94,12 +96,14 @@ func TestFileStore(t *testing.T) {
if _, ok := dirToIDAfterDel[dirPath]; ok { if _, ok := dirToIDAfterDel[dirPath]; ok {
t.Fatalf("sharing(%s) should not exist", dirPath) t.Fatalf("sharing(%s) should not exist", dirPath)
} }
shared := store.IsSharing(ctx, adminId, dirPath) shared, err := store.IsSharing(ctx, adminId, dirPath)
if shared { if err != nil {
t.Fatal(err)
} else if shared {
t.Fatalf("get sharing(%t) should not shared but exist", shared) t.Fatalf("get sharing(%t) should not shared but exist", shared)
} }
info, err := store.GetFileInfo(ctx, adminId, dirPath) info, err := store.GetFileInfo(ctx, dirPath)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} else if len(info.ShareID) != 0 { } else if len(info.ShareID) != 0 {
@ -161,7 +165,7 @@ func TestFileStore(t *testing.T) {
// get infos // get infos
for itemPath, expected := range pathInfos { for itemPath, expected := range pathInfos {
info, err := store.GetFileInfo(ctx, adminId, itemPath) info, err := store.GetFileInfo(ctx, itemPath)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -196,11 +200,11 @@ func TestFileStore(t *testing.T) {
// set sha1 // set sha1
testSha1 := "sha1" testSha1 := "sha1"
for itemPath := range pathInfos { for itemPath := range pathInfos {
err := store.SetSha1(ctx, adminId, itemPath, testSha1) err := store.SetSha1(ctx, itemPath, testSha1)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
info, err := store.GetFileInfo(ctx, adminId, itemPath) info, err := store.GetFileInfo(ctx, itemPath)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -261,7 +265,7 @@ func TestFileStore(t *testing.T) {
} }
for itemPath := range pathInfos { for itemPath := range pathInfos {
_, err := store.GetFileInfo(ctx, adminId, itemPath) _, err := store.GetFileInfo(ctx, itemPath)
if !errors.Is(err, db.ErrFileInfoNotFound) { if !errors.Is(err, db.ErrFileInfoNotFound) {
t.Fatal(err) t.Fatal(err)
} }