From 5d683c635f5afd9c8f7078c8778e7e3a831a283f Mon Sep 17 00:00:00 2001
From: Yann Locatelli <yann.locatelli2@gmail.com>
Date: Wed, 24 Jul 2024 15:33:13 +0200
Subject: [PATCH] Update CoreIMU_test.cpp

---
 drivers/CoreIMU/tests/CoreIMU_test.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/drivers/CoreIMU/tests/CoreIMU_test.cpp b/drivers/CoreIMU/tests/CoreIMU_test.cpp
index 4f1a3a643..cb2ba2dbc 100644
--- a/drivers/CoreIMU/tests/CoreIMU_test.cpp
+++ b/drivers/CoreIMU/tests/CoreIMU_test.cpp
@@ -152,6 +152,14 @@ TEST_F(CoreIMUTest, enableOnWakeUpInterrupt)
 	coreimu.enableOnWakeUpInterrupt();
 }
 
+TEST_F(CoreIMUTest, disableOnWakeUpInterrupt)
+{
+	EXPECT_CALL(mocki2c, write).Times(AtLeast(1));
+	EXPECT_CALL(mocki2c, read).Times(AtLeast(1));
+
+	coreimu.disableOnWakeUpInterrupt();
+}
+
 TEST_F(CoreIMUTest, switchCallbacks)
 {
 	auto mock_data_available_callback = MockFunction<void(const leka::interface::IMU::SensorData &data)> {};
@@ -190,4 +198,13 @@ TEST_F(CoreIMUTest, switchCallbacks)
 		on_rise_callback = spy_InterruptIn_getRiseCallback();
 		on_rise_callback();
 	}
+
+	{
+		// Disable Wake Up
+		EXPECT_CALL(mock_data_available_callback, Call).Times(0);
+		EXPECT_CALL(mock_wake_up_callback, Call).Times(0);
+		coreimu.disableOnWakeUpInterrupt();
+		on_rise_callback = spy_InterruptIn_getRiseCallback();
+		on_rise_callback();
+	}
 }