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

---
 test/button_test.c | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

(limited to 'test/button_test.c')

diff --git a/test/button_test.c b/test/button_test.c
index 3c48e43..bef557c 100644
--- a/test/button_test.c
+++ b/test/button_test.c
@@ -1,8 +1,17 @@
 #include "button.h"
 #include "mode.h"
+#include "stubbed/testutils.h"
 #include <assert.h>
 #include <avr/io.h>
 
+void before_each() {
+  PINA = (1 << PA0); // Active on low
+  button_tick_count = 0;
+  button_inactive_for_ticks = 0;
+}
+
+void after_each() {}
+
 void test_button_press() {
   assert(check_button_state() == ButtonIdle);
 
@@ -51,13 +60,6 @@ void test_button_inactive_after_press() {
   assert(check_button_state() == ButtonActive);
 }
 
-void before_each() {
-  PINA = (1 << PA0); // Active on low
-  button_tick_count = 0;
-  button_inactive_for_ticks = 0;
-}
-#define TEST before_each();
-
 int main() {
   TEST test_button_press();
   TEST test_button_long_press();
-- 
cgit v1.3.1