Fix test

avajs · Nov 9, 2023 · ec70158 · ec70158
1 parent fdca2ed
commit ec70158
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/lib/api-event-iterator.js b/lib/api-event-iterator.js
@@ -1,10 +1,14 @@
-import {on} from 'node:events';
-
 export async function * asyncEventIteratorFromApi(api) {
-	for await (const [plan] of on(api, 'run')) {
-
-		for await (const [stateChange] of on(plan.status, 'stateChange')) {
+	for await (const plan of api.events('run')) {
+		for await (const stateChange of plan.status.events('stateChange')) {
 			yield stateChange;
+
+			if (stateChange.type === 'end' || stateChange.type === 'interrupt') {
+				break;
+			}
 		}
+
+		// todo: support multiple runs (watch mode)
+		break;
 	}
 }