diff options
author | 2017-09-03 20:32:55 -0700 | |
---|---|---|
committer | 2017-09-03 20:34:03 -0700 | |
commit | 3fa27a2a339c09516b5ab52c5b0f1ef01f0928e2 (patch) | |
tree | 7fba13fde253b7f003989170ca61bf86b4175cb7 /tests/unit/apiserver/__init__.py | |
parent | c19115bafe0141326b189f317a684bf4681c8bd0 (diff) |
apiserver: mock socket.gethostbyname
we don't want use external DNS requests during unittest
Change-Id: I5ed67b700ef1dab4b650ae5071a3cf641a17ae4c
Signed-off-by: Ross Brattain <ross.b.brattain@intel.com>
Diffstat (limited to 'tests/unit/apiserver/__init__.py')
-rw-r--r-- | tests/unit/apiserver/__init__.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/unit/apiserver/__init__.py b/tests/unit/apiserver/__init__.py index 1c9d5a672..5e1ed2ea1 100644 --- a/tests/unit/apiserver/__init__.py +++ b/tests/unit/apiserver/__init__.py @@ -1,5 +1,6 @@ from __future__ import absolute_import +import mock import os import socket import unittest @@ -16,6 +17,10 @@ class APITestCase(unittest.TestCase): self.db_fd, self.db_path = tempfile.mkstemp() consts.SQLITE = 'sqlite:///{}'.format(self.db_path) + # 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: @@ -30,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'} |