use PartialMessage and class instance

Author: turbocrime

packages/services/src/ctx/authorize.ts

@@ -1,10 +1,10 @@
-import { PlainMessage } from '@bufbuild/protobuf';
+import { PartialMessage } from '@bufbuild/protobuf';
 import { createContextKey } from '@connectrpc/connect';
 import {
   AuthorizeRequest,
   AuthorizeResponse,
 } from '@penumbra-zone/protobuf/penumbra/custody/v1/custody_pb';
 
 export const authorizeCtx = createContextKey<
-  (req: PlainMessage<AuthorizeRequest>) => Promise<PlainMessage<AuthorizeResponse>>
+  (req: PartialMessage<AuthorizeRequest>) => Promise<AuthorizeResponse>
 >(undefined as never);

packages/services/src/custody-service/authorize.test.ts

@@ -1,4 +1,4 @@
-import { PlainMessage } from '@bufbuild/protobuf';
+import { PartialMessage } from '@bufbuild/protobuf';
 import { createContextValues, createHandlerContext, HandlerContext } from '@connectrpc/connect';
 import { CustodyService } from '@penumbra-zone/protobuf';
 import { Metadata } from '@penumbra-zone/protobuf/penumbra/core/asset/v1/asset_pb';
@@ -23,7 +23,7 @@ describe('Authorize request handler', () => {
   let req: AuthorizeRequest;
 
   const mockFullViewingKeyCtx = vi.fn(() => Promise.resolve(testFullViewingKey));
-  const mockAuthorizeCtx = vi.fn(({ plan }: PlainMessage<AuthorizeRequest>) =>
+  const mockAuthorizeCtx = vi.fn(({ plan }: PartialMessage<AuthorizeRequest>) =>
     Promise.resolve(
       new AuthorizeResponse({ data: authorizePlan(testSpendKey, new TransactionPlan(plan)) }),
     ),