Added tests for the CLI parser for gist subcommands

Author: guyzmo

tests/helpers.py

@@ -25,6 +25,11 @@ def __init__(self, *args, **kwarg):
         self._did_create = None
         self._did_fork = None
         self._did_user = False
+        self._did_gist_list = None
+        self._did_gist_fetch = None
+        self._did_gist_clone = None
+        self._did_gist_create = None
+        self._did_gist_delete = None
 
     def pull(self, *args, **kwarg):
         self._did_pull = (args, kwarg)
@@ -50,6 +55,45 @@ def create(self, *args, **kwarg):
     def fork(self, *args, **kwarg):
         self._did_fork = (args, kwarg)
 
+    def gist_list(self, *args, **kwarg):
+        self._did_gist_list = (args, kwarg)
+        if len(args) == 0:
+            return [('id1', 'value1'),
+                    ('id2', 'value2'),
+                    ('id3', 'value3')]
+        elif len(args) == 1:
+            if args[0] == 'bad':
+                raise Exception('bad gist!')
+            else:
+                return [('lang1', 'size1', 'name1'),
+                        ('lang2', 'size2', 'name2'),
+                        ('lang3', 'size3', 'name3')]
+
+    def gist_fetch(self, *args, **kwarg):
+        self._did_gist_fetch = (args, kwarg)
+        if args[0] == 'bad':
+            raise Exception('bad gist!')
+        elif args[1] == 'bad':
+            raise Exception('bad gist file!')
+        else:
+            return "content of a gist"
+
+    def gist_clone(self, *args, **kwarg):
+        self._did_gist_clone = (args, kwarg)
+        if args[0] == 'bad':
+            raise Exception('bad gist!')
+
+    def gist_create(self, *args, **kwarg):
+        self._did_gist_create = (args, kwarg)
+        if 'exists' in args[0]:
+            raise Exception('gist exists!')
+        return 'https://gists/42'
+
+    def gist_delete(self, *args, **kwarg):
+        self._did_gist_delete = (args, kwarg)
+        if args[0] == 'bad':
+            raise Exception('bad gist!')
+
     @property
     def user(self):
         self._did_user = True
@@ -101,6 +145,7 @@ def setup_args(self, d, args={}):
             'fetch': False,
             'fork': False,
             'list': False,
+            'ls': False,
             'open': False,
             '--secret': False,
             '<description>': None,
@@ -160,23 +205,47 @@ def main_fork(self, repo, rc=0, args={}):
         }, args)), "Non {} result for fork".format(rc)
         return RepositoryService._current._did_fork
 
-    def main_request_list(self, repo, rc=0, args={}):
-        assert True
+    def main_gist_list(self, rc=0, args={}):
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'list': True,
+        }, args)), "Non {} result for gist list".format(rc)
+        return RepositoryService._current._did_gist_list
 
-    def main_request_fetch(self, repo, rc=0, args={}):
-        assert True
+    def main_gist_ls(self, rc=0, args={}):
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'ls': True,
+        }, args)), "Non {} result for gist ls".format(rc)
+        return RepositoryService._current._did_gist_list
 
-    def main_gist_list(self, rc=0, args={}):
-        assert True
+    def main_gist_clone(self, rc=0, args={}):
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'clone': True,
+        }, args)), "Non {} result for gist clone".format(rc)
+        return RepositoryService._current._did_gist_clone
 
-    def main_gist_fetch(self, capsys, rc=0, args={}):
-        assert True
+    def main_gist_fetch(self, rc=0, args={}):
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'fetch': True,
+        }, args)), "Non {} result for gist fetch".format(rc)
+        return RepositoryService._current._did_gist_fetch
 
     def main_gist_create(self, rc=0, args={}):
-        assert True
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'create': True,
+        }, args)), "Non {} result for gist create".format(rc)
+        return RepositoryService._current._did_gist_create
 
     def main_gist_delete(self, rc=0, args={}):
-        assert True
+        assert rc == main(self.setup_args({
+            'gist': True,
+            'delete': True,
+        }, args)), "Non {} result for gist delete".format(rc)
+        return RepositoryService._current._did_gist_delete
 
     def main_open(self, repo, rc=0, args={}):
         os.mkdir(os.path.join(self.tempdir.name, repo.split('/')[-1]))

tests/integration/test_main.py

@@ -154,53 +154,133 @@ def test_fork__branch_clone(self):
         assert ('guyzmo', 'git-repo') == repo_slug
         assert {'branch': 'foobar', 'clone': True} == seen_args
 
-    def test_gist_list(self):
-        repo_slug, seen_args = self.main_gist_list(0, args={})
-
-    def test_gist_list__with_gist(self):
-        repo_slug, seen_args = self.main_gist_list(0, args={})
-
-    def test_gist_list__with_bad_gist(self):
-        repo_slug, seen_args = self.main_gist_list(0, args={})
-
-    def test_gist_clone__with_gist(self):
-        repo_slug, seen_args = self.main_gist_clone(0, args={})
-
-    def test_gist_clone__with_bad_gist(self):
-        repo_slug, seen_args = self.main_gist_clone(0, args={})
+    def test_gist_list(self, capsys, caplog):
+        did_list = self.main_gist_list(0, args={})
+        out, err = capsys.readouterr()
+        assert did_list == ((), {})
+        assert 'id' in caplog.text and 'title' in caplog.text
+        assert out ==  ''.join([
+            'id1                                                     \tvalue1\n',
+            'id2                                                     \tvalue2\n',
+            'id3                                                     \tvalue3\n'
+            ])
+
+    def test_gist_ls(self, capsys, caplog):
+        did_list = self.main_gist_ls(0, args={})
+        out, err = capsys.readouterr()
+        assert did_list == ((), {})
+        assert 'id' in caplog.text and 'title' in caplog.text
+        assert out ==  ''.join([
+            'id1                                                     \tvalue1\n',
+            'id2                                                     \tvalue2\n',
+            'id3                                                     \tvalue3\n'
+            ])
+
+    def test_gist_list__with_gist(self, capsys, caplog):
+        did_list = self.main_gist_list(0, args={'<gist>': 'foobar'})
+        out, err = capsys.readouterr()
+        assert did_list == (('foobar',), {})
+        assert 'language' in caplog.text and 'size' in caplog.text and 'name' in caplog.text
+        assert out ==  ''.join([
+            'lang1          \tsize1  \tname1\n',
+            'lang2          \tsize2  \tname2\n',
+            'lang3          \tsize3  \tname3\n'
+            ])
+
+    def test_gist_list__with_bad_gist(self, caplog):
+        did_list = self.main_gist_list(2, args={'<gist>': 'bad', '--verbose': 0})
+        assert did_list == (('bad',), {})
+        assert 'language       \t   size\tname' in caplog.text
+        assert 'Fatal error: bad gist!' in caplog.text
+
+    def test_gist_clone__with_gist(self, caplog):
+        did_clone = self.main_gist_clone(0, args={'<gist>': '42'})
+        assert did_clone == (('42',), {})
+        assert 'git init' in caplog.text
+        assert 'Successfully cloned `42` into `./42`!' in caplog.text
+
+    def test_gist_clone__with_bad_gist(self, caplog):
+        did_clone = self.main_gist_clone(2, args={'<gist>': 'bad', '--verbose': 0})
+        assert did_clone == (('bad',), {})
+        assert 'Fatal error: bad gist!\n' in caplog.text
 
     def test_gist_fetch__with_gist(self, capsys):
-        repo_slug, seen_args = self.main_gist_fetch(capsys, 0, args={})
+        did_fetch = self.main_gist_fetch(0, args={'<gist>': '42'})
+        out, err = capsys.readouterr()
+        assert did_fetch == (('42', None), {})
+        assert out == 'content of a gist\n'
+        assert err == ''
 
-    def test_gist_fetch__with_bad_gist(self, capsys):
-        repo_slug, seen_args = self.main_gist_fetch(capsys, 0, args={})
+    def test_gist_fetch__with_bad_gist(self, caplog):
+        did_fetch = self.main_gist_fetch(2, args={'<gist>': 'bad', '--verbose': 0})
+        assert did_fetch == (('bad', None), {})
+        assert 'Fatal error: bad gist!\n' in caplog.text
 
     def test_gist_fetch__with_gist_file(self, capsys):
-        repo_slug, seen_args = self.main_gist_fetch(capsys, 0, args={})
-
-    def test_gist_fetch__with_bad_gist_file(self, capsys):
-        repo_slug, seen_args = self.main_gist_fetch(capsys, 0, args={})
-
-    def test_gist_create__gist_file(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_create__gist_file_list(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_create__gist_dir(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_create__gist_file_list_secret(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_create__gist_file_secret(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_create__gist_dir_secret(self):
-        repo_slug, seen_args = self.main_gist_create(0, args={})
-
-    def test_gist_delete(self):
-        repo_slug, seen_args = self.main_gist_delete(0, args={})
+        did_fetch = self.main_gist_fetch(0, args={'<gist>': '42', '<gist_file>': 'fubar'})
+        out, err = capsys.readouterr()
+        assert did_fetch == (('42', 'fubar'), {})
+        assert out == 'content of a gist\n'
+        assert err == ''
+
+    def test_gist_fetch__with_bad_gist_file(self, caplog):
+        did_fetch = self.main_gist_fetch(2, args={'<gist>': '42', '<gist_file>': 'bad', '--verbose': 0})
+        assert did_fetch == (('42', 'bad'), {})
+        assert 'Fatal error: bad gist file!\n' in caplog.text
+
+    def test_gist_create(self, caplog):
+        did_create = self.main_gist_create(0, args={
+            '<gist_path>': ['f1'],
+            '<description>': 'fubar',
+            '--secret': False,
+            '--verbose': 0
+
+        })
+        assert did_create == ((['f1'], 'fubar', False), {})
+        assert 'Successfully created gist `https://gists/42`!\n' in caplog.text
+
+    def test_gist_create__exists(self, caplog):
+        did_create = self.main_gist_create(2, args={
+            '<gist_path>': ['exists'],
+            '<description>': 'fubar',
+            '--secret': False,
+            '--verbose': 0
+        })
+        assert did_create == ((['exists'], 'fubar', False), {})
+        assert 'Fatal error: gist exists!' in caplog.text
+
+    def test_gist_create__secret(self, caplog):
+        did_create = self.main_gist_create(0, args={
+            '<gist_path>': ['f1'],
+            '<description>': 'fubar',
+            '--secret': True,
+            '--verbose': 0
+
+        })
+        assert did_create == ((['f1'], 'fubar', True), {})
+        assert 'Successfully created gist `https://gists/42`!\n' in caplog.text
+
+    def test_gist_create__secret__exists(self, caplog):
+        did_create = self.main_gist_create(2, args={
+            '<gist_path>': ['exists'],
+            '<description>': 'fubar',
+            '--secret': True,
+            '--verbose': 0
+        })
+        assert did_create == ((['exists'], 'fubar', True), {})
+        assert 'Fatal error: gist exists!' in caplog.text
+
+    def test_gist_delete__force(self, caplog):
+        did_delete = self.main_gist_delete(0, args={'--force': True})
+        assert 'Successfully deleted gist!' in caplog.text
+
+    def test_gist_delete__ask(self, caplog):
+        import io, sys
+        stdin = sys.stdin
+        sys.stdin = io.StringIO('y\nburn!')
+        did_delete = self.main_gist_delete(0, args={'--force': True})
+        sys.stdin = stdin
+        assert 'Successfully deleted gist!' in caplog.text
 
     def test_open(self):
         repo_slug, seen_args = self.main_open('guyzmo/git-repo', 0)