From f9c602e62cb253d728e519404a39543121d19282 Mon Sep 17 00:00:00 2001
From: Prathamesh Mutkure <pmutkure009@gmail.com>
Date: Thu, 23 Nov 2023 15:58:08 +0530
Subject: [PATCH] Added Test for Layout Manager

Signed-off-by: Prathamesh Mutkure <pmutkure009@gmail.com>
---
 .../src/components/DependencyGraph/DAG.test.js         | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/packages/jaeger-ui/src/components/DependencyGraph/DAG.test.js b/packages/jaeger-ui/src/components/DependencyGraph/DAG.test.js
index b3c71c01d7..45e0b8ffed 100644
--- a/packages/jaeger-ui/src/components/DependencyGraph/DAG.test.js
+++ b/packages/jaeger-ui/src/components/DependencyGraph/DAG.test.js
@@ -15,6 +15,7 @@
 import React from 'react';
 import ShallowRenderer from 'react-test-renderer/shallow';
 import { render, screen } from '@testing-library/react';
+import { shallow } from 'enzyme';
 import DAG, { renderNode } from './DAG';
 
 // mock canvas API (we don't care about canvas results)
@@ -167,3 +168,12 @@ describe('renderNode', () => {
     expect(element.textContent).toBe('');
   });
 });
+
+describe('clean up', () => {
+  it('stops LayoutManager before unmounting', () => {
+    const wrapper = shallow(<DAG serviceCalls={[]} />);
+    const stopAndReleaseSpy = jest.spyOn(wrapper.instance().layoutManager, 'stopAndRelease');
+    wrapper.unmount();
+    expect(stopAndReleaseSpy).toHaveBeenCalledTimes(1);
+  });
+});