Merge pull request #2823 from PrefectHQ/dylan/bugfix/make-sure-all-ty…

…pes-are-updated BugFix: ensure all types are updated for workpoolworker
PrefectHQ · Nov 12, 2024 · 4dfa3f8 · 4dfa3f8
2 parents ee103d5 + 9622411
commit 4dfa3f8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/mocks/worker.ts b/src/mocks/worker.ts
@@ -11,7 +11,7 @@ export const randomWorker: MockFunction<WorkPoolWorker, [Partial<WorkPoolWorker>
     lastHeartbeatTime: this.create('date'),
     status: this.create('workerStatus'),
     clientVersion: this.create('string'),
-    metadata: { 'integrations': [this.create('string'), this.create('string')] },
+    metadata: { integrations: [{ name: this.create('string'), version: this.create('string') }], 'foo': 'bar' },
     heartbeatIntervalSeconds: this.create('number'),
     ...overrides,
   })

diff --git a/src/models/WorkPoolWorker.ts b/src/models/WorkPoolWorker.ts
@@ -1,7 +1,7 @@
 import { WorkPoolWorkerStatus } from '@/models/WorkPoolWorkerStatus'
 
-type Integration = { name: string, version: string }
-type Metadata = Record<string, unknown> & { integrations?: Integration[] }
+export type Integration = { name: string, version: string }
+export type Metadata = { integrations?: Integration[] } & Record<string, unknown>
 
 export interface IWorkPoolWorker {
   readonly id: string,
@@ -27,7 +27,7 @@ export class WorkPoolWorker implements IWorkPoolWorker {
   public status: WorkPoolWorkerStatus
   public heartbeatIntervalSeconds: number
   public clientVersion: string | null
-  public metadata: Record<string, unknown> | null
+  public metadata: Metadata | null
 
   public constructor(workPoolWorker: IWorkPoolWorker) {
     this.id = workPoolWorker.id