From c8f7f394d4408b1b35e0071fcbadbaf21b8a58ed Mon Sep 17 00:00:00 2001
From: Aleksandar Petrov <8142643+aleks-p@users.noreply.github.com>
Date: Wed, 29 Nov 2023 15:41:28 -0400
Subject: [PATCH] Fix frontend tests

---
 .../QueryEditor/QueryEditor.test.tsx                          | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx b/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
index 75c0f98e564..643089f5f0e 100644
--- a/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
+++ b/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
@@ -111,6 +111,10 @@ function setupDs() {
     },
   ] as ProfileTypeMessage[]);
 
+  ds.getLabelNames = jest.fn().mockResolvedValue([
+      'label_one'
+  ]);
+
   return ds;
 }