ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-02-02 01:08:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add test for Bob download callback

Browse Source
This commit is contained in:
David Stainton 2015-06-30 16:03:28 -07:00 committed by Daira Hopwood
parent d1d43d72da
commit 4db3adbbd1
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/allmydata/test/test_magic_folder.py
View File

@ -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))