bittorrent-benchmarks/benchmarks/deluge/tests/test_deluge_static_experiment.py

40 lines
1.4 KiB
Python

from benchmarks.core.experiments.static_experiment import StaticDisseminationExperiment
from benchmarks.core.utils import RandomTempData, megabytes
from benchmarks.deluge.deluge_node import DelugeMeta
from benchmarks.deluge.tests.test_deluge_node import assert_is_seed
def test_should_run_with_a_single_seeder(tracker, deluge_node1, deluge_node2, deluge_node3):
size = megabytes(10)
experiment = StaticDisseminationExperiment(
network=[deluge_node1, deluge_node2, deluge_node3],
seeders=[1],
data=RandomTempData(
size=size,
meta=DelugeMeta('dataset-1', announce_url=tracker)
)
)
experiment.run()
assert_is_seed(deluge_node1, 'dataset-1', size)
assert_is_seed(deluge_node2, 'dataset-1', size)
assert_is_seed(deluge_node3, 'dataset-1', size)
def test_should_run_with_multiple_seeders(tracker, deluge_node1, deluge_node2, deluge_node3):
size = megabytes(10)
experiment = StaticDisseminationExperiment(
network=[deluge_node1, deluge_node2, deluge_node3],
seeders=[1, 2],
data=RandomTempData(
size=size,
meta=DelugeMeta('dataset-1', announce_url=tracker)
)
)
experiment.run()
assert_is_seed(deluge_node1, 'dataset-1', size)
assert_is_seed(deluge_node2, 'dataset-1', size)
assert_is_seed(deluge_node3, 'dataset-1', size)