Convert functional test to a framework and decouple from client

Using client library here is not convenient, as it does not have
new features we need to test. Also we should only test API itself.

Instead, we'll use new Base class and mocked_server context manager
to build a separate functional test for client.

Change-Id: I44250d3df0d202e5fe5f94e6ec553a1f86053637
This commit is contained in:
Dmitry Tantsur 2015-07-22 15:22:51 +02:00
parent 3fff75de0e
commit ad576bd806
2 changed files with 69 additions and 46 deletions

View File

@ -14,6 +14,7 @@
import eventlet import eventlet
eventlet.monkey_patch() eventlet.monkey_patch()
import contextlib
import json import json
import os import os
import shutil import shutil
@ -21,7 +22,6 @@ import sys
import tempfile import tempfile
import unittest import unittest
import ironic_inspector_client as client
import mock import mock
import requests import requests
@ -50,9 +50,9 @@ connection = sqlite:///%(db_file)s
DEFAULT_SLEEP = 2 DEFAULT_SLEEP = 2
class Test(base.NodeTest): class Base(base.NodeTest):
def setUp(self): def setUp(self):
super(Test, self).setUp() super(Base, self).setUp()
self.cli = utils.get_client() self.cli = utils.get_client()
self.cli.reset_mock() self.cli.reset_mock()
@ -82,30 +82,52 @@ class Test(base.NodeTest):
self.node.power_state = 'power off' self.node.power_state = 'power off'
def call_ramdisk(self): def call(self, method, endpoint, data=None, expect_errors=False):
res = requests.post('http://127.0.0.1:5050/v1/continue', if data is not None:
data=json.dumps(self.data)) data = json.dumps(data)
endpoint = 'http://127.0.0.1:5050' + endpoint
headers = {'X-Auth-Token': 'token'}
res = getattr(requests, method.lower())(endpoint, data=data,
headers=headers)
if not expect_errors:
res.raise_for_status() res.raise_for_status()
return res return res
def call_introspect(self, uuid, new_ipmi_username=None,
new_ipmi_password=None):
endpoint = '/v1/introspection/%s' % uuid
if new_ipmi_password:
endpoint += '?new_ipmi_password=%s' % new_ipmi_password
if new_ipmi_username:
endpoint += '&new_ipmi_username=%s' % new_ipmi_username
return self.call('post', endpoint)
def call_get_status(self, uuid):
return self.call('get', '/v1/introspection/%s' % uuid).json()
def call_continue(self, data):
return self.call('post', '/v1/continue', data=data).json()
class Test(Base):
def test_bmc(self): def test_bmc(self):
client.introspect(self.uuid, auth_token='token') self.call_introspect(self.uuid)
eventlet.greenthread.sleep(DEFAULT_SLEEP) eventlet.greenthread.sleep(DEFAULT_SLEEP)
self.cli.node.set_power_state.assert_called_once_with(self.uuid, self.cli.node.set_power_state.assert_called_once_with(self.uuid,
'reboot') 'reboot')
status = client.get_status(self.uuid, auth_token='token') status = self.call_get_status(self.uuid)
self.assertEqual({'finished': False, 'error': None}, status) self.assertEqual({'finished': False, 'error': None}, status)
res = self.call_ramdisk() res = self.call_continue(self.data)
self.assertEqual({'uuid': self.uuid}, res.json()) self.assertEqual({'uuid': self.uuid}, res)
eventlet.greenthread.sleep(DEFAULT_SLEEP) eventlet.greenthread.sleep(DEFAULT_SLEEP)
self.cli.node.update.assert_any_call(self.uuid, self.patch) self.cli.node.update.assert_any_call(self.uuid, self.patch)
self.cli.port.create.assert_called_once_with( self.cli.port.create.assert_called_once_with(
node_uuid=self.uuid, address='11:22:33:44:55:66') node_uuid=self.uuid, address='11:22:33:44:55:66')
status = client.get_status(self.uuid, auth_token='token') status = self.call_get_status(self.uuid)
self.assertEqual({'finished': True, 'error': None}, status) self.assertEqual({'finished': True, 'error': None}, status)
def test_setup_ipmi(self): def test_setup_ipmi(self):
@ -116,19 +138,18 @@ class Test(base.NodeTest):
'value': 'pwd'}, 'value': 'pwd'},
] ]
self.node.maintenance = True self.node.maintenance = True
client.introspect(self.uuid, auth_token='token', self.call_introspect(self.uuid, new_ipmi_username='admin',
new_ipmi_username='admin', new_ipmi_password='pwd') new_ipmi_password='pwd')
eventlet.greenthread.sleep(DEFAULT_SLEEP) eventlet.greenthread.sleep(DEFAULT_SLEEP)
self.assertFalse(self.cli.node.set_power_state.called) self.assertFalse(self.cli.node.set_power_state.called)
status = client.get_status(self.uuid, auth_token='token') status = self.call_get_status(self.uuid)
self.assertEqual({'finished': False, 'error': None}, status) self.assertEqual({'finished': False, 'error': None}, status)
res = self.call_ramdisk() res = self.call_continue(self.data)
res_json = res.json() self.assertEqual('admin', res['ipmi_username'])
self.assertEqual('admin', res_json['ipmi_username']) self.assertEqual('pwd', res['ipmi_password'])
self.assertEqual('pwd', res_json['ipmi_password']) self.assertTrue(res['ipmi_setup_credentials'])
self.assertTrue(res_json['ipmi_setup_credentials'])
eventlet.greenthread.sleep(DEFAULT_SLEEP) eventlet.greenthread.sleep(DEFAULT_SLEEP)
self.cli.node.update.assert_any_call(self.uuid, self.patch) self.cli.node.update.assert_any_call(self.uuid, self.patch)
@ -136,13 +157,12 @@ class Test(base.NodeTest):
self.cli.port.create.assert_called_once_with( self.cli.port.create.assert_called_once_with(
node_uuid=self.uuid, address='11:22:33:44:55:66') node_uuid=self.uuid, address='11:22:33:44:55:66')
status = client.get_status(self.uuid, auth_token='token') status = self.call_get_status(self.uuid)
self.assertEqual({'finished': True, 'error': None}, status) self.assertEqual({'finished': True, 'error': None}, status)
@mock.patch.object(utils, 'check_auth') @contextlib.contextmanager
@mock.patch.object(utils, 'get_client') def mocked_server():
def run(client_mock, keystone_mock):
d = tempfile.mkdtemp() d = tempfile.mkdtemp()
try: try:
conf_file = os.path.join(d, 'test.conf') conf_file = os.path.join(d, 'test.conf')
@ -150,6 +170,8 @@ def run(client_mock, keystone_mock):
with open(conf_file, 'wb') as fp: with open(conf_file, 'wb') as fp:
fp.write(CONF % {'db_file': db_file}) fp.write(CONF % {'db_file': db_file})
with mock.patch.object(utils, 'check_auth'):
with mock.patch.object(utils, 'get_client'):
eventlet.greenthread.spawn_n(main.main, eventlet.greenthread.spawn_n(main.main,
args=['--config-file', conf_file], args=['--config-file', conf_file],
in_functional_test=True) in_functional_test=True)
@ -165,14 +187,16 @@ def run(client_mock, keystone_mock):
eventlet.greenthread.sleep(3) eventlet.greenthread.sleep(3)
else: else:
break break
suite = unittest.TestLoader().loadTestsFromTestCase(Test) # start testing
res = unittest.TextTestRunner().run(suite) yield
# Make sure all processes finished executing # Make sure all processes finished executing
eventlet.greenthread.sleep(1) eventlet.greenthread.sleep(1)
return 0 if res.wasSuccessful() else 1
finally: finally:
shutil.rmtree(d) shutil.rmtree(d)
if __name__ == '__main__': if __name__ == '__main__':
sys.exit(run()) with mocked_server():
suite = unittest.TestLoader().loadTestsFromTestCase(Test)
res = unittest.TextTestRunner().run(suite)
sys.exit(0 if res.wasSuccessful else 1)

View File

@ -5,4 +5,3 @@ coverage>=3.6
doc8 # Apache-2.0 doc8 # Apache-2.0
hacking<0.11,>=0.10.0 hacking<0.11,>=0.10.0
mock>=1.2 mock>=1.2
python-ironic-inspector-client>=1.0.1