Tests for passing in a repo arg.

tests/test_vcs.py

@@ -11,22 +11,69 @@
 import logging
 import os
 import shutil
+import sys
 import unittest
 
+PY3 = sys.version > '3'
+if PY3:
+    from unittest.mock import patch
+    input_str = 'builtins.input'
+else:
+    import __builtin__
+    from mock import patch
+    input_str = '__builtin__.raw_input'
+    from cStringIO import StringIO
+
 from cookiecutter import vcs
 
 
 # Log debug and above to console
 logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.DEBUG)
 
+
 class TestVCS(unittest.TestCase):
 
     def test_git_clone(self):
-        repo_dir = vcs.git_clone('https://github.com/audreyr/cookiecutter-pypackage.git')
+        repo_dir = vcs.git_clone(
+            'https://github.com/audreyr/cookiecutter-pypackage.git'
+        )
         self.assertEqual(repo_dir, 'cookiecutter-pypackage')
         self.assertTrue(os.path.isfile('cookiecutter-pypackage/README.rst'))
         if os.path.isdir('cookiecutter-pypackage'):
             shutil.rmtree('cookiecutter-pypackage')
 
+
+class TestVCSPrompt(unittest.TestCase):
+
+    def setUp(self):
+        if os.path.isdir('cookiecutter-pypackage'):
+            shutil.rmtree('cookiecutter-pypackage')
+        os.mkdir('cookiecutter-pypackage/')
+
+    @patch(input_str, lambda: 'y')
+    def test_git_clone_overwrite(self):
+        if not PY3:
+            sys.stdin = StringIO('y\n\n')
+        repo_dir = vcs.git_clone(
+            'https://github.com/audreyr/cookiecutter-pypackage.git'
+        )
+        self.assertEqual(repo_dir, 'cookiecutter-pypackage')
+        self.assertTrue(os.path.isfile('cookiecutter-pypackage/README.rst'))
+
+    @patch(input_str, lambda: 'n')
+    def test_git_clone_cancel(self):
+        if not PY3:
+            sys.stdin = StringIO('n\n\n')
+        self.assertRaises(
+            SystemExit,
+            vcs.git_clone,
+            'https://github.com/audreyr/cookiecutter-pypackage.git'
+        )
+
+    def tearDown(self):
+        if os.path.isdir('cookiecutter-pypackage'):
+            shutil.rmtree('cookiecutter-pypackage')
+
+
 if __name__ == '__main__':
     unittest.main()