up

internal/app/screens_test.go

@@ -0,0 +1,28 @@
+package app
+
+import (
+	"context"
+	"testing"
+
+	"evening_detective_server/internal/modules/processor_jwt"
+	"evening_detective_server/proto"
+
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func TestSetScreenModeRejectsRemovedModes(t *testing.T) {
+	ctx := processor_jwt.ContextWithClaims(context.Background(), &processor_jwt.JWTClaims{
+		Roles: []string{"organizer"},
+	})
+	s := &server{}
+	for _, mode := range []int32{3, 5} {
+		_, err := s.SetScreenMode(ctx, &proto.SetScreenModeReq{
+			GameID: 1,
+			Mode:   proto.ScreenMode(mode),
+		})
+		if status.Code(err) != codes.InvalidArgument {
+			t.Errorf("SetScreenMode(%d) = %v, want InvalidArgument", mode, err)
+		}
+	}
+}