From 96b074de6104161c62e09b2b1a725316dbf1c5f3 Mon Sep 17 00:00:00 2001
From: Akshay Nair <phenax5@gmail.com>
Date: Sun, 1 Mar 2026 10:35:53 +0530
Subject: Add tests for mode state

---
 src/mode.c | 3 ++-
 src/mode.h | 6 ++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/mode.c b/src/mode.c
index b6fad97..6130372 100644
--- a/src/mode.c
+++ b/src/mode.c
@@ -34,6 +34,7 @@ void apply_button_state(enum ButtonState button_state) {
       new_minutes = minutes;
       new_hours = hours;
       current_digit_index = 0;
+      time_set_timeout_ticks = SET_MODE_TIMEOUT;
       current_mode = ModeSetTime;
       break;
     default:
@@ -49,7 +50,7 @@ void apply_button_state(enum ButtonState button_state) {
         return;
       }
     } else {
-      time_set_timeout_ticks = 10 * 1024;
+      time_set_timeout_ticks = SET_MODE_TIMEOUT;
     }
 
     switch (button_state) {
diff --git a/src/mode.h b/src/mode.h
index db65fc8..6bd5a35 100644
--- a/src/mode.h
+++ b/src/mode.h
@@ -6,6 +6,8 @@ enum Mode {
   ModeSetTime,
 };
 
+#define SET_MODE_TIMEOUT 1000
+
 enum ButtonState {
   ButtonIdle = 0,
   ButtonActive = 1,
@@ -19,4 +21,8 @@ extern volatile uint8_t new_hours;
 
 void apply_button_state(enum ButtonState button_state);
 
+// Just for tests
+extern volatile uint8_t current_digit_index;
+extern volatile uint16_t time_set_timeout_ticks;
+
 #endif
-- 
cgit v1.3.1