mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 11:36:36 +08:00
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
|
"github.com/ShukeBta/MediaStationGo/internal/repository"
|
|
"github.com/ShukeBta/MediaStationGo/internal/service"
|
|
)
|
|
|
|
func TestProfileHideAdultRequiresPasswordOnlyWhenChanged(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.User{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
user := &model.User{Username: "viewer", PasswordHash: "hash", Role: "user", HideAdult: true}
|
|
if err := repos.User.Create(t.Context(), user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := &service.Container{Repo: repos}
|
|
|
|
same := true
|
|
changed, err := profileHideAdultChanged(t.Context(), svc, user.ID, service.ProfileUpdate{HideAdult: &same})
|
|
if err != nil {
|
|
t.Fatalf("same value returned error: %v", err)
|
|
}
|
|
if changed {
|
|
t.Fatal("same hide_adult value should not require password")
|
|
}
|
|
|
|
next := false
|
|
changed, err = profileHideAdultChanged(t.Context(), svc, user.ID, service.ProfileUpdate{HideAdult: &next})
|
|
if err != nil {
|
|
t.Fatalf("changed value returned error: %v", err)
|
|
}
|
|
if !changed {
|
|
t.Fatal("changed hide_adult value should require password")
|
|
}
|
|
}
|