From 77ef1ea1a2bd63a4d9fa6b487c57e536d8205c5f Mon Sep 17 00:00:00 2001
From: Benoit Pierre <benoit.pierre@gmail.com>
Date: Wed, 25 Dec 2024 15:46:14 +0100
Subject: [PATCH] tests: cleanup and speedup screenshoter tests

---
 spec/unit/screenshoter_spec.lua | 27 ++++++++-------------------
 1 file changed, 8 insertions(+), 19 deletions(-)

diff --git a/spec/unit/screenshoter_spec.lua b/spec/unit/screenshoter_spec.lua
index 668ad1e69..ab91a1680 100644
--- a/spec/unit/screenshoter_spec.lua
+++ b/spec/unit/screenshoter_spec.lua
@@ -1,7 +1,7 @@
 describe("ReaderScreenshot module", function()
     local DataStorage, DocumentRegistry, ReaderUI, lfs, UIManager, Event, Screen
-    local sample_epub = "spec/front/unit/data/leaves.epub"
     local readerui
+
     setup(function()
         require("commonrequire")
         disable_plugins()
@@ -15,45 +15,34 @@ describe("ReaderScreenshot module", function()
 
         readerui = ReaderUI:new{
             dimen = Screen:getSize(),
-            document = DocumentRegistry:openDocument(sample_epub),
+            document = DocumentRegistry:openDocument("spec/front/unit/data/sample.txt"),
         }
     end)
 
     teardown(function()
         readerui:handleEvent(Event:new("SetRotationMode", Screen.DEVICE_ROTATED_UPRIGHT))
-        readerui:closeDocument()
         readerui:onClose()
     end)
 
+    after_each(function()
+        UIManager:quit()
+    end)
+
     it("should get screenshot in portrait", function()
         local name = DataStorage:getDataDir() .. "/screenshots/reader_screenshot_portrait.png"
         readerui:handleEvent(Event:new("SetRotationMode", Screen.DEVICE_ROTATED_UPRIGHT))
-        UIManager:quit()
         UIManager:show(readerui)
-        UIManager:scheduleIn(1, function()
-            UIManager:close(readerui)
-            -- We haven't torn it down yet
-            ReaderUI.instance = readerui
-        end)
-        UIManager:run()
+        fastforward_ui_events()
         readerui.screenshot:onScreenshot(name)
         assert.truthy(lfs.attributes(name, "mode"))
-        UIManager:quit()
     end)
 
     it("should get screenshot in landscape", function()
         local name = DataStorage:getDataDir() .. "/screenshots/reader_screenshot_landscape.png"
         readerui:handleEvent(Event:new("SetRotationMode", Screen.DEVICE_ROTATED_CLOCKWISE))
-        UIManager:quit()
         UIManager:show(readerui)
-        UIManager:scheduleIn(2, function()
-            UIManager:close(readerui)
-            -- We haven't torn it down yet
-            ReaderUI.instance = readerui
-        end)
-        UIManager:run()
+        fastforward_ui_events()
         readerui.screenshot:onScreenshot(name)
         assert.truthy(lfs.attributes(name, "mode"))
-        UIManager:quit()
     end)
 end)