From aecd00f3d8993e481ad2a3a3adad239bdf798978 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cau=C3=AA=20Marcondes?=
 <55978943+cauemarcondes@users.noreply.github.com>
Date: Wed, 22 Apr 2020 06:34:40 +0100
Subject: [PATCH] [APM]fixing custom link unit tests (#64045)

---
 .../Settings/CustomizeUI/CustomLink/index.test.tsx    | 11 +++++++----
 .../__test__/TransactionActionMenu.test.tsx           |  8 +++++---
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/x-pack/legacy/plugins/apm/public/components/app/Settings/CustomizeUI/CustomLink/index.test.tsx b/x-pack/legacy/plugins/apm/public/components/app/Settings/CustomizeUI/CustomLink/index.test.tsx
index 0c04b7cccbd23..e5c20b260e097 100644
--- a/x-pack/legacy/plugins/apm/public/components/app/Settings/CustomizeUI/CustomLink/index.test.tsx
+++ b/x-pack/legacy/plugins/apm/public/components/app/Settings/CustomizeUI/CustomLink/index.test.tsx
@@ -35,7 +35,7 @@ const data = [
 ];
 
 describe('CustomLink', () => {
-  let callApmApiSpy: Function;
+  let callApmApiSpy: jasmine.Spy;
   beforeAll(() => {
     callApmApiSpy = spyOn(apmApi, 'callApmApi').and.returnValue({});
   });
@@ -101,7 +101,7 @@ describe('CustomLink', () => {
       ]);
     });
 
-    it('checks if create custom link button is available and working', () => {
+    it('checks if create custom link button is available and working', async () => {
       const { queryByText, getByText } = render(
         <LicenseContext.Provider value={goldLicense}>
           <MockApmPluginContextWrapper>
@@ -113,6 +113,7 @@ describe('CustomLink', () => {
       act(() => {
         fireEvent.click(getByText('Create custom link'));
       });
+      await wait(() => expect(callApmApiSpy).toHaveBeenCalled());
       expect(queryByText('Create link')).toBeInTheDocument();
     });
   });
@@ -144,8 +145,10 @@ describe('CustomLink', () => {
       act(() => {
         fireEvent.click(component.getByText('Create custom link'));
       });
-      await wait(() => component.queryByText('Create link'));
-      expect(component.queryByText('Create link')).toBeInTheDocument();
+      await wait(() =>
+        expect(component.queryByText('Create link')).toBeInTheDocument()
+      );
+      await wait(() => expect(callApmApiSpy).toHaveBeenCalled());
       return component;
     };
 
diff --git a/x-pack/legacy/plugins/apm/public/components/shared/TransactionActionMenu/__test__/TransactionActionMenu.test.tsx b/x-pack/legacy/plugins/apm/public/components/shared/TransactionActionMenu/__test__/TransactionActionMenu.test.tsx
index ce42bd3e39ad1..8dc2076eab5b5 100644
--- a/x-pack/legacy/plugins/apm/public/components/shared/TransactionActionMenu/__test__/TransactionActionMenu.test.tsx
+++ b/x-pack/legacy/plugins/apm/public/components/shared/TransactionActionMenu/__test__/TransactionActionMenu.test.tsx
@@ -5,7 +5,7 @@
  */
 
 import React from 'react';
-import { render, fireEvent, act } from '@testing-library/react';
+import { render, fireEvent, act, wait } from '@testing-library/react';
 import { TransactionActionMenu } from '../TransactionActionMenu';
 import { Transaction } from '../../../../../../../../plugins/apm/typings/es_schemas/ui/transaction';
 import * as Transactions from './mockData';
@@ -143,8 +143,9 @@ describe('TransactionActionMenu component', () => {
   });
 
   describe('Custom links', () => {
+    let callApmApiSpy: jasmine.Spy;
     beforeAll(() => {
-      spyOn(apmApi, 'callApmApi').and.returnValue({});
+      callApmApiSpy = spyOn(apmApi, 'callApmApi').and.returnValue({});
     });
     afterAll(() => {
       jest.resetAllMocks();
@@ -257,7 +258,7 @@ describe('TransactionActionMenu component', () => {
       });
       expectTextsInDocument(component, ['Custom Links']);
     });
-    it('opens flyout with filters prefilled', () => {
+    it('opens flyout with filters prefilled', async () => {
       const license = new License({
         signature: 'test signature',
         license: {
@@ -287,6 +288,7 @@ describe('TransactionActionMenu component', () => {
         fireEvent.click(component.getByText('Create custom link'));
       });
       expectTextsInDocument(component, ['Create link']);
+      await wait(() => expect(callApmApiSpy).toHaveBeenCalled());
       const getFilterKeyValue = (key: string) => {
         return {
           [(component.getAllByText(key)[0] as HTMLOptionElement)