From 7fdd1df7be1ba611eef99e1dd997555ed0d31b3a Mon Sep 17 00:00:00 2001
From: Nerivec <62446222+Nerivec@users.noreply.github.com>
Date: Sat, 15 Jun 2024 04:22:28 +0200
Subject: [PATCH] Fix mock.

---
 test/availability.test.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/availability.test.js b/test/availability.test.js
index 2a23126365..6a7facb379 100644
--- 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 () => {