pass workingState into formatter.{before,after}All()

public-transport · Oct 31, 2024 · 26b0f6b · 26b0f6b
1 parent 5321ae0
commit 26b0f6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -224,7 +224,7 @@ LANGUAGE sql;
 		if ('string' === typeof beforeAll && beforeAll) {
 			yield beforeAll
 		} else if ('function' === typeof beforeAll) {
-			yield beforeAll(opt)
+			yield beforeAll(opt, workingState)
 		}
 
 		if (task.file) {
@@ -249,7 +249,7 @@ LANGUAGE sql;
 		if ('string' === typeof afterAll && afterAll) {
 			yield afterAll + ';\n'
 		} else if ('function' === typeof afterAll) {
-			yield afterAll(opt) + ';\n'
+			yield afterAll(opt, workingState) + ';\n'
 		}
 	}