native macOS codings agent orchestrator prowl.onev.cat
Something went wrong. Try again.
3.0 kB · 79 lines
Swift
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980import Testing@testable import ProwlCLIShared
struct WorkflowControlCursorTests { @Test func nestedLoopsAndContinueRetainOnlyExplicitState() throws { let definition = try #require(WorkflowDocumentParser.parse(""" schema: prowl.workflow/v1 id: count name: Count state: count: {type: integer, initial: 0} steps: - id: loop while: state.count < 3 steps: - id: increment set: {count: state.count + 1} - id: skip if: state.count == 2 then: [{id: next, continue: true}] - id: observe notify: '{{ state.count }}' - id: end notify: done """).definition) var cursor = try WorkflowControlCursor(definition: definition) var observed: [String] = [] for _ in 0..<20 { switch try cursor.next(values: [:], budget: 64) { case .step(let step): if case .notify(let text) = step.action { observed.append(try WorkflowExpression.renderText(text, values: cursor.values(over: [:]))) } cursor.complete() case .finished: break case .yielded: continue } if cursor.isFinished { break } } #expect(observed == ["1", "3", "done"]) }
@Test func loopRechecksUseTheLoopPositionAndRetainedState() throws { for condition in [ "context.step.iteration < 2", "context.step.id == 'loop' && state.count < 2", ] { let definition = try #require(WorkflowDocumentParser.parse(""" schema: prowl.workflow/v1 id: position name: Position state: {count: {type: integer, initial: 0}} steps: - id: loop while: "\(condition)" max_iterations: 2 steps: - id: tick set: {count: state.count + 1} """).definition) var cursor = try WorkflowControlCursor(definition: definition) let context: [String: WorkflowJSONValue] = ["context": .object([ "step": .object(["id": .string("previous"), "iteration": .integer(99)])])] #expect(try cursor.next(values: context) == .finished) #expect(cursor.state.values["count"] == .integer(2)) } }
@Test func unlimitedPureLoopYieldsAndCapDoesNotCompleteSuccessfully() throws { let step = WorkflowStepDefinition(id: "loop", action: .control(.loop(condition: "true", maximum: nil, steps: [.init(id: "tick", action: .control(.set([:])))]))) var cursor = try WorkflowControlCursor(definition: .init(id: "test", name: "Test", steps: [step])) #expect(try cursor.next(values: [:], budget: 64) == .yielded) let capped = WorkflowStepDefinition(id: "loop", action: .control(.loop(condition: "true", maximum: 2, steps: [.init(id: "tick", action: .control(.set([:])))]))) var limited = try WorkflowControlCursor(definition: .init(id: "test", name: "Test", steps: [capped])) #expect(throws: (any Error).self) { try limited.next(values: [:], budget: 64) } }}