From d94957b79f9413cb5925c81a7861a3e1a04891e1 Mon Sep 17 00:00:00 2001
From: Antoine Nguyen <tonio@ngyn.org>
Date: Sun, 8 Oct 2017 14:39:40 +0200
Subject: [PATCH] Fixed unit test.

---
 run.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/run.py b/run.py
index 0e95ed5..bfa4b5f 100755
--- a/run.py
+++ b/run.py
@@ -7,6 +7,7 @@ try:
     import configparser
 except ImportError:
     import ConfigParser as configparser
+import sys
 
 from modoboa_installer import compatibility_matrix
 from modoboa_installer import package
@@ -15,7 +16,7 @@ from modoboa_installer import ssl
 from modoboa_installer import utils
 
 
-def main():
+def main(input_args):
     """Install process."""
     parser = argparse.ArgumentParser()
     parser.add_argument("--debug", action="store_true", default=False,
@@ -36,7 +37,7 @@ def main():
         help="Generate configuration file with user interaction")
     parser.add_argument("domain", type=str,
                         help="The main domain of your future mail server")
-    args = parser.parse_args()
+    args = parser.parse_args(input_args)
 
     if args.debug:
         utils.ENV["debug"] = True
@@ -91,4 +92,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main(sys.argv[1:])