import logging
import unittest
from pathlib import Path

from simulation import utils
from simulation.key_creator import KeyCreator
from simulation.networks import Network
from simulation.operations import simulator

log = logging.getLogger(__name__)


class TestCyclicSearch(unittest.TestCase):
    def test_all_searches(self):
        network = Network.from_config(
            {"connect_type": "cyclical", "num_search_tests": 10, "groups": [{"size": 10}]},
            KeyCreator(),
        )
        path = Path("simulation_output/tests/cyclical") / f"{utils.get_formatted_time()}"

        result = simulator.simulate(network, path)
        self.assertEqual(result.success_percentage, 1)


class TestRootedSearch(unittest.TestCase):
    def test_all_searches(self):
        network = Network.from_config(
            {"connect_type": "rooted", "num_search_tests": 10, "groups": [{"size": 10}]},
            KeyCreator(),
        )
        path = Path("simulation_output/tests/rooted") / f"{utils.get_formatted_time()}"

        result = simulator.simulate(network, path)
        self.assertEqual(result.success_percentage, 1)