36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
import time
|
|
import os
|
|
|
|
import gevent
|
|
import pytest
|
|
import mock
|
|
|
|
from Crypt import CryptConnection
|
|
from Connection import ConnectionServer
|
|
from Config import config
|
|
from Site import Site
|
|
|
|
@pytest.mark.usefixtures("resetTempSettings")
|
|
@pytest.mark.usefixtures("resetSettings")
|
|
class TestWorker:
|
|
def testDownload(self, file_server, site, site_temp):
|
|
client = ConnectionServer("127.0.0.1", 1545)
|
|
assert site.storage.directory == config.data_dir+"/"+site.address
|
|
assert site_temp.storage.directory == config.data_dir+"-temp/"+site.address
|
|
|
|
# Init source server
|
|
site.connection_server = file_server
|
|
file_server.sites[site.address] = site
|
|
|
|
# Init client server
|
|
site_temp.connection_server = client
|
|
site_temp.announce = mock.MagicMock(return_value=True) # Don't try to find peers from the net
|
|
|
|
# Download to client from source
|
|
site_temp.addPeer("127.0.0.1", 1544)
|
|
site_temp.download().join(timeout=5)
|
|
|
|
assert not site_temp.bad_files
|
|
|
|
site_temp.storage.deleteFiles()
|