mirror of
https://github.com/dylanaraps/pywal.git
synced 2024-06-01 11:07:29 -04:00
tests: Test args
This commit is contained in:
parent
3fdad9fb89
commit
6d116540dd
|
@ -24,7 +24,7 @@ from . import util
|
|||
from . import wallpaper
|
||||
|
||||
|
||||
def get_args():
|
||||
def get_args(args):
|
||||
"""Get the script arguments."""
|
||||
description = "wal - Generate colorschemes on the fly"
|
||||
arg = argparse.ArgumentParser(description=description)
|
||||
|
@ -62,7 +62,7 @@ def get_args():
|
|||
arg.add_argument("-v", action="store_true",
|
||||
help="Print \"wal\" version.")
|
||||
|
||||
return arg.parse_args()
|
||||
return arg.parse_args(args)
|
||||
|
||||
|
||||
def process_args(args):
|
||||
|
@ -115,7 +115,7 @@ def process_args(args):
|
|||
|
||||
def main():
|
||||
"""Main script function."""
|
||||
args = get_args()
|
||||
args = get_args(sys.argv[1:])
|
||||
process_args(args)
|
||||
|
||||
# This saves 10ms.
|
||||
|
|
37
tests/test_main.py
Normal file
37
tests/test_main.py
Normal file
|
@ -0,0 +1,37 @@
|
|||
"""Test __main__ functions."""
|
||||
import unittest
|
||||
from pywal import __main__
|
||||
|
||||
|
||||
class TestMain(unittest.TestCase):
|
||||
"""Test the gen_colors functions."""
|
||||
|
||||
def test_no_args(self):
|
||||
"""> Generate a colorscheme and fail."""
|
||||
with self.assertRaises(SystemExit):
|
||||
args = __main__.get_args([""])
|
||||
__main__.process_args(args)
|
||||
|
||||
def test_conflict(self):
|
||||
"""> Test arg parsing (-i, -f)"""
|
||||
with self.assertRaises(SystemExit):
|
||||
args = __main__.get_args(["-i", "-f"])
|
||||
__main__.process_args(args)
|
||||
|
||||
def test_version(self):
|
||||
"""> Test arg parsing (-v)"""
|
||||
args = __main__.get_args(["-v"])
|
||||
self.assertTrue(args.v)
|
||||
|
||||
def test_quiet(self):
|
||||
"""> Test arg parsing (-q)"""
|
||||
args = __main__.get_args(["-q"])
|
||||
self.assertTrue(args.q)
|
||||
|
||||
def test_ext_script(self):
|
||||
"""> Test arg parsing (-o)"""
|
||||
args = __main__.get_args(["-o", "true"])
|
||||
self.assertTrue(args.o)
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
Loading…
Reference in a new issue