From f3c06069ad473d45c63d24ea47e0889ff74df43a Mon Sep 17 00:00:00 2001
From: Bastien Dumont <bdumont@grandlyon.com>
Date: Thu, 21 Mar 2024 17:06:50 +0100
Subject: [PATCH] fix tests

---
 .../Consumption/ConsumptionView.spec.tsx        | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/components/Consumption/ConsumptionView.spec.tsx b/src/components/Consumption/ConsumptionView.spec.tsx
index c65440192..722bc1e09 100644
--- a/src/components/Consumption/ConsumptionView.spec.tsx
+++ b/src/components/Consumption/ConsumptionView.spec.tsx
@@ -1,4 +1,4 @@
-import { render, screen } from '@testing-library/react'
+import { render, screen, waitFor } from '@testing-library/react'
 import { FluidState, FluidType, TimeStep } from 'enums'
 import React from 'react'
 import { Provider } from 'react-redux'
@@ -122,7 +122,7 @@ describe('ConsumptionView component', () => {
     ).toBeTruthy()
   })
 
-  it('should render mutlifluid consumption if at least one fluid is connected', () => {
+  it('should render mutlifluid consumption if at least one fluid is connected without konnectorViewerCard', async () => {
     const store = createMockEcolyoStore({
       chart: mockChartStateShowOffline,
       global: {
@@ -136,12 +136,16 @@ describe('ConsumptionView component', () => {
         <ConsumptionView fluidType={FluidType.MULTIFLUID} />
       </Provider>
     )
+    await waitFor(() => null, { container })
     expect(
-      container.getElementsByClassName('consumptionview-content')[0]
+      container.getElementsByTagName('mock-consumptionDetails').item(0)
     ).toBeInTheDocument()
+    expect(
+      container.getElementsByTagName('mock-konnectorViewerCard').item(0)
+    ).not.toBeInTheDocument()
   })
 
-  it('should render Electricity when elec is connected', () => {
+  it('should render Electricity when elec is connected with konnectorViewCard', async () => {
     const store = createMockEcolyoStore({
       chart: mockChartStateShowOffline,
       global: {
@@ -155,11 +159,12 @@ describe('ConsumptionView component', () => {
         <ConsumptionView fluidType={FluidType.ELECTRICITY} />
       </Provider>
     )
+    await waitFor(() => null, { container })
     expect(
-      container.getElementsByClassName('consumptionview-content').item(0)
+      container.getElementsByTagName('mock-consumptionDetails').item(0)
     ).toBeInTheDocument()
     expect(
-      container.getElementsByTagName('mock-consumptionDetails').item(0)
+      container.getElementsByTagName('mock-konnectorViewerCard').item(0)
     ).toBeInTheDocument()
   })
 
-- 
GitLab