From b14081625c3d5fac8c339db30832f98b75e40691 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Sun, 23 Jul 2017 23:31:16 +1000 Subject: [PATCH] tests: Implement all arguments. --- tests/test_main.py | 33 +++++++++++++++++++++++++++++++++ tests/test_sequences.py | 6 +++--- 2 files changed, 36 insertions(+), 3 deletions(-) diff --git a/tests/test_main.py b/tests/test_main.py index 321f565..400962e 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,6 +1,9 @@ """Test __main__ functions.""" import unittest +import os + from pywal import __main__ +from pywal.settings import __cache_dir__ class TestMain(unittest.TestCase): @@ -35,12 +38,42 @@ class TestMain(unittest.TestCase): __main__.process_args(args) self.assertTrue(args.q) + def test_alpha(self): + """> Test arg parsing (-a)""" + args = __main__.get_args(["-a", "99"]) + __main__.process_args(args) + self.assertTrue(args.a) + def test_ext_script(self): """> Test arg parsing (-o)""" args = __main__.get_args(["-o", "true"]) __main__.process_args(args) self.assertTrue(args.o) + def test_clean(self): + """> Test arg parsing (-c)""" + args = __main__.get_args(["-c"]) + __main__.process_args(args) + self.assertFalse(os.path.isdir(__cache_dir__ / "schemes")) + + def test_reload(self): + """> Test arg parsing (-r)""" + with self.assertRaises(SystemExit): + args = __main__.get_args(["-r"]) + __main__.process_args(args) + + def test_image(self): + """> Test arg parsing (-i)""" + args = __main__.get_args(["-i", "tests/test_files/test.jpg"]) + __main__.process_args(args) + self.assertTrue(args.i) + + def test_json(self): + """> Test arg parsing (-f)""" + args = __main__.get_args(["-f", "tests/test_files/test_file.json"]) + __main__.process_args(args) + self.assertTrue(args.f) + if __name__ == "__main__": unittest.main() diff --git a/tests/test_sequences.py b/tests/test_sequences.py index bd03516..fe88909 100755 --- a/tests/test_sequences.py +++ b/tests/test_sequences.py @@ -21,16 +21,16 @@ class Testsequences(unittest.TestCase): def test_set_special_alpha(self): """> Create special escape sequence with alpha.""" - util.Color.alpha_num = 40 + util.Color.alpha_num = 99 result = sequences.set_special(11, COLORS["special"]["background"]) - self.assertEqual(result, "\033]11;[40]#1F211E\007") + self.assertEqual(result, "\033]11;[99]#1F211E\007") def test_set_color(self): """> Create color escape sequence.""" result = sequences.set_color(11, COLORS["colors"]["color0"]) self.assertEqual(result, "\033]4;11;#1F211E\007") - def test_send_srquences(self): + def test_send_sequences(self): """> Send sequences to all open terminals.""" with unittest.mock.patch('sys.stdout', new=io.StringIO()) as fake_out: sequences.send(COLORS, False)