From 3a1a97d84df3c3e99a00d16884126672c0e1936e Mon Sep 17 00:00:00 2001
From: Isabella Siu <isabella.siu@grafana.com>
Date: Fri, 13 Sep 2024 17:54:16 -0400
Subject: [PATCH] handle feature toggles properly

---
 src/opensearchDatasource.test.ts | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/src/opensearchDatasource.test.ts b/src/opensearchDatasource.test.ts
index 622c3c88..21528fde 100644
--- a/src/opensearchDatasource.test.ts
+++ b/src/opensearchDatasource.test.ts
@@ -1370,9 +1370,16 @@ describe('OpenSearchDatasource', function (this: any) {
   });
 
   describe('getOpenSearchVersion backend flow', () => {
-    it('should return OpenSearch version', async () => {
+    beforeEach(() => {
       // @ts-ignore-next-line
       config.featureToggles.openSearchBackendFlowEnabled = true;
+    });
+
+    afterEach(() => {
+      // @ts-ignore-next-line
+      config.featureToggles.openSearchBackendFlowEnabled = false;
+    });
+    it('should return OpenSearch version', async () => {
       const mockResource = jest.fn().mockResolvedValue({
         version: { distribution: 'opensearch', number: '2.6.0' },
       });
@@ -1387,8 +1394,6 @@ describe('OpenSearchDatasource', function (this: any) {
     });
 
     it('should return ElasticSearch version', async () => {
-      // @ts-ignore-next-line
-      config.featureToggles.openSearchBackendFlowEnabled = true;
       ctx.ds.getResource = jest.fn().mockResolvedValue({
         version: { number: '7.6.0' },
       });
@@ -1400,8 +1405,6 @@ describe('OpenSearchDatasource', function (this: any) {
     });
 
     it('should error for invalid version', async () => {
-      // @ts-ignore-next-line
-      config.featureToggles.openSearchBackendFlowEnabled = true;
       ctx.ds.getResource = jest.fn().mockResolvedValue({
         version: { number: '7.11.1' },
       });
@@ -1411,8 +1414,6 @@ describe('OpenSearchDatasource', function (this: any) {
     });
 
     it('should return ElasticSearch for ElasticSearch 7.10.2 without tagline', async () => {
-      // @ts-ignore-next-line
-      config.featureToggles.openSearchBackendFlowEnabled = true;
       ctx.ds.getResource = jest.fn().mockResolvedValue({
         version: { number: '7.10.2' },
       });
@@ -1423,8 +1424,6 @@ describe('OpenSearchDatasource', function (this: any) {
     });
 
     it('should return OpenSearch for ElasticSearch 7.10.2 with tagline', async () => {
-      // @ts-ignore-next-line
-      config.featureToggles.openSearchBackendFlowEnabled = true;
       ctx.ds.getResource = jest.fn().mockResolvedValue({
         version: { number: '7.10.2' },
         tagline: 'The OpenSearch Project: https://opensearch.org/',