2 files changed, 10 insertions, 9 deletions
diff --git a/src/scadere/check.py b/src/scadere/check.py
index 7ef2d4a..5764382 100644
--- a/src/scadere/check.py
+++ b/src/scadere/check.py
@@ -23,7 +23,7 @@ from email.utils import parsedate_to_datetime as parsedate
 from itertools import chain
 from socket import AF_INET, socket
 from ssl import create_default_context as tls_context
-from sys import stderr, stdout
+from sys import argv, stderr, stdout
 
 from . import __version__, GNUHelpFormatter, NetLoc
 
@@ -66,7 +66,7 @@ def check(netlocs, after, output, fake_ca=None):
                       file=output)
 
 
-def main():
+def main(arguments=argv[1:]):
     """Run TLS checker."""
     description = ('Check TLS certificate expiration of HOST,'
                    ' where PORT defaults to 443.')
@@ -82,11 +82,11 @@ def main():
     parser.add_argument('-o', '--output', metavar='PATH',
                         type=FileType('w'), default=stdout,
                         help='output file (default to stdout)')
-    args = parser.parse_args()
-    with args.output:
+    args = parser.parse_args(arguments)
+    with args.output:  # pragma: no cover
         after = datetime.now(tz=timezone.utc) + timedelta(days=args.days)
         check(args.netloc, after, args.output)
 
 
-if __name__ == '__main__':
+if __name__ == '__main__':  # pragma: no cover
     main()
diff --git a/src/scadere/listen.py b/src/scadere/listen.py
index 4870e2f..80929ea 100644
--- a/src/scadere/listen.py
+++ b/src/scadere/listen.py
@@ -26,6 +26,7 @@ from urllib.parse import parse_qs, urljoin, urlsplit
 from xml.etree.ElementTree import (Element as xml_element,
                                    SubElement as xml_subelement,
                                    indent, tostring as str_from_xml)
+from sys import argv
 
 from . import __version__, GNUHelpFormatter, NetLoc
 
@@ -168,7 +169,7 @@ async def listen(certs, base_url, host, port):  # pragma: no cover
         await server.serve_forever()
 
 
-def main():
+def main(arguments=argv[1:]):
     """Launch server."""
     description = ('Serve the TLS certificate expiration feed'
                    ' from INPUT file for base URL at HOST:PORT,'
@@ -183,9 +184,9 @@ def main():
     parser.add_argument('base_url', metavar='URL')
     parser.add_argument('netloc', metavar='[HOST][:PORT]', nargs='?',
                         type=NetLoc(None), default=('localhost', None))
-    args = parser.parse_args()
-    run(listen(args.certs, args.base_url, *args.netloc))
+    args = parser.parse_args(arguments)
+    run(listen(args.certs, args.base_url, *args.netloc))  # pragma: no cover
 
 
-if __name__ == '__main__':
+if __name__ == '__main__':  # pragma: no cover
     main()