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)