Implementar y probar el componente main

Author: Esteban Echeverry

taskit/__main__.py

@@ -1,4 +1,5 @@
 import os
+import sys
 from pathlib import Path
 from taskit.application.coordinators.admin_coordinator import (
     AdminCoordinator)
@@ -13,10 +14,9 @@
 from taskit.infrastructure.cli.taskit import cli, State
 
 
-def prepare_json_database() -> str:
-    taskit_dir = Path.home().joinpath('.taskit')
-    os.makedirs(str(taskit_dir), exist_ok=True)
-    db_path = str(taskit_dir.joinpath('db.json'))
+def prepare_json_database(taskit_dir: str) -> str:
+    os.makedirs(taskit_dir, exist_ok=True)
+    db_path = os.sep.join([taskit_dir, 'db.json'])
     with open(db_path, 'a+') as f:
         f.seek(0)
         contents = f.read()
@@ -42,9 +42,17 @@ def build_state(json_file: str) -> State:
 
 
 def main():
-    db_file = prepare_json_database()
+    taskit_dir = (
+        os.environ.get('TASKIT_DIR') or str(Path.home().joinpath('.taskit'))
+    )
+    db_file = prepare_json_database(taskit_dir)
     state = build_state(db_file)
-    cli(obj=state)
+
+    try:
+        cli(obj=state)
+    except Exception as e:
+        print("\nEXECUTION ERROR:", e)
+        sys.exit(-1)
 
 
 if __name__ == '__main__':

tests/test_main.py

@@ -1,5 +1,44 @@
-from taskit.__main__ import main
+import os
+import sys
+import taskit.__main__
+from subprocess import run
+from pytest import fixture
+from unittest.mock import Mock, patch
+from taskit.__main__ import main, build_state
 
 
-# def test_main():
-#     main()
+@fixture(scope='module')
+def taskit_dir(tmpdir_factory: fixture) -> str:
+    taskit_dir = tmpdir_factory.mktemp('taskitdir')
+    return str(taskit_dir)
+
+
+def test_main(taskit_dir) -> None:
+    sandbox = patch.dict(os.environ, {'TASKIT_DIR': taskit_dir})
+    sandbox.start()
+    python_bin = sys.executable
+    result = run([python_bin, '-m', 'taskit'])
+    sandbox.stop()
+    assert result.returncode == 0
+
+
+def test_main_db_file_not_empty(taskit_dir) -> None:
+    sandbox = patch.dict(os.environ, {'TASKIT_DIR': taskit_dir})
+    sandbox.start()
+    db_path = os.sep.join([taskit_dir, 'db.json'])
+    with open(db_path, 'w+') as f:
+        f.write('{"projects": {}}')
+    python_bin = sys.executable
+    result = run([python_bin, '-m', 'taskit'])
+    sandbox.stop()
+    assert result.returncode == 0
+
+
+def test_main_exception_handling(taskit_dir, monkeypatch) -> None:
+    sandbox = patch.dict(os.environ, {'TASKIT_DIR': taskit_dir})
+    sandbox.start()
+    monkeypatch.setattr("taskit.__main__.build_state", lambda x: "NOTHING")
+    python_bin = sys.executable
+    result = run([python_bin, '-m', 'taskit', 'report', 'tasks'])
+    sandbox.stop()
+    assert result.returncode != 0