X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=tests%2Funit%2Fapiserver%2F__init__.py;h=5e1ed2ea1b94197f11e92f1326551e3981195bf5;hb=1e5e1977fab21941e6bd2158e6cf10680ceaa2f6;hp=02141529689542327bb4d48a8280cb8ce006173c;hpb=fd1d3c0504c5e5b153b78bd0d5bd8fe3f4bc37d4;p=yardstick.git

diff --git a/tests/unit/apiserver/__init__.py b/tests/unit/apiserver/__init__.py
index 021415296..5e1ed2ea1 100644
--- a/tests/unit/apiserver/__init__.py
+++ b/tests/unit/apiserver/__init__.py
@@ -1,6 +1,8 @@
 from __future__ import absolute_import
 
+import mock
 import os
+import socket
 import unittest
 import tempfile
 
@@ -14,7 +16,16 @@ class APITestCase(unittest.TestCase):
     def setUp(self):
         self.db_fd, self.db_path = tempfile.mkstemp()
         consts.SQLITE = 'sqlite:///{}'.format(self.db_path)
-        from api import server
+
+        # server calls gethostbyname which takes 4 seconds, and we should mock it anyway
+        self.socket_mock = mock.patch.dict("sys.modules", {"socket": mock.MagicMock(
+            **{"gethostbyname.return_value": "127.0.0.1", "gethostname.return_value": "localhost"})})
+        self.socket_mock.start()
+        try:
+            from api import server
+        except socket.gaierror:
+            self.app = None
+            return
 
         server.app.config['TESTING'] = True
         self.app = server.app.test_client()
@@ -24,6 +35,7 @@ class APITestCase(unittest.TestCase):
     def tearDown(self):
         os.close(self.db_fd)
         os.unlink(self.db_path)
+        self.socket_mock.stop()
 
     def _post(self, url, data):
         headers = {'Content-Type': 'application/json'}