fix: add tests

Author: myandrienko

src/components/Dialog/__tests__/DialogsManager.test.js

@@ -8,13 +8,15 @@ describe('DialogManager', () => {
     const dialogManager = new DialogManager({ id });
     expect(dialogManager.id).toBe(id);
   });
+
   it('initiates with default options', () => {
     const mockedId = '12345';
     const spy = jest.spyOn(Date.prototype, 'getTime').mockReturnValueOnce(mockedId);
     const dialogManager = new DialogManager();
     expect(dialogManager.id).toBe(mockedId);
     spy.mockRestore();
   });
+
   it('creates a new closed dialog', () => {
     const dialogManager = new DialogManager();
     expect(Object.keys(dialogManager.state.getLatestValue().dialogsById)).toHaveLength(0);
@@ -142,4 +144,33 @@ describe('DialogManager', () => {
     expect(dialogManager.openDialogCount).toBe(1);
     expect(Object.keys(dialogManager.state.getLatestValue().dialogsById)).toHaveLength(1);
   });
+
+  it('marks dialog for removal', () => {
+    jest.useFakeTimers();
+
+    const dialogManager = new DialogManager();
+    dialogManager.getOrCreate({ id: dialogId });
+    dialogManager.open({ id: dialogId });
+    dialogManager.markForRemoval(dialogId);
+
+    jest.runAllTimers();
+
+    expect(dialogManager.openDialogCount).toBe(0);
+    expect(Object.keys(dialogManager.state.getLatestValue().dialogsById)).toHaveLength(0);
+  });
+
+  it('cancels dialog removal if it is referenced again quickly', () => {
+    jest.useFakeTimers();
+
+    const dialogManager = new DialogManager();
+    dialogManager.getOrCreate({ id: dialogId });
+    dialogManager.open({ id: dialogId });
+    dialogManager.markForRemoval(dialogId);
+    dialogManager.getOrCreate({ id: dialogId });
+
+    jest.runAllTimers();
+
+    expect(dialogManager.openDialogCount).toBe(1);
+    expect(Object.keys(dialogManager.state.getLatestValue().dialogsById)).toHaveLength(1);
+  });
 });