Add test for download feature on Release

b4stien · b4stien · commit 386514649a8c · 2015-11-06T12:07:16.000+01:00
diff --git a/tests/test_repos.py b/tests/test_repos.py
@@ -951,3 +951,50 @@ def test_download(self):
         self.response('', 404)
         self.request.side_effect = None
         assert self.asset.download() is False
+
+
+class TestRelease(BaseCase):
+    def __init__(self, methodName='runTest'):
+        super(TestRelease, self).__init__(methodName)
+        self.release = repos.release.Release(load('release'))
+
+    def test_archive(self):
+        headers = {'content-disposition': 'filename=foo'}
+        self.response('archive', 200, **headers)
+        self.get('https://api.github.com/repos/sigmavirus24/github3.py/tarball/v0.7.1')
+        self.conf.update({'stream': True})
+
+        assert self.release.archive(None) is False
+
+        assert os.path.isfile('foo') is False
+        assert self.release.archive('tarball')
+        assert os.path.isfile('foo')
+        os.unlink('foo')
+        self.mock_assertions()
+
+        self.request.return_value.raw.seek(0)
+        self.request.return_value._content_consumed = False
+
+        assert os.path.isfile('path_to_file') is False
+        assert self.release.archive('tarball', 'path_to_file')
+        assert os.path.isfile('path_to_file')
+        os.unlink('path_to_file')
+
+        self.request.return_value.raw.seek(0)
+        self.request.return_value._content_consumed = False
+
+        self.get('https://api.github.com/repos/sigmavirus24/github3.py/zipball/v0.7.1')
+        assert self.release.archive('zipball')
+        os.unlink('foo')
+
+        self.request.return_value.raw.seek(0)
+        self.request.return_value._content_consumed = False
+
+        o = mock.mock_open()
+        with mock.patch('{0}.open'.format(__name__), o, create=True):
+            with open('archive', 'wb+') as fd:
+                self.release.archive('tarball', fd)
+
+        o.assert_called_once_with('archive', 'wb+')
+        fd = o()
+        fd.write.assert_called_once_with(b'archive_data')
