import unittest
import types
from config_service.api.v1.resources.appilcations_rules.application_rules_utils import random_available_ports


class TestSplitPorts(unittest.TestCase):
    def setUp(self) -> None:
        # Given
        self.exclude_ports = [1024, 1025, 1027, 1035, 1050, 1027]
        self.start_port = 1024
        self.end_port = 1060

    def tearDown(self) -> None:
        pass

    def test_random_available_ports_returns_generator(self):
        # When
        available_ports = random_available_ports(self.start_port, self.end_port, self.exclude_ports)
        # Then
        self.assertIsInstance(available_ports, types.GeneratorType)

    def test_random_available_ports_doesnt_return_excluded_port(self):
        # Given
        start_port = 1024  # shadows self.start_port
        end_port = 1074  # shadows self.end_port
        # When
        available_ports = list(random_available_ports(start_port, end_port, self.exclude_ports))
        # Then
        self.assertNotIn(self.exclude_ports, available_ports)

    def test_random_available_ports_generates_next_available_port(self):
        available_ports = random_available_ports(self.start_port, self.end_port, self.exclude_ports)
        self.assertEqual(next(available_ports), 1026)