AnnounceLocal plugin

plugins/AnnounceLocal/Test/TestAnnounce.py

@@ -0,0 +1,105 @@
+import time
+import copy
+
+import gevent
+import pytest
+
+from AnnounceLocal import AnnounceLocalPlugin
+from File import FileServer
+from Test import Spy
+
+@pytest.fixture
+def announcer(file_server, site):
+    file_server.sites[site.address] = site
+    file_server.local_announcer.listen_port = 1100
+    file_server.local_announcer.sender_info["broadcast_port"] = 1100
+    file_server.local_announcer.listen_ip = file_server.local_announcer.getMyIps()[0]
+    gevent.spawn(file_server.local_announcer.start)
+    time.sleep(0.3)
+
+    assert file_server.local_announcer.running
+    return file_server.local_announcer
+
+@pytest.fixture
+def announcer_remote(site_temp):
+    file_server_remote = FileServer("127.0.0.1", 1545)
+    file_server_remote.sites[site_temp.address] = site_temp
+    file_server_remote.local_announcer.listen_port = 1101
+    file_server_remote.local_announcer.sender_info["broadcast_port"] = 1101
+    file_server_remote.local_announcer.listen_ip = file_server_remote.local_announcer.getMyIps()[0]
+    gevent.spawn(file_server_remote.local_announcer.start)
+    time.sleep(0.3)
+
+    assert file_server_remote.local_announcer.running
+    return file_server_remote.local_announcer
+
+@pytest.mark.usefixtures("resetSettings")
+@pytest.mark.usefixtures("resetTempSettings")
+class TestAnnounce:
+    def testSenderInfo(self, file_server):
+        # gevent.spawn(announcer.listen)
+
+        sender_info = file_server.local_announcer.sender_info
+        assert sender_info["port"] > 0
+        assert len(sender_info["peer_id"]) == 20
+        assert sender_info["rev"] > 0
+
+    def testIgnoreSelfMessages(self, file_server, site):
+        file_server.sites[site.address] = site
+        announcer = file_server.local_announcer
+
+        # No response to messages that has same peer_id as server
+        assert not announcer.handleMessage(("0.0.0.0", 123), {"cmd": "discoverRequest", "sender": announcer.sender_info, "params": {}})[1]
+
+        # Response to messages with different peer id
+        sender_info = copy.copy(announcer.sender_info)
+        sender_info["peer_id"] += "-"
+        addr, res = announcer.handleMessage(("0.0.0.0", 123), {"cmd": "discoverRequest", "sender": sender_info, "params": {}})
+        assert res["params"]["sites_changed"] > 0
+
+    def testDiscoverRequest(self, announcer, announcer_remote):
+        assert len(announcer_remote.known_peers) == 0
+        with Spy.Spy(announcer_remote, "handleMessage") as responses:
+            announcer_remote.broadcast({"cmd": "discoverRequest", "params": {}}, port=announcer.listen_port)
+            time.sleep(0.1)
+
+        response_cmds = [response[1]["cmd"] for response in responses]
+        assert response_cmds == ["discoverResponse", "siteListResponse"]
+        assert len(responses[-1][1]["params"]["sites"]) == 1
+
+        # It should only request siteList if sites_changed value is different from last response
+        with Spy.Spy(announcer_remote, "handleMessage") as responses:
+            announcer_remote.broadcast({"cmd": "discoverRequest", "params": {}}, port=announcer.listen_port)
+            time.sleep(0.1)
+
+        response_cmds = [response[1]["cmd"] for response in responses]
+        assert response_cmds == ["discoverResponse"]
+
+    def testPeerDiscover(self, announcer, announcer_remote, site):
+        assert announcer.server.peer_id != announcer_remote.server.peer_id
+        assert len(announcer.server.sites.values()[0].peers) == 0
+        announcer.broadcast({"cmd": "discoverRequest"}, port=announcer_remote.listen_port)
+        time.sleep(0.1)
+        assert len(announcer.server.sites.values()[0].peers) == 1
+
+    def testRecentPeerList(self, announcer, announcer_remote, site):
+        assert len(site.peers_recent) == 0
+        assert len(site.peers) == 0
+        with Spy.Spy(announcer, "handleMessage") as responses:
+            announcer.broadcast({"cmd": "discoverRequest", "params": {}}, port=announcer_remote.listen_port)
+            time.sleep(0.1)
+        assert [response[1]["cmd"] for response in responses] == ["discoverResponse", "siteListResponse"]
+        assert len(site.peers_recent) == 1
+        assert len(site.peers) == 1
+
+        # It should update peer without siteListResponse
+        last_time_found = site.peers.values()[0].time_found
+        site.peers_recent.clear()
+        with Spy.Spy(announcer, "handleMessage") as responses:
+            announcer.broadcast({"cmd": "discoverRequest", "params": {}}, port=announcer_remote.listen_port)
+            time.sleep(0.1)
+        assert [response[1]["cmd"] for response in responses] == ["discoverResponse"]
+        assert len(site.peers_recent) == 1
+        assert site.peers.values()[0].time_found > last_time_found
+
+