Rooba
/
stash
mirror of https://github.com/stashapp/stash.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix test database generator (#3112) [skip ci]

This commit is contained in:
WithoutPants 2022-11-10 15:01:47 +11:00 committed by GitHub
parent 9df66024d1
commit d0b0be4dd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
scripts/test_db_generator/makeTestDB.go
View File

@ -227,20 +227,16 @@ func makePerformers(n int) {
if err := retry(100, func() error { if err := retry(100, func() error {
return withTxn(func(ctx context.Context) error { return withTxn(func(ctx context.Context) error {
name := generatePerformerName() name := generatePerformerName()
performer := models.Performer{ performer := &models.Performer{
Name: sql.NullString{String: name, Valid: true}, Name: name,
Checksum: md5.FromString(name), Checksum: md5.FromString(name),
Favorite: sql.NullBool{
Bool: false,
Valid: true,
},
} }
// TODO - set tags // TODO - set tags
_, err := repo.Performer.Create(ctx, performer) err := repo.Performer.Create(ctx, performer)
if err != nil { if err != nil {
err = fmt.Errorf("error creating performer with name: %s: %s", performer.Name.String, err.Error()) err = fmt.Errorf("error creating performer with name: %s: %s", performer.Name, err.Error())
} }
return err return err
}) })
@ -444,8 +440,8 @@ func makeGalleries(n int) {
for ; i < batch && i < n; i++ { for ; i < batch && i < n; i++ {
gallery := generateGallery(i) gallery := generateGallery(i)
gallery.StudioID = getRandomStudioID(ctx) gallery.StudioID = getRandomStudioID(ctx)
gallery.TagIDs = getRandomTags(ctx, 0, 15) gallery.TagIDs = models.NewRelatedIDs(getRandomTags(ctx, 0, 15))
gallery.PerformerIDs = getRandomPerformers(ctx) gallery.PerformerIDs = models.NewRelatedIDs(getRandomPerformers(ctx))
path := md5.FromString("gallery/" + strconv.Itoa(i)) path := md5.FromString("gallery/" + strconv.Itoa(i))
f, err := makeZipFile(ctx, path) f, err := makeZipFile(ctx, path)
@ -564,10 +560,10 @@ func getRandomStudioID(ctx context.Context) *int {
func makeSceneRelationships(ctx context.Context, s *models.Scene) { func makeSceneRelationships(ctx context.Context, s *models.Scene) {
// add tags // add tags
s.TagIDs = getRandomTags(ctx, 0, 15) s.TagIDs = models.NewRelatedIDs(getRandomTags(ctx, 0, 15))
// add performers // add performers
s.PerformerIDs = getRandomPerformers(ctx) s.PerformerIDs = models.NewRelatedIDs(getRandomPerformers(ctx))
} }
func makeImageRelationships(ctx context.Context, i *models.Image) { func makeImageRelationships(ctx context.Context, i *models.Image) {
@ -576,12 +572,12 @@ func makeImageRelationships(ctx context.Context, i *models.Image) {
// add tags // add tags
if rand.Intn(100) == 0 { if rand.Intn(100) == 0 {
i.TagIDs = getRandomTags(ctx, 1, 15) i.TagIDs = models.NewRelatedIDs(getRandomTags(ctx, 1, 15))
} }
// add performers // add performers
if rand.Intn(100) <= 1 { if rand.Intn(100) <= 1 {
i.PerformerIDs = getRandomPerformers(ctx) i.PerformerIDs = models.NewRelatedIDs(getRandomPerformers(ctx))
} }
} }