generated from VLADIMIR/template
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
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)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestScreenTeamsRejectsInvalidAccess(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
req *proto.GetScreenTeamsReq
|
|
}{
|
|
{"missing", nil},
|
|
{"missing token", &proto.GetScreenTeamsReq{ScreenID: "1"}},
|
|
{"bad ID", &proto.GetScreenTeamsReq{ScreenID: "bad", AccessToken: "token"}},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
_, err := (&server{}).GetScreenTeams(context.Background(), tt.req)
|
|
if status.Code(err) != codes.InvalidArgument {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|