From 8b6d6e3836aa29e26d3a6b5eb0a67b9a54650a4b Mon Sep 17 00:00:00 2001 From: Pedro Algarvio Date: Tue, 5 Jul 2011 09:02:45 +0100 Subject: [PATCH] Fix broken `SessionProxy` tests. --- deluge/tests/test_sessionproxy.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/deluge/tests/test_sessionproxy.py b/deluge/tests/test_sessionproxy.py index 593fbe075..d97a4d622 100644 --- a/deluge/tests/test_sessionproxy.py +++ b/deluge/tests/test_sessionproxy.py @@ -1,7 +1,6 @@ import time -import sys from twisted.trial import unittest -from twisted.internet.defer import maybeDeferred, succeed, DeferredList +from twisted.internet.defer import maybeDeferred, succeed import deluge.ui.sessionproxy import deluge.component as component @@ -16,6 +15,9 @@ class Core(object): self.torrents["c"] = {"key1": 1, "key2": 2, "key3": 3} self.prev_status = {} + def get_session_state(self): + return maybeDeferred(self.torrents.keys) + def get_torrent_status(self, torrent_id, keys, diff=False): if not keys: keys = self.torrents[torrent_id].keys()