fix

AztecProtocol · Dec 13, 2024 · 9e17522 · 9e17522
1 parent 1493142
commit 9e17522
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/yarn-project/circuit-types/src/interfaces/p2p.test.ts b/yarn-project/circuit-types/src/interfaces/p2p.test.ts
@@ -8,13 +8,13 @@ import { type P2PApi, P2PApiSchema, type PeerInfo } from './p2p.js';
 
 describe('P2PApiSchema', () => {
   let handler: MockP2P;
-  let context: JsonRpcTestContext<P2PApi<P2PClientType.Full>>;
+  let context: JsonRpcTestContext<P2PApi>;
 
   const tested = new Set<string>();
 
   beforeEach(async () => {
     handler = new MockP2P();
-    context = await createJsonRpcTestSetup<P2PApi<P2PClientType.Full>>(handler, P2PApiSchema);
+    context = await createJsonRpcTestSetup<P2PApi>(handler, P2PApiSchema);
   });
 
   afterEach(() => {
@@ -66,7 +66,7 @@ const peers: PeerInfo[] = [
   { status: 'cached', id: 'id', addresses: ['address'], enr: 'enr', dialAttempts: 1 },
 ];
 
-class MockP2P implements P2PApi<P2PClientType.Full> {
+class MockP2P implements P2PApi {
   getAttestationsForSlot(slot: bigint, proposalId?: string | undefined): Promise<BlockAttestation[]> {
     expect(slot).toEqual(1n);
     expect(proposalId).toEqual('proposalId');

diff --git a/yarn-project/circuit-types/src/interfaces/p2p.ts b/yarn-project/circuit-types/src/interfaces/p2p.ts
@@ -62,11 +62,11 @@ export interface P2PClient extends P2PApiWithoutAttestations {
   getAttestationsForSlot(slot: bigint, proposalId?: string): Promise<BlockAttestation[]>;
 }
 
-export type P2PApi<T extends P2PClientType> = T extends P2PClientType.Full
+export type P2PApi<T extends P2PClientType = P2PClientType.Full> = T extends P2PClientType.Full
   ? P2PClient & P2PApiWithoutAttestations
   : P2PApiWithoutAttestations;
 
-export const P2PApiSchema: ApiSchemaFor<P2PApi<P2PClientType.Full>> = {
+export const P2PApiSchema: ApiSchemaFor<P2PApi> = {
   getAttestationsForSlot: z
     .function()
     .args(schemas.BigInt, optional(z.string()))