From c4e057eddf89b944609c496e563bc72b833f482f Mon Sep 17 00:00:00 2001
From: "grafana-delivery-bot[bot]"
 <132647405+grafana-delivery-bot[bot]@users.noreply.github.com>
Date: Tue, 16 Apr 2024 16:09:43 +0100
Subject: [PATCH] [v11.0.x] Loki: Fix test flake (#86353)

Loki: Fix test flake (#85356)

attempt to fix flake

(cherry picked from commit 0c78d80ef10b4e7fd52e06d613619370095537ff)

Co-authored-by: Galen Kistler <109082771+gtk-grafana@users.noreply.github.com>
---
 .../loki/components/LokiQueryField.test.tsx        | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/public/app/plugins/datasource/loki/components/LokiQueryField.test.tsx b/public/app/plugins/datasource/loki/components/LokiQueryField.test.tsx
index 5c304700ffb..17b2fd47e6f 100644
--- a/public/app/plugins/datasource/loki/components/LokiQueryField.test.tsx
+++ b/public/app/plugins/datasource/loki/components/LokiQueryField.test.tsx
@@ -6,6 +6,16 @@ import { dateTime } from '@grafana/data';
 import { createLokiDatasource } from '../__mocks__/datasource';
 
 import { LokiQueryField } from './LokiQueryField';
+import { Props as MonacoProps } from './monaco-query-field/MonacoQueryFieldProps';
+
+jest.mock('./monaco-query-field/MonacoQueryFieldLazy', () => {
+  const fakeQueryField = (props: MonacoProps) => {
+    return <input onBlur={(e) => props.onBlur(e.currentTarget.value)} data-testid={'dummy-code-input'} type={'text'} />;
+  };
+  return {
+    MonacoQueryFieldLazy: fakeQueryField,
+  };
+});
 
 type Props = ComponentProps<typeof LokiQueryField>;
 describe('LokiQueryField', () => {
@@ -34,7 +44,7 @@ describe('LokiQueryField', () => {
     const { rerender } = render(<LokiQueryField {...props} />);
 
     await waitFor(async () => {
-      expect(await screen.findByText('Loading...')).toBeInTheDocument();
+      expect(await screen.findByTestId('dummy-code-input')).toBeInTheDocument();
     });
 
     expect(props.datasource.languageProvider.fetchLabels).not.toHaveBeenCalled();
@@ -58,7 +68,7 @@ describe('LokiQueryField', () => {
     const { rerender } = render(<LokiQueryField {...props} />);
 
     await waitFor(async () => {
-      expect(await screen.findByText('Loading...')).toBeInTheDocument();
+      expect(await screen.findByTestId('dummy-code-input')).toBeInTheDocument();
     });
 
     expect(props.datasource.languageProvider.fetchLabels).not.toHaveBeenCalled();