diff --git a/tests/test_main.py b/tests/test_main.py index 9262c8f..196e3cf 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,28 +1,43 @@ """Test __main__ functions.""" import unittest -from unittest.mock import MagicMock +import unittest.mock from pywal import __main__ from pywal import reload +from pywal import wallpaper +from pywal import util from pywal.settings import CACHE_DIR class TestMain(unittest.TestCase): """Test the gen_colors functions.""" - def test_clean(self): - """> Test arg parsing (-c)""" + def test_args_a(self): + """> Test arg parsing (-a).""" + args = __main__.get_args(["-a", "50"]) + __main__.process_args(args) + self.assertEqual(util.Color.alpha_num, "50") + + def test_args_c(self): + """> Test arg parsing (-c).""" args = __main__.get_args(["-c"]) __main__.process_args(args) self.assertFalse((CACHE_DIR / "schemes").is_dir()) def test_args_e(self): - """> Test arg parsing (-e)""" - reload.env = MagicMock() + """> Test arg parsing (-e).""" + reload.env = unittest.mock.MagicMock() args = __main__.get_args(["-e"]) __main__.process_args(args) self.assertFalse(reload.env.called) + def test_args_n(self): + """> Test arg parsing (-n).""" + wallpaper.change = unittest.mock.MagicMock() + args = __main__.get_args(["-n"]) + __main__.process_args(args) + self.assertFalse(wallpaper.change.called) + if __name__ == "__main__": unittest.main()