From 4e80a68b227b92d16dadea4fd01d29f058944df0 Mon Sep 17 00:00:00 2001 From: David Stainton Date: Tue, 21 Apr 2015 08:17:24 +0100 Subject: [PATCH] Fix db persistence unit test for uploader --- src/allmydata/test/test_drop_upload.py | 28 +++++++++----------------- 1 file changed, 10 insertions(+), 18 deletions(-) diff --git a/src/allmydata/test/test_drop_upload.py b/src/allmydata/test/test_drop_upload.py index 14cecb81..08476f99 100644 --- a/src/allmydata/test/test_drop_upload.py +++ b/src/allmydata/test/test_drop_upload.py @@ -143,35 +143,27 @@ class DropUploadTestMixin(GridTestMixin, ShouldFailMixin, ReallyEqualMixin, NonA self.stats_provider = self.client.stats_provider d = self.client.create_dirnode() d.addCallback(self._made_upload_dir) - d.addCallback(self._cleanup) - d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_uploaded'), 0)) - d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_queued'), 0)) - d.addCallback(lambda ign: self.client.removeService(self.uploader)) - - def msg(val): - print "old uploader removed" - return None - d.addCallback(msg) def create_file(val): + d2 = defer.Deferred() + self.uploader.set_uploaded_callback(d2.callback) myFile = os.path.join(self.local_dir, "what") f = open(myFile, "wb") f.write("meow") f.close() - print "test file created" - return None + self.notify_close_write(FilePath(myFile)) + return d2 d.addCallback(create_file) - + d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_uploaded'), 1)) + d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_queued'), 0)) + d.addCallback(self._cleanup) + d.addCallback(lambda ign: self.client.removeService(self.uploader)) + d.addCallback(self._made_upload_dir) - def resume_uploader(val): - print "created new uploader" - d = defer.Deferred() - self.uploader.set_uploaded_callback(d.callback) - return d - d.addCallback(resume_uploader) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_queued'), 0)) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('drop_upload.objects_uploaded'), 1)) d.addBoth(self._cleanup) + return d def _test(self): -- 2.45.2