From c8e4c1752f18b9bd9d0a0318729778e52d47475a Mon Sep 17 00:00:00 2001 From: David Stainton Date: Tue, 30 Jun 2015 16:03:28 -0700 Subject: [PATCH] Add test for Bob download callback --- src/allmydata/test/test_magic_folder.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/src/allmydata/test/test_magic_folder.py b/src/allmydata/test/test_magic_folder.py index 190b1918..59ef855c 100644 --- a/src/allmydata/test/test_magic_folder.py +++ b/src/allmydata/test/test_magic_folder.py @@ -333,14 +333,24 @@ class MagicFolderTestMixin(MagicFolderCLITestMixin, ShouldFailMixin, ReallyEqual self.alice_magicfolder.set_processed_callback(d2.callback, ignore_count=0) return d2 d.addCallback(wait_for_upload) - def prepare_for_stats(result): + def prepare_for_alice_stats(result): self.stats_provider = self.alice_magicfolder._client.stats_provider - d.addCallback(prepare_for_stats) + d.addCallback(prepare_for_alice_stats) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('magic_folder.objects_succeeded'), 1)) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('magic_folder.files_uploaded'), 1)) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('magic_folder.objects_queued'), 0)) d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('magic_folder.directories_created'), 0)) + def wait_for_download(result): + d2 = defer.Deferred() + self.bob_magicfolder.set_download_callback(d2.callback, ignore_count=0) + return d2 + d.addCallback(wait_for_download) + def prepare_for_bob_stats(result): + self.stats_provider = self.bob_magicfolder._client.stats_provider + d.addCallback(prepare_for_bob_stats) + d.addCallback(lambda ign: self.failUnlessReallyEqual(self._get_count('magic_folder.objects_succeeded'), 1)) + def cleanup_Alice_and_Bob(result): d = defer.succeed(None) d.addCallback(lambda ign: self.alice_magicfolder.finish(for_tests=True)) -- 2.45.2