Add test suite for SessionProxy

1 files changed, 112 insertions, 0 deletions

diff --git a/tests/test_sessionproxy.py b/tests/test_sessionproxy.py
new file mode 100644
index 000000000..1214214d8
--- /dev/null
+++ b/tests/test_sessionproxy.py
@@ -0,0 +1,112 @@
+import time
+import sys
+from twisted.trial import unittest
+from twisted.internet.defer import maybeDeferred, succeed, DeferredList
+import deluge.ui.sessionproxy
+import deluge.component as component
+
+class Core(object):
+    def __init__(self):
+        self.reset()
+
+    def reset(self):
+        self.torrents = {}
+        self.torrents["a"] = {"key1": 1, "key2": 2, "key3": 3}
+        self.torrents["b"] = {"key1": 1, "key2": 2, "key3": 3}
+        self.torrents["c"] = {"key1": 1, "key2": 2, "key3": 3}
+        self.prev_status = {}
+
+    def get_torrent_status(self, torrent_id, keys, diff=False):
+        if not keys:
+            keys = self.torrents[torrent_id].keys()
+
+        if not diff:
+            ret = {}
+            for key in keys:
+                ret[key] = self.torrents[torrent_id][key]
+
+            return succeed(ret)
+
+        else:
+            ret = {}
+            if torrent_id in self.prev_status:
+                for key in keys:
+                    if self.prev_status[torrent_id][key] != self.torrents[torrent_id][key]:
+                        ret[key] = self.torrents[torrent_id][key]
+            else:
+                ret = self.torrents[torrent_id]
+            self.prev_status[torrent_id] = dict(self.torrents[torrent_id])
+            return succeed(ret)
+
+    def get_torrents_status(self, filter_dict, keys, diff=False):
+        if not filter_dict:
+            filter_dict["id"] = self.torrents.keys()
+        if not keys:
+            keys = self.torrents["a"].keys()
+        if not diff:
+            if "id" in filter_dict:
+                torrents = filter_dict["id"]
+                ret = {}
+                for torrent in torrents:
+                    ret[torrent] = {}
+                    for key in keys:
+                        ret[torrent][key] = self.torrents[torrent][key]
+                return succeed(ret)
+        else:
+            if "id" in filter_dict:
+                torrents = filter_dict["id"]
+                ret = {}
+                for torrent in torrents:
+                    ret[torrent] = {}
+                    if torrent in self.prev_status:
+                        for key in self.prev_status[torrent]:
+                            if self.prev_status[torrent][key] != self.torrents[torrent][key]:
+                                ret[torrent][key] = self.torrents[torrent][key]
+                    else:
+                        ret[torrent] = dict(self.torrents[torrent])
+
+                    self.prev_status[torrent] = dict(self.torrents[torrent])
+                return succeed(ret)
+
+class Client(object):
+    def __init__(self):
+        self.core = Core()
+
+    def __noop__(self, *args, **kwargs):
+        return None
+    def __getattr__(self, *args, **kwargs):
+        return self.__noop__
+
+client = Client()
+
+deluge.ui.sessionproxy.client = client
+class SessionProxyTestCase(unittest.TestCase):
+    def setUp(self):
+        self.sp = deluge.ui.sessionproxy.SessionProxy()
+        client.core.reset()
+        d = self.sp.start()
+        return d
+
+    def tearDown(self):
+        return component.deregister("SessionProxy")
+
+    def test_startup(self):
+        self.assertEquals(client.core.torrents["a"], self.sp.torrents["a"][1])
+
+    def test_get_torrent_status_no_change(self):
+        d = self.sp.get_torrent_status("a", [])
+        d.addCallback(self.assertEquals, client.core.torrents["a"])
+        return d
+
+    def test_get_torrent_status_change_with_cache(self):
+        client.core.torrents["a"]["key1"] = 2
+        d = self.sp.get_torrent_status("a", ["key1"])
+        d.addCallback(self.assertEquals, {"key1": 1})
+        return d
+
+    def test_get_torrent_status_change_without_cache(self):
+        client.core.torrents["a"]["key1"] = 2
+        time.sleep(self.sp.cache_time + 0.1)
+        d = self.sp.get_torrent_status("a", [])
+        d.addCallback(self.assertEquals, client.core.torrents["a"])
+        return d