Add unit tests for command line progress

These test both percentage steps and a spinner where the number of steps is unknown.
riskysnail · Feb 8, 2024 · 60041f3 · 60041f3
1 parent 3e19b5f
commit 60041f3
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/gramps/cli/test/user_test.py b/gramps/cli/test/user_test.py
@@ -173,6 +173,26 @@ def _progress_begin_step_end(self):
             self.user.step_progress()
         self.user.end_progress()
 
+    def test_progress_percent(self):
+        self.user.begin_progress("Foo", "Bar", 20)
+        self.user._fileout.write.assert_called_with("Bar")
+        for i in range(20):
+            self.user.step_progress()
+            pct = (i + 1) * 5
+            self.user._fileout.write.assert_called_with("\r{:3d}% ".format(pct))
+        self.user.end_progress()
+        self.user._fileout.write.assert_called_with("\n")
+
+    def test_progress_spinner(self):
+        self.user.begin_progress("Foo", "Bar", 0)
+        self.user._fileout.write.assert_called_with("Bar")
+        for i in range(5):
+            self.user.step_progress()
+            spn = user._SPINNER[(i + 1) % 4]
+            self.user._fileout.write.assert_called_with("\r  {}  ".format(spn))
+        self.user.end_progress()
+        self.user._fileout.write.assert_called_with("\n")
+
 
 if __name__ == "__main__":
     unittest.main()