feat(ng-div): add AngularContext.isRunning()

simontonsoftware · Aug 12, 2023 · ce1a8bb · ce1a8bb
1 parent 81b4822
commit ce1a8bb
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/projects/ng-dev/src/lib/angular-context/angular-context.spec.ts b/projects/ng-dev/src/lib/angular-context/angular-context.spec.ts
@@ -230,6 +230,17 @@ describe('AngularContext', () => {
     });
   });
 
+  describe('.isRunning()', () => {
+    it('works', () => {
+      const ctx = new AngularContext();
+      expect(ctx.isRunning()).toBe(false);
+      ctx.run(() => {
+        expect(ctx.isRunning()).toBe(true);
+      });
+      expect(ctx.isRunning()).toBe(false);
+    });
+  });
+
   describe('.inject()', () => {
     it('fetches from the root injector', () => {
       const ctx = new AngularContext();

diff --git a/projects/ng-dev/src/lib/angular-context/angular-context.ts b/projects/ng-dev/src/lib/angular-context/angular-context.ts
@@ -101,6 +101,7 @@ export class AngularContext {
    */
   startTime = new Date();
 
+  #isRunning = false;
   #loader = FakeAsyncHarnessEnvironment.documentRootLoader(this);
 
   /**
@@ -140,6 +141,7 @@ export class AngularContext {
    */
   run(test: () => Promise<void> | void): void {
     try {
+      this.#isRunning = true;
       this.#runWithMockedTime(() => {
         this.init();
         try {
@@ -153,9 +155,17 @@ export class AngularContext {
       });
     } finally {
       AngularContext.#current = undefined;
+      this.#isRunning = false;
     }
   }
 
+  /**
+   * Returns whether this context is currently executing the {@linkcode #run} callback.
+   */
+  isRunning(): boolean {
+    return this.#isRunning;
+  }
+
   /**
    * Gets a service or other injectable from the root injector.
    *
@@ -207,9 +217,7 @@ export class AngularContext {
    * @param unit The unit of time `amount` represents. Accepts anything described in `@s-libs/s-core`'s [TimeUnit]{@linkcode https://simontonsoftware.github.io/s-js-utils/typedoc/enums/timeunit.html} enum.
    */
   tick(amount = 0, unit = 'ms'): void {
-    if (
-      isUndefined((window as any).Zone.current.get('FakeAsyncTestZoneSpec'))
-    ) {
+    if (!this.#isRunning) {
       throw new Error(
         '.tick() only works inside the .run() callback (because it needs to be in a fakeAsync zone)',
       );