test_files.py 文件源码

def test_download(self):
        url_download = 'http://some-url-for-dowload/som-path/file'
        self.client.get.return_value = mock_response('http://some-url-for-dowload/som-path/file',
                                                     httplib.OK,
                                                     None,
                                                     'files', 'download.txt')

        def check_data(data, json, **kwargs):
            self.assertIn('stream', kwargs)
            self.assertTrue(kwargs['stream'])

        mock_response.check_data = check_data

        @mock.patch('__builtin__.open', spec=open, return_value=MockFile())
        def fire_test(mock_open):
            self.files.download(url_download, 'somewhere.txt')
            self.client.get.assert_called_with(self.client.get.return_value.url,
                                               verify=False,
                                               stream=True)
            self.assertEqual(''.join(mock_open.return_value.buffer), 'Some data downloaded')

        fire_test()