Fix mock.

Koenkk · Jun 15, 2024 · e022902 · e022902
1 parent dde550e
commit e022902
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/availability.test.js b/test/availability.test.js
@@ -348,7 +348,7 @@ describe('Availability', () => {
         const availability = controller.extensions.find((extension) => extension instanceof Availability);
         const publishAvailabilitySpy = jest.spyOn(availability, 'publishAvailability');
 
-        devices.bulb_color.ping = jest.fn().mockImplementation(() => new Promise((resolve) => setTimeout(resolve, 1000)));
+        devices.bulb_color.ping.mockImplementationOnce(() => new Promise((resolve) => setTimeout(resolve, 1000)));
         availability.addToPingQueue(devices.bulb_color);
         availability.addToPingQueue(devices.bulb_color_2);
 
@@ -358,6 +358,7 @@ describe('Availability', () => {
         expect(availability.pingQueue).toEqual([]);
         // Validate the stop-interrupt implicitly by checking that it prevents further function invocations
         expect(publishAvailabilitySpy).not.toHaveBeenCalled();
+        devices.bulb_color.ping = jest.fn();// ensure reset
     });
 
     it('Should prevent instance restart', async () => {