tests: Added test for msg()

1 files changed, 10 insertions, 0 deletions

diff --git a/tests/test_util.py b/tests/test_util.py
index 01d1c9c..3a3ecdb 100755
--- a/tests/test_util.py
+++ b/tests/test_util.py
@@ -1,5 +1,7 @@
 """Test util functions."""
 import unittest
+import unittest.mock
+import io
 import os
 import pathlib
 import time
@@ -95,6 +97,14 @@ class TestUtil(unittest.TestCase):
         self.assertTrue(result)
         os.remove(test_file)
 
+    def test_msg(self):
+        """> Test displaying a message."""
+        # Since this function just prints a message we redirect
+        # it's output so that we can read it.
+        with unittest.mock.patch('sys.stdout', new=io.StringIO()) as fake_out:
+            util.msg("test", True)
+            self.assertEqual(fake_out.getvalue().strip(), "test")
+
 
 if __name__ == "__main__":
     unittest.main()