Simplify immutable download API: use just filenode.read(consumer, offset, size)

[tahoe-lafs/tahoe-lafs.git] / src / allmydata / test / test_download.py

diff --git a/src/allmydata/test/test_download.py b/src/allmydata/test/test_download.py
index a4743d2eb6c62a8e3038d850342007d68da80c91..b54bf017437b4326caae99e7ea11ed280ebb3eaa 100644 (file)
--- a/src/allmydata/test/test_download.py
+++ b/src/allmydata/test/test_download.py
@@ -8,6 +8,7 @@ from twisted.trial import unittest
 from allmydata import uri
 from allmydata.storage.server import storage_index_to_dir
 from allmydata.util import base32, fileutil
+from allmydata.util.consumer import download_to_data
 from allmydata.immutable import upload
 from allmydata.test.no_network import GridTestMixin
 
@@ -173,7 +174,7 @@ class DownloadTest(GridTestMixin, unittest.TestCase):
 
     def download_immutable(self, ignored=None):
         n = self.c0.create_node_from_uri(immutable_uri)
-        d = n.download_to_data()
+        d = download_to_data(n)
         def _got_data(data):
             self.failUnlessEqual(data, plaintext)
         d.addCallback(_got_data)