patch the get_port function

Author: Keerthan Mala

rootfs/api/tests/__init__.py

@@ -7,6 +7,10 @@
 from django.test.runner import DiscoverRunner
 
 
+def mock_port(*args, **kwargs):
+    return 5000
+
+
 # Mock out router requests and add in some jitter
 # Used for application is available in router checks
 def fake_responses(request, context):

rootfs/api/tests/test_app.py

@@ -17,6 +17,7 @@
 from scheduler import KubeException
 
 from . import adapter
+from . import mock_port
 import requests_mock
 
 
@@ -30,6 +31,7 @@ def _mock_run(*args, **kwargs):
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class AppTest(APITestCase):
     """Tests creation of applications"""
 

rootfs/api/tests/test_build.py

@@ -19,11 +19,13 @@
 from scheduler import KubeException
 
 from . import adapter
+from . import mock_port
 import requests_mock
 
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class BuildTest(APITransactionTestCase):
 
     """Tests build notification from build system"""

rootfs/api/tests/test_config.py

@@ -15,11 +15,13 @@
 from api.models import App, Config
 
 from . import adapter
+from . import mock_port
 import requests_mock
 
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class ConfigTest(APITransactionTestCase):
 
     """Tests setting and updating config values"""

rootfs/api/tests/test_hooks.py

@@ -11,6 +11,7 @@
 from rest_framework.authtoken.models import Token
 
 from . import adapter
+from . import mock_port
 import requests_mock
 
 RSA_PUBKEY = (
@@ -34,6 +35,7 @@
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class HookTest(APITransactionTestCase):
 
     """Tests API hooks used to trigger actions from external components"""

rootfs/api/tests/test_pods.py

@@ -18,11 +18,13 @@
 from scheduler import KubeException
 
 from . import adapter
+from . import mock_port
 import requests_mock
 
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class PodTest(APITransactionTestCase):
     """Tests creation of pods on nodes"""
 

rootfs/api/tests/test_release.py

@@ -17,11 +17,13 @@
 from api.models import Release
 from scheduler import KubeHTTPException
 from . import adapter
+from . import mock_port
 import requests_mock
 
 
 @requests_mock.Mocker(real_http=True, adapter=adapter)
 @mock.patch('api.models.release.publish_release', lambda *args: None)
+@mock.patch('scheduler.KubeHTTPClient._get_port', mock_port)
 class ReleaseTest(APITransactionTestCase):
 
     """Tests push notification from build system"""