From e105c8bb68fa630159363e3c206ffe5535ea8f92 Mon Sep 17 00:00:00 2001
From: Daniel Thompson <daniel@redfelineninja.org.uk>
Date: Sat, 14 Nov 2020 20:36:41 +0000
Subject: apps: testapp: Add an alarm test

Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>

diff --git a/wasp/apps/testapp.py b/wasp/apps/testapp.py
index 4c90109..69cb1a3 100644
--- a/wasp/apps/testapp.py
+++ b/wasp/apps/testapp.py
@@ -9,6 +9,8 @@ import machine
 import wasp
 import icons
 
+from apps.pager import PagerApp
+
 class TestApp():
     """Simple test application.
 
@@ -21,7 +23,7 @@ class TestApp():
     ICON = icons.app
 
     def __init__(self):
-        self.tests = ('Button', 'Crash', 'Colours', 'Fill', 'Fill-H', 'Fill-V', 'Notifications', 'RLE', 'String', 'Touch', 'Wrap')
+        self.tests = ('Alarm', 'Button', 'Crash', 'Colours', 'Fill', 'Fill-H', 'Fill-V', 'Notifications', 'RLE', 'String', 'Touch', 'Wrap')
         self.test = self.tests[0]
         self.scroll = wasp.widgets.ScrollIndicator()
 
@@ -41,7 +43,9 @@ class TestApp():
 
     def press(self, button, state):
         draw = wasp.watch.drawable
-        if self.test == 'Button':
+        if self.test == 'Alarm':
+            self._test_alarm()
+        elif self.test == 'Button':
             draw.string('{}: {}'.format(button, state), 0, 108, width=240)
         elif self.test == 'Crash':
             self.crash()
@@ -98,6 +102,20 @@ class TestApp():
         elif self.test == 'Wrap':
             self._benchmark_wrap()
 
+    def _alarm(self):
+        wasp.system.wake()
+        wasp.system.switch(PagerApp('Alarm triggered'))
+
+    def _test_alarm(self):
+        def nop():
+            pass
+        now = wasp.watch.rtc.time()
+        wasp.system.set_alarm(now + 30, self._alarm)
+        wasp.system.set_alarm(now + 30, nop)
+        if not wasp.system.cancel_alarm(now + 30, nop):
+            bug()
+        wasp.watch.drawable.string("Done.", 12, 24+80)
+
     def _benchmark_rle(self):
         draw = wasp.watch.drawable
         draw.fill(0, 0, 30, 240, 240-30)
@@ -177,6 +195,9 @@ class TestApp():
         draw.string('{} test'.format(self.test),
                 0, 6, width=240)
 
+        if self.test == 'Alarm':
+            draw.string("Press button to", 12, 24+24)
+            draw.string("set alarm.", 12, 24+48)
         if self.test == 'Crash':
             draw.string("Press button to", 12, 24+24)
             draw.string("throw exception.", 12, 24+48)
diff --git a/wasp/wasp.py b/wasp/wasp.py
index a415666..a44a61c 100644
--- a/wasp/wasp.py
+++ b/wasp/wasp.py
@@ -312,11 +312,12 @@ class Manager():
     def wake(self):
         """Return to a running state.
         """
-        watch.display.poweron()
-        if 'wake' in dir(self.app):
-            self.app.wake()
-        watch.backlight.set(self._brightness)
-        watch.touch.wake()
+        if not self.sleep_at:
+            watch.display.poweron()
+            if 'wake' in dir(self.app):
+                self.app.wake()
+            watch.backlight.set(self._brightness)
+            watch.touch.wake()
 
         self.keep_awake()
 
-- 
cgit v0.10.2