Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-18 04:44:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restructure data layer (#2532)

Browse Source 
* Add new txn manager interface
* Add txn management to sqlite
* Rename get to getByID
* Add contexts to repository methods
* Update query builders
* Add context to reader writer interfaces
* Use repository in resolver
* Tighten interfaces
* Tighten interfaces in dlna
* Tighten interfaces in match package
* Tighten interfaces in scraper package
* Tighten interfaces in scan code
* Tighten interfaces on autotag package
* Remove ReaderWriter usage
* Merge database package into sqlite
This commit is contained in:
WithoutPants
2022-05-19 17:49:32 +10:00
parent 7b5bd80515
commit 964b559309
244 changed files with 7377 additions and 6699 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
pkg/sqlite/stash_id_test.go
View File

@@ -4,6 +4,7 @@
package sqlite_test
import (
"context"
"testing"
"github.com/stashapp/stash/pkg/models"
@@ -11,16 +12,16 @@ import (
)
type stashIDReaderWriter interface {
GetStashIDs(performerID int) ([]*models.StashID, error)
UpdateStashIDs(performerID int, stashIDs []models.StashID) error
GetStashIDs(ctx context.Context, performerID int) ([]*models.StashID, error)
UpdateStashIDs(ctx context.Context, performerID int, stashIDs []models.StashID) error
}
func testStashIDReaderWriter(t *testing.T, r stashIDReaderWriter, id int) {
func testStashIDReaderWriter(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int) {
// ensure no stash IDs to begin with
testNoStashIDs(t, r, id)
testNoStashIDs(ctx, t, r, id)
// ensure GetStashIDs with non-existing also returns none
testNoStashIDs(t, r, -1)
testNoStashIDs(ctx, t, r, -1)
// add stash ids
const stashIDStr = "stashID"
@@ -31,28 +32,28 @@ func testStashIDReaderWriter(t *testing.T, r stashIDReaderWriter, id int) {
}
// update stash ids and ensure was updated
if err := r.UpdateStashIDs(id, []models.StashID{stashID}); err != nil {
if err := r.UpdateStashIDs(ctx, id, []models.StashID{stashID}); err != nil {
t.Error(err.Error())
}
testStashIDs(t, r, id, []*models.StashID{&stashID})
testStashIDs(ctx, t, r, id, []*models.StashID{&stashID})
// update non-existing id - should return error
if err := r.UpdateStashIDs(-1, []models.StashID{stashID}); err == nil {
if err := r.UpdateStashIDs(ctx, -1, []models.StashID{stashID}); err == nil {
t.Error("expected error when updating non-existing id")
}
// remove stash ids and ensure was updated
if err := r.UpdateStashIDs(id, []models.StashID{}); err != nil {
if err := r.UpdateStashIDs(ctx, id, []models.StashID{}); err != nil {
t.Error(err.Error())
}
testNoStashIDs(t, r, id)
testNoStashIDs(ctx, t, r, id)
}
func testNoStashIDs(t *testing.T, r stashIDReaderWriter, id int) {
func testNoStashIDs(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int) {
t.Helper()
stashIDs, err := r.GetStashIDs(id)
stashIDs, err := r.GetStashIDs(ctx, id)
if err != nil {
t.Error(err.Error())
return
@@ -61,9 +62,9 @@ func testNoStashIDs(t *testing.T, r stashIDReaderWriter, id int) {
assert.Len(t, stashIDs, 0)
}
func testStashIDs(t *testing.T, r stashIDReaderWriter, id int, expected []*models.StashID) {
func testStashIDs(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int, expected []*models.StashID) {
t.Helper()
stashIDs, err := r.GetStashIDs(id)
stashIDs, err := r.GetStashIDs(ctx, id)
if err != nil {
t.Error(err.Error())
return