Add tests for mode state

2 files changed, 8 insertions, 1 deletions

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