Added test cases for main.py

Author: jerrymakesjelly

autoremovetorrents/exception/connectionfailure.py

@@ -1,3 +1,3 @@
 class ConnectionFailure(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/exception/deletionfailure.py

@@ -1,3 +1,3 @@
 class DeletionFailure(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/exception/loginfailure.py

@@ -2,4 +2,4 @@
 
 class LoginFailure(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/exception/nosuchclient.py

@@ -2,4 +2,4 @@
 
 class NoSuchClient(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/exception/nosuchtorrent.py

@@ -1,3 +1,3 @@
 class NoSuchTorrent(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/exception/remotefailure.py

@@ -1,3 +1,3 @@
 class RemoteFailure(RuntimeError):
     def __init__(self, arg):
-        self.args = arg
+        self.args = (arg,)

autoremovetorrents/main.py

@@ -6,8 +6,9 @@
 import yaml
 from . import logger
 from .task import Task
+from autoremovetorrents.version import __version__
 
-def main():
+def main(argv):
     # View Mode
     view_mode = False
     # The path of the configuration file
@@ -19,7 +20,7 @@ def main():
 
     # Get arguments
     try:
-        opts = getopt.getopt(sys.argv[1:], 'vc:t:', ['view', 'conf=', 'task='])[0]
+        opts = getopt.getopt(argv, 'vc:t:', ['view', 'conf=', 'task='])[0]
     except getopt.GetoptError:
         print('Invalid arguments.')
         sys.exit(255)
@@ -33,6 +34,8 @@ def main():
 
     # Run autoremove
     try:
+        # Show version
+        lg.info('Auto Remove Torrents %s' % __version__)
         # Load configurations
         lg.info('Loading configurations...')
         with open(conf_path, 'r') as stream:
@@ -51,4 +54,4 @@ def main():
         lg.critical('An error occured. Please contact the administrator for more information.')
 
 if __name__ == '__main__':
-    main()
+    main(sys.argv[1:])

pytest/test_main/command_lines.txt

@@ -0,0 +1,8 @@
+--conf=pytest/test_main/config.yml
+-c pytest/test_main/config.yml
+-v -c pytest/test_main/config.yml
+--view -c pytest/test_main/config.yml
+-v --task=task1 -c pytest/test_main/config.yml
+-t task1 -c pytest/test_main/config.yml
+-t task10000 -c pytest/test_main/config.yml
+abcdefghijklmnopqrstuvwxyz

pytest/test_main/test_main.py

@@ -0,0 +1,15 @@
+import os
+from autoremovetorrents.main import main
+
+def test_main():
+    basic_dir = os.path.realpath(os.path.dirname(__file__))
+    print('Basic directory: %s' % basic_dir)
+    
+    # Open file of command lines
+    with open(os.path.join(basic_dir, 'command_lines.txt')) as f:
+        lines = f.readlines()
+        # Get command lines
+        for line in lines:
+            print('Command line: %s' % line)
+            argv = line.split()
+            main(argv) # Execute it