add hook callback

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

diff --git a/src/allmydata/test/test_magic_folder.py b/src/allmydata/test/test_magic_folder.py
index 56cd58e9791db1980eb9aed395258887fda738f3..f5179769253f707eff277d07e16b0eea84407e4f 100644 (file)
--- a/src/allmydata/test/test_magic_folder.py
+++ b/src/allmydata/test/test_magic_folder.py
@@ -1240,6 +1240,7 @@ class MockTest(MagicFolderTestMixin, unittest.TestCase):
         def hook(res):
             uploaded_d = self.magicfolder.uploader.set_hook('processed')
             return uploaded_d
+        d.addCallback(hook)
         d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('uploader.files_uploaded'), 1))
         d.addCallback(lambda ign: self.magicfolder.finish())
         return d