@highstate/backend 0.18.0 → 0.20.0

package/src/orchestrator/operation.refresh.test.ts

@@ -18,6 +18,8 @@ describe("Operation - Refresh", () => {
       instanceStateService,
       projectModelService,
       unitExtraService,
+      entitySnapshotService,
+      unitOutputService,
       createUnit,
       createDeployedUnitState,
       createOperation,
@@ -59,6 +61,8 @@ describe("Operation - Refresh", () => {
         instanceStateService,
         projectModelService,
         unitExtraService,
+        entitySnapshotService,
+        unitOutputService,
         logger,
       )
 

package/src/orchestrator/operation.test-utils.ts

@@ -1,17 +1,20 @@
 import type { InstanceId, InstanceModel, VersionedName } from "@highstate/contract"
 import type { ArtifactService } from "../artifact"
 import type {
+  EntitySnapshotService,
   InstanceLockService,
   InstanceStateService,
   OperationService,
   ProjectModelService,
   SecretService,
   UnitExtraService,
+  UnitOutputService,
 } from "../business"
 import type { Operation } from "../database"
 import type { LibraryBackend } from "../library"
 import type {
   OperationType,
+  RawPulumiOutputs,
   RunnerBackend,
   UnitDestroyOptions,
   UnitOptions,
@@ -95,11 +98,9 @@ export type RunnerTestController = {
   emitError: (stateId: string, message: string) => void
   emitCompletion: (
     stateId: string,
-    update?: Omit<
-      Extract<UnitStateUpdate, { type: "completion" }>,
-      "type" | "unitId" | "operationType"
-    > & {
+    update?: {
       operationType?: OperationType
+      rawOutputs?: RawPulumiOutputs | null
     },
   ) => void
 }
@@ -115,6 +116,8 @@ export const operationTest = test.extend<{
   instanceStateService: MockedObject<InstanceStateService>
   projectModelService: MockedObject<ProjectModelService>
   unitExtraService: MockedObject<UnitExtraService>
+  entitySnapshotService: MockedObject<EntitySnapshotService>
+  unitOutputService: MockedObject<UnitOutputService>
 
   createMockLibrary: () => LibraryModel
   createUnit: (name: string, type?: VersionedName) => InstanceModel
@@ -198,18 +201,17 @@ export const operationTest = test.extend<{
         const operationType =
           update.operationType ?? lastOperationTypeByStateId.get(stateId) ?? "update"
 
+        const rawOutputs = update.rawOutputs ?? {
+          // Note: tests generally don't care about actual Pulumi outputs.
+          // Provide a non-empty object so business parsing can run if needed.
+          $test: { value: true },
+        }
+
         getQueue(stateId).push({
           type: "completion",
           unitId: stateId as unknown as InstanceId,
           operationType,
-          outputHash: update.outputHash ?? null,
-          statusFields: update.statusFields ?? null,
-          terminals: update.terminals ?? null,
-          pages: update.pages ?? null,
-          triggers: update.triggers ?? null,
-          secrets: update.secrets ?? null,
-          workers: update.workers ?? null,
-          exportedArtifactIds: update.exportedArtifactIds ?? null,
+          rawOutputs,
         })
       },
     }
@@ -360,6 +362,35 @@ export const operationTest = test.extend<{
     await use(unitExtraService)
   },
 
+  unitOutputService: async ({}, use) => {
+    const unitOutputService = vi.mockObject({
+      parseUnitOutputs: vi.fn().mockResolvedValue({
+        outputHash: null,
+        statusFields: null,
+        terminals: null,
+        pages: null,
+        triggers: null,
+        secrets: null,
+        workers: null,
+        exportedArtifactIds: null,
+        hasResourceHooks: false,
+        entitySnapshotError: null,
+        entitySnapshotPayload: null,
+      }),
+    } as unknown as UnitOutputService)
+
+    await use(unitOutputService)
+  },
+
+  entitySnapshotService: async ({}, use) => {
+    const entitySnapshotService = vi.mockObject({
+      reconstructLatestExportedOutputValues: vi.fn().mockResolvedValue(new Map()),
+      persistUnitEntitySnapshots: vi.fn().mockResolvedValue(undefined),
+    } as unknown as EntitySnapshotService)
+
+    await use(entitySnapshotService)
+  },
+
   createMockLibrary: async ({}, use) => {
     const createMockLibrary = (): LibraryModel => {
       const testEntity = defineEntity({
@@ -444,6 +475,7 @@ export const operationTest = test.extend<{
         status: "deployed",
         source: "resident",
         kind: instance.kind,
+        hasResourceHooks: false,
         parentId: null,
         parentInstanceId: instance.parentId ?? null,
         selfHash: null,
@@ -492,6 +524,7 @@ export const operationTest = test.extend<{
         options: {
           forceUpdateDependencies: false,
           ignoreDependencies: false,
+          ignoreChangedDependencies: false,
           forceUpdateChildren: false,
           destroyDependentInstances: true,
           invokeDestroyTriggers: true,
@@ -564,6 +597,7 @@ export const operationTest = test.extend<{
         libraryBackend,
         instanceStateService,
         projectModelService,
+        undefined,
         logger,
       )
     }
@@ -576,9 +610,12 @@ export const operationTest = test.extend<{
       const instanceByStateId = new Map<string, InstanceModel>(
         input.instances.map(instance => [instance.id, instance]),
       )
+      const lastStatusByStateId = new Map<string, InstanceOperationStatus>()
 
       instanceStateService.createOperationStates.mockImplementation(async (_projectId, tuples) => {
         return tuples.map(([opState, instancePatch]) => {
+          lastStatusByStateId.set(opState.stateId, opState.status)
+
           return {
             instanceId: opState.model.id,
             ...instancePatch,
@@ -602,7 +639,9 @@ export const operationTest = test.extend<{
           const status =
             typeof options.operationState?.status === "string"
               ? options.operationState.status
-              : "pending"
+              : (lastStatusByStateId.get(stateId) ?? "pending")
+
+          lastStatusByStateId.set(stateId, status)
 
           const instance = instanceByStateId.get(stateId) ?? input.instances[0]
           if (!instance) {