diff --git a/circle/vm/tests/test_models.py b/circle/vm/tests/test_models.py index affc8b9..e561f80 100644 --- a/circle/vm/tests/test_models.py +++ b/circle/vm/tests/test_models.py @@ -1,14 +1,26 @@ from django.test import TestCase +from ..models.common import ( + Lease +) from ..models.instance import ( - InstanceTemplate, Instance + find_unused_port, InstanceTemplate, Instance ) from ..models.network import ( Interface ) -from ..models.common import ( - Lease -) + + +class PortFinderTestCase(TestCase): + + def test_find_unused_port_without_used_ports(self): + port = find_unused_port(port_range=(1000, 2000)) + assert port is not None + + def test_find_unused_port_with_fully_saturated_range(self): + r = (10, 20) + port = find_unused_port(port_range=r, used_ports=range(*r)) + assert port is None class TemplateTestCase(TestCase):