From c14503597c86a3b3294bc57e40fb6938d6300a19 Mon Sep 17 00:00:00 2001
From: Pavel Zavora <pavel.zavora@bonitoo.io>
Date: Wed, 5 May 2021 07:50:45 +0200
Subject: [PATCH] fix(ui): test EDIT_CELL_QUERY_STATUS action

---
 ui/test/dashboards/reducers/ui.test.ts | 34 ++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/ui/test/dashboards/reducers/ui.test.ts b/ui/test/dashboards/reducers/ui.test.ts
index 5b3acd39d..4d44add7b 100644
--- a/ui/test/dashboards/reducers/ui.test.ts
+++ b/ui/test/dashboards/reducers/ui.test.ts
@@ -14,6 +14,8 @@ import {
   templateVariableLocalSelected,
   setActiveCell,
   updateTemplates,
+  editCellQueryStatus,
+  loadDashboard,
 } from 'src/dashboards/actions'
 
 let state
@@ -212,4 +214,36 @@ describe('DataExplorer.Reducers.UI', () => {
       )
     })
   })
+  describe('EDIT_CELL_QUERY_STATUS', () => {
+    it('can edit cell query status', () => {
+      let actual = reducer(
+        initialState,
+        editCellQueryStatus('query-id1', {success: 'yes'})
+      )
+      expect(actual.cellQueryStatuses).toEqual({'query-id1': {success: 'yes'}})
+      actual = reducer(actual, editCellQueryStatus('query-id2', {error: 'no'}))
+      expect(actual.cellQueryStatuses).toEqual({
+        'query-id1': {success: 'yes'},
+        'query-id2': {error: 'no'},
+      })
+    })
+    it('resets query status on dashboards load', () => {
+      let actual = reducer(
+        initialState,
+        editCellQueryStatus('query-id1', {success: 'yes'})
+      )
+      expect(actual.cellQueryStatuses).toEqual({'query-id1': {success: 'yes'}})
+      actual = reducer(actual, loadDashboards(dashboards, d1.id))
+      expect(actual.cellQueryStatuses).toEqual({})
+    })
+    it('resets query status on dashboard load', () => {
+      let actual = reducer(
+        initialState,
+        editCellQueryStatus('query-id1', {success: 'yes'})
+      )
+      expect(actual.cellQueryStatuses).toEqual({'query-id1': {success: 'yes'}})
+      actual = reducer(actual, loadDashboard(d1))
+      expect(actual.cellQueryStatuses).toEqual({})
+    })
+  })
 })