add script test

Saurabh Kumar · Saurabh Kumar · commit 5f181b43b566 · 2014-09-07T01:05:10.000+05:30
diff --git a/README.md b/README.md
@@ -1,5 +1,7 @@
 # django-dotenv-rw
 
+[![Build Status](https://travis-ci.org/theskumar/django-dotenv-rw.svg?branch=master)](https://travis-ci.org/theskumar/django-dotenv-rw)
+
 Forked from awesome but simpler [django-dotenv](https://github.com/jacobian/django-dotenv).  Removes black magic, makes loading .env in settings.py easier, adds remote .env file management capabilities.  Works as a drop-in replacement for django-dotenv.
 
 [foreman](https://github.com/ddollar/foreman) reads from `.env`. `manage.py`
@@ -10,7 +12,7 @@ doesn't. Let's fix that.
 ## Installation
 
 ```
-pip install git+ssh://git@github.com/tedtieken/django-dotenv-rw.git
+pip install git+ssh://git@github.com/theskumar/django-dotenv-rw.git
 ```
 
 ## Usage: loading settings from a .env file into your django environment
diff --git a/dotenv.py b/dotenv.py
@@ -130,21 +130,21 @@ def cli(file, action, key, value, force):
         stored_value = get_key(file, key)
         if stored_value:
             click.echo("%s=%s" % (key, stored_value))
-            exit(1)
+            exit(0)
         else:
             click.echo("%s doesn't seems to have been set yet.")
-            exit(0)
+            exit(1)
 
     elif action == 'set':
         if not value:
             click.echo("Error: value is missing.")
-            exit(0)
+            exit(1)
         success, key, value = set_key(file, key, value)
         if success:
             click.echo("%s=%s" % (key, value))
-            exit(1)
-        else:
             exit(0)
+        else:
+            exit(1)
 
     elif action == 'unset':
         success, key = unset_key(file, key)
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -13,3 +13,11 @@ def test_read_write():
     stored_value = dotenv.get_key(dotenv_path, 'HELLO')
     assert stored_value == 'WORLD'
     sh.rm(dotenv_path)
+
+
+def test_script():
+    sh.touch(dotenv_path)
+    sh.dotenv('set', 'HELLO', 'WORLD', '-f', dotenv_path)
+    output = sh.dotenv('get', 'HELLO', '-f', dotenv_path)
+    assert output == 'HELLO=WORLD\n'
+    sh.rm(dotenv_path)
