fix tests

woogles-io · Feb 6, 2025 · dc04275 · dc04275
1 parent df5573c
commit dc04275
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/pkg/mod/service_test.go b/pkg/mod/service_test.go
@@ -7,7 +7,10 @@ import (
 
 	"github.com/matryer/is"
 	"github.com/woogles-io/liwords/pkg/apiserver"
+	"github.com/woogles-io/liwords/pkg/auth/rbac"
 	"github.com/woogles-io/liwords/pkg/entity"
+	"github.com/woogles-io/liwords/pkg/stores/common"
+	"github.com/woogles-io/liwords/pkg/stores/models"
 	"github.com/woogles-io/liwords/pkg/stores/user"
 	pb "github.com/woogles-io/liwords/rpc/api/proto/mod_service"
 )
@@ -25,9 +28,18 @@ func TestAuthenticateMod(t *testing.T) {
 
 	recreateDB()
 	us := userStore()
-	ms := &ModService{userStore: us}
 
-	_, err := authenticateMod(ctx, ms, &pb.ModActionsList{
+	pool, err := common.OpenTestingDB(pkg)
+	is.NoErr(err)
+	q := models.New(pool)
+	ms := &ModService{userStore: us, queries: q}
+	err = q.AssignRole(ctx, models.AssignRoleParams{
+		Username: common.ToPGTypeText("Moderator"),
+		RoleName: string(rbac.Moderator),
+	})
+	is.NoErr(err)
+
+	_, err = authenticateMod(ctx, ms, &pb.ModActionsList{
 		Actions: []*pb.ModAction{},
 	})
 	is.NoErr(err)
@@ -47,9 +59,13 @@ func TestAuthenticateModNoAuth(t *testing.T) {
 
 	recreateDB()
 	us := userStore()
-	ms := &ModService{userStore: us}
 
-	_, err := authenticateMod(ctx, ms, &pb.ModActionsList{
+	pool, err := common.OpenTestingDB(pkg)
+	is.NoErr(err)
+	q := models.New(pool)
+	ms := &ModService{userStore: us, queries: q}
+
+	_, err = authenticateMod(ctx, ms, &pb.ModActionsList{
 		Actions: []*pb.ModAction{},
 	})
 	is.Equal(err, apiserver.Unauthenticated(errNotAuthorized.Error()))

diff --git a/pkg/stores/common/db.go b/pkg/stores/common/db.go
@@ -435,3 +435,7 @@ func checkRowsAffected(rowsAffected int, cfg *CommonDBConfig) error {
 	}
 	return nil
 }
+
+func ToPGTypeText(str string) pgtype.Text {
+	return pgtype.Text{Valid: true, String: str}
+}