tests

src/services/consumption.service.spec.ts

@@ -581,4 +581,32 @@ describe('Consumption service', () => {
       expect(result.length).toEqual(4)
     })
   })
+
+  describe('getFluidsWithDataForTimePeriod', () => {
+    const allFluids = [FluidType.ELECTRICITY, FluidType.WATER, FluidType.GAS]
+    const timePeriods: TimePeriod = {
+      startDate: DateTime.local(2023, 5, 1),
+      endDate: DateTime.local(2023, 6, 1),
+    }
+    it('should return 2 fluids', async () => {
+      mockFetchFluidData.mockResolvedValueOnce(null)
+      mockFetchFluidData.mockResolvedValueOnce(mockFetchDataActual)
+      mockFetchFluidData.mockResolvedValueOnce(mockFetchDataActual)
+      const fluidsWithData =
+        await consumptionDataManager.getFluidsWithDataForTimePeriod(
+          allFluids,
+          timePeriods
+        )
+      expect(fluidsWithData).toEqual([FluidType.WATER, FluidType.GAS])
+    })
+    it('should return no fluids', async () => {
+      mockFetchFluidData.mockResolvedValue(null)
+      const fluidsWithData =
+        await consumptionDataManager.getFluidsWithDataForTimePeriod(
+          allFluids,
+          timePeriods
+        )
+      expect(fluidsWithData).toEqual([])
+    })
+  })
 })