def setUp(self): response = Mock(spec=requests.Response) response.raw = Mock( spec=requests.packages.urllib3.response.HTTPResponse) response.raw.closed = False response.raw.getheader.return_value = 1028 client = Mock(spec=DropboxClient) client.files_download.return_value = ({}, response) self.reader = ChunkedReader(client, '/file1.txt')
def test_open_read(self, mock_download): """Test opening a file for read.""" response = Mock(spec=requests.Response) response.raw = Mock( spec=requests.packages.urllib3.response.HTTPResponse) response.raw.getheader.return_value = 0 mock_download.return_value = ({}, response) reader = self.fs.open('/file.txt') self.assertIsInstance(reader, ChunkedReader)