mirror of
https://github.com/stashapp/stash.git
synced 2025-12-07 08:54:10 +01:00
5495d72849
* Restructure data layer part 2 (#2599) * Refactor and separate image model * Refactor image query builder * Handle relationships in image query builder * Remove relationship management methods * Refactor gallery model/query builder * Add scenes to gallery model * Convert scene model * Refactor scene models * Remove unused methods * Add unit tests for gallery * Add image tests * Add scene tests * Convert unnecessary scene value pointers to values * Convert unnecessary pointer values to values * Refactor scene partial * Add scene partial tests * Refactor ImagePartial * Add image partial tests * Refactor gallery partial update * Add partial gallery update tests * Use zero/null package for null values * Add files and scan system * Add sqlite implementation for files/folders * Add unit tests for files/folders * Image refactors * Update image data layer * Refactor gallery model and creation * Refactor scene model * Refactor scenes * Don't set title from filename * Allow galleries to freely add/remove images * Add multiple scene file support to graphql and UI * Add multiple file support for images in graphql/UI * Add multiple file for galleries in graphql/UI * Remove use of some deprecated fields * Remove scene path usage * Remove gallery path usage * Remove path from image * Move funscript to video file * Refactor caption detection * Migrate existing data * Add post commit/rollback hook system * Lint. Comment out import/export tests * Add WithDatabase read only wrapper * Prepend tasks to list * Add 32 pre-migration * Add warnings in release and migration notes
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
//go:build integration
|
|
// +build integration
|
|
|
|
package sqlite_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type stashIDReaderWriter interface {
|
|
GetStashIDs(ctx context.Context, performerID int) ([]*models.StashID, error)
|
|
UpdateStashIDs(ctx context.Context, performerID int, stashIDs []*models.StashID) error
|
|
}
|
|
|
|
func testStashIDReaderWriter(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int) {
|
|
// ensure no stash IDs to begin with
|
|
testNoStashIDs(ctx, t, r, id)
|
|
|
|
// ensure GetStashIDs with non-existing also returns none
|
|
testNoStashIDs(ctx, t, r, -1)
|
|
|
|
// add stash ids
|
|
const stashIDStr = "stashID"
|
|
const endpoint = "endpoint"
|
|
stashID := &models.StashID{
|
|
StashID: stashIDStr,
|
|
Endpoint: endpoint,
|
|
}
|
|
|
|
// update stash ids and ensure was updated
|
|
if err := r.UpdateStashIDs(ctx, id, []*models.StashID{stashID}); err != nil {
|
|
t.Error(err.Error())
|
|
}
|
|
|
|
testStashIDs(ctx, t, r, id, []*models.StashID{stashID})
|
|
|
|
// update non-existing id - should return error
|
|
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(ctx, id, []*models.StashID{}); err != nil {
|
|
t.Error(err.Error())
|
|
}
|
|
|
|
testNoStashIDs(ctx, t, r, id)
|
|
}
|
|
|
|
func testNoStashIDs(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int) {
|
|
t.Helper()
|
|
stashIDs, err := r.GetStashIDs(ctx, id)
|
|
if err != nil {
|
|
t.Error(err.Error())
|
|
return
|
|
}
|
|
|
|
assert.Len(t, stashIDs, 0)
|
|
}
|
|
|
|
func testStashIDs(ctx context.Context, t *testing.T, r stashIDReaderWriter, id int, expected []*models.StashID) {
|
|
t.Helper()
|
|
stashIDs, err := r.GetStashIDs(ctx, id)
|
|
if err != nil {
|
|
t.Error(err.Error())
|
|
return
|
|
}
|
|
|
|
assert.Equal(t, stashIDs, expected)
|
|
}
|