Fix bug that someomes blocked plugins accessing connectionserver sitelist
This commit is contained in:
parent
07633ba79d
commit
c14e722303
5 changed files with 16 additions and 14 deletions
|
@ -65,7 +65,7 @@ class FileServer(ConnectionServer):
|
||||||
|
|
||||||
self.port_opened = {}
|
self.port_opened = {}
|
||||||
|
|
||||||
self.sites = {}
|
self.sites = self.site_manager.sites
|
||||||
self.last_request = time.time()
|
self.last_request = time.time()
|
||||||
self.files_parsing = {}
|
self.files_parsing = {}
|
||||||
self.ui_server = None
|
self.ui_server = None
|
||||||
|
|
|
@ -91,7 +91,7 @@ class TestFileRequest:
|
||||||
def testPex(self, file_server, site, site_temp):
|
def testPex(self, file_server, site, site_temp):
|
||||||
file_server.sites[site.address] = site
|
file_server.sites[site.address] = site
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
connection = client.getConnection(file_server.ip, 1544)
|
connection = client.getConnection(file_server.ip, 1544)
|
||||||
|
|
||||||
|
|
|
@ -15,7 +15,7 @@ class TestPeer:
|
||||||
def testPing(self, file_server, site, site_temp):
|
def testPing(self, file_server, site, site_temp):
|
||||||
file_server.sites[site.address] = site
|
file_server.sites[site.address] = site
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
connection = client.getConnection(file_server.ip, 1544)
|
connection = client.getConnection(file_server.ip, 1544)
|
||||||
|
|
||||||
|
@ -34,7 +34,7 @@ class TestPeer:
|
||||||
def testDownloadFile(self, file_server, site, site_temp):
|
def testDownloadFile(self, file_server, site, site_temp):
|
||||||
file_server.sites[site.address] = site
|
file_server.sites[site.address] = site
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
connection = client.getConnection(file_server.ip, 1544)
|
connection = client.getConnection(file_server.ip, 1544)
|
||||||
|
|
||||||
|
@ -129,7 +129,7 @@ class TestPeer:
|
||||||
def testFindHash(self, file_server, site, site_temp):
|
def testFindHash(self, file_server, site, site_temp):
|
||||||
file_server.sites[site.address] = site
|
file_server.sites[site.address] = site
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Add file_server as peer to client
|
# Add file_server as peer to client
|
||||||
|
|
|
@ -26,7 +26,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
||||||
|
|
||||||
|
@ -35,6 +35,8 @@ class TestSiteDownload:
|
||||||
|
|
||||||
site_temp.download(blind_includes=True).join(timeout=5)
|
site_temp.download(blind_includes=True).join(timeout=5)
|
||||||
|
|
||||||
|
assert site_temp.storage.isFile("content.json")
|
||||||
|
|
||||||
# Rename non-optional file
|
# Rename non-optional file
|
||||||
os.rename(site.storage.getPath("data/img/domain.png"), site.storage.getPath("data/img/domain-new.png"))
|
os.rename(site.storage.getPath("data/img/domain.png"), site.storage.getPath("data/img/domain-new.png"))
|
||||||
|
|
||||||
|
@ -75,7 +77,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
||||||
|
|
||||||
|
@ -131,7 +133,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Download normally
|
# Download normally
|
||||||
|
@ -179,7 +181,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Download normally
|
# Download normally
|
||||||
|
@ -343,7 +345,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Don't try to find peers from the net
|
# Don't try to find peers from the net
|
||||||
|
@ -419,7 +421,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Connect peers
|
# Connect peers
|
||||||
|
@ -470,7 +472,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Connect peers
|
# Connect peers
|
||||||
|
@ -517,7 +519,7 @@ class TestSiteDownload:
|
||||||
|
|
||||||
# Init client server
|
# Init client server
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
||||||
|
|
||||||
|
|
|
@ -117,7 +117,7 @@ class TestTor:
|
||||||
file_server.tor_manager = tor_manager
|
file_server.tor_manager = tor_manager
|
||||||
|
|
||||||
client = FileServer(file_server.ip, 1545)
|
client = FileServer(file_server.ip, 1545)
|
||||||
client.sites[site_temp.address] = site_temp
|
client.sites = {site_temp.address: site_temp}
|
||||||
site_temp.connection_server = client
|
site_temp.connection_server = client
|
||||||
|
|
||||||
# Add file_server as peer to client
|
# Add file_server as peer to client
|
||||||
|
|
Loading…
Reference in a new issue