extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2025-02-03 12:00:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

tests: Added test for msg()

Browse Source
This commit is contained in:
Dylan Araps 2017-07-23 20:29:42 +10:00
parent a38115407c
commit 64e0ca26f1
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_util.py
View File

@ -1,5 +1,7 @@
"""Test util functions.""" """Test util functions."""
import unittest import unittest
import unittest.mock
import io
import os import os
import pathlib import pathlib
import time import time
@ -95,6 +97,14 @@ class TestUtil(unittest.TestCase):
self.assertTrue(result) self.assertTrue(result)
os.remove(test_file) 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__": if __name__ == "__main__":
unittest.main() unittest.main()