def test_cleanup(self):
"""
DispatcherHelper.cleanup() should call .cleanup() on its
PersistenceHelper and WorkerHelper.
"""
disp_helper = DispatcherHelper(None)
disp_helper.persistence_helper = FakeCleanupCheckHelper()
disp_helper.worker_helper = FakeCleanupCheckHelper()
self.assertEqual(disp_helper.persistence_helper.cleaned_up, False)
self.assertEqual(disp_helper.worker_helper.cleaned_up, False)
success_result_of(disp_helper.cleanup())
self.assertEqual(disp_helper.persistence_helper.cleaned_up, True)
self.assertEqual(disp_helper.worker_helper.cleaned_up, True)
