up

internal/app/screens_test.go

@@ -26,3 +26,22 @@ func TestSetScreenModeRejectsRemovedModes(t *testing.T) {
 		}
 	}
 }
+
+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)
+			}
+		})
+	}
+}