improve test coverage

Saurabh Kumar · Saurabh Kumar · commit 3fb3f2caefed · 2015-03-01T11:01:24.000+05:30
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -0,0 +1 @@
+from .fixtures import *  # noqa
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -0,0 +1,7 @@
+import pytest
+from click.testing import CliRunner
+
+
+@pytest.fixture()
+def cli():
+    return CliRunner()
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -33,18 +33,32 @@ def test_unset():
     assert success is None
 
 
-def test_console_script():
-    sh.touch(dotenv_path)
-    sh.dotenv('-f', dotenv_path, 'set', 'HELLO', 'WORLD')
-    output = sh.dotenv('-f', dotenv_path, 'get', 'HELLO', )
-    assert output == 'HELLO="WORLD"\n'
-    sh.rm(dotenv_path)
+def test_console_script(cli):
+    with cli.isolated_filesystem():
+        sh.touch(dotenv_path)
+        sh.dotenv('-f', dotenv_path, 'set', 'HELLO', 'WORLD')
+        output = sh.dotenv('-f', dotenv_path, 'get', 'HELLO', )
+        assert output == 'HELLO="WORLD"\n'
+        sh.rm(dotenv_path)
 
+    # should fail for not existing file
+    result = cli.invoke(dotenv.set, ['my_key', 'my_value'])
+    assert result.exit_code != 0
 
-def test_default_path():
-    sh.touch(dotenv_path)
-    sh.cd(here)
-    sh.dotenv('set', 'HELLO', 'WORLD')
-    output = sh.dotenv('get', 'HELLO')
-    assert output == 'HELLO="WORLD"\n'
-    sh.rm(dotenv_path)
+    # should fail for not existing file
+    result = cli.invoke(dotenv.get, ['my_key'])
+    assert result.exit_code != 0
+
+    # should fail for not existing file
+    result = cli.invoke(dotenv.list, [])
+    assert result.exit_code != 0
+
+
+def test_default_path(cli):
+    with cli.isolated_filesystem():
+        sh.touch(dotenv_path)
+        sh.cd(here)
+        sh.dotenv('set', 'HELLO', 'WORLD')
+        output = sh.dotenv('get', 'HELLO')
+        assert output == 'HELLO="WORLD"\n'
+        sh.rm(dotenv_path)
