Add counter prop

atomex-protocol · Sep 6, 2022 · 21efdb8 · 21efdb8
1 parent 51eaf5c
commit 21efdb8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/core/timeoutScheduler.ts b/src/core/timeoutScheduler.ts
@@ -1,15 +1,23 @@
 import type { Disposable } from '../common';
 
 export class TimeoutScheduler implements Disposable {
-  private counter = 0;
   private counterExpirationWatcherId: ReturnType<typeof setTimeout> | undefined;
   private actionWatcherId: ReturnType<typeof setTimeout> | undefined;
+  private _counter = 0;
 
   constructor(
     private readonly timeouts: number[],
     private readonly counterExpirationMs?: number
   ) { }
 
+  get counter() {
+    return this._counter;
+  }
+
+  private set counter(value: number) {
+    this._counter = value;
+  }
+
   async dispose(): Promise<void> {
     if (this.counterExpirationWatcherId)
       clearInterval(this.counterExpirationWatcherId);

diff --git a/tests/core/timeoutScheduler.test.ts b/tests/core/timeoutScheduler.test.ts
@@ -14,26 +14,32 @@ describe('TimeoutScheduler', () => {
 
     scheduler = new TimeoutScheduler([500, 700]);
 
+    expect(scheduler.counter).toEqual(0);
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(1);
+    expect(scheduler.counter).toEqual(1);
 
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(1);
     await wait(200);
     expect(counter).toEqual(2);
+    expect(scheduler.counter).toEqual(2);
 
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(2);
     await wait(200);
     expect(counter).toEqual(3);
+    expect(scheduler.counter).toEqual(3);
 
     scheduler.resetCounter();
+    expect(scheduler.counter).toEqual(0);
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(4);
+    expect(scheduler.counter).toEqual(1);
   });
 
   test('runs scheduled actions and clears timer automatically', async () => {
@@ -42,26 +48,32 @@ describe('TimeoutScheduler', () => {
 
     scheduler = new TimeoutScheduler([500, 700], 1000);
 
+    expect(scheduler.counter).toEqual(0);
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(1);
+    expect(scheduler.counter).toEqual(1);
 
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(1);
     await wait(200);
     expect(counter).toEqual(2);
+    expect(scheduler.counter).toEqual(2);
 
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(2);
     await wait(200);
     expect(counter).toEqual(3);
+    expect(scheduler.counter).toEqual(3);
 
     //Wait internal counter expiration
     await wait(1000);
+    expect(scheduler.counter).toEqual(0);
     scheduler.setTimeout(action);
     await wait(500);
     expect(counter).toEqual(4);
+    expect(scheduler.counter).toEqual(1);
   });
 });