diff --git a/src/services/consumption.service.spec.ts b/src/services/consumption.service.spec.ts index 4dadcfe21a64d157979ded1353c0f54c0113226f..b848e4c06ea52f1be608af23042543bfff83ea44 100644 --- a/src/services/consumption.service.spec.ts +++ b/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([]) + }) + }) })