Source code for ironicclient.tests.unit.osc.v1.test_baremetal_driver

#  Copyright (c) 2016 Mirantis, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#

import copy

from osc_lib.tests import utils as oscutils

from ironicclient.osc.v1 import baremetal_driver
from ironicclient.tests.unit.osc.v1 import fakes as baremetal_fakes


[docs]class TestBaremetalDriver(baremetal_fakes.TestBaremetal):
[docs] def setUp(self): super(TestBaremetalDriver, self).setUp() self.baremetal_mock = self.app.client_manager.baremetal self.baremetal_mock.reset_mock()
[docs]class TestListBaremetalDriver(TestBaremetalDriver):
[docs] def setUp(self): super(TestListBaremetalDriver, self).setUp() self.baremetal_mock.driver.list.return_value = [ baremetal_fakes.FakeBaremetalResource( None, copy.deepcopy(baremetal_fakes.BAREMETAL_DRIVER), loaded=True) ] self.cmd = baremetal_driver.ListBaremetalDriver(self.app, None)
[docs] def test_baremetal_driver_list(self): arglist = [] verifylist = [] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) collist = ( "Supported driver(s)", "Active host(s)") self.assertEqual(collist, tuple(columns)) datalist = (( baremetal_fakes.baremetal_driver_name, ', '.join(baremetal_fakes.baremetal_driver_hosts)), ) self.assertEqual(datalist, tuple(data))
[docs]class TestPassthruCallBaremetalDriver(TestBaremetalDriver):
[docs] def setUp(self): super(TestPassthruCallBaremetalDriver, self).setUp() self.baremetal_mock.driver.vendor_passthru.return_value = ( baremetal_fakes.BAREMETAL_DRIVER_PASSTHRU ) self.cmd = baremetal_driver.PassthruCallBaremetalDriver(self.app, None)
[docs] def test_baremetal_driver_passthru_call_with_min_args(self): arglist = [ baremetal_fakes.baremetal_driver_name, baremetal_fakes.baremetal_driver_passthru_method, ] verifylist = [ ('driver', baremetal_fakes.baremetal_driver_name), ('method', baremetal_fakes.baremetal_driver_passthru_method), ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) # Set expected values args = [ baremetal_fakes.baremetal_driver_name, baremetal_fakes.baremetal_driver_passthru_method, ] kwargs = { 'http_method': 'POST', 'args': {} } (self.baremetal_mock.driver.vendor_passthru. assert_called_once_with(*args, **kwargs))
[docs] def test_baremetal_driver_passthru_call_with_all_args(self): arglist = [ baremetal_fakes.baremetal_driver_name, baremetal_fakes.baremetal_driver_passthru_method, '--arg', 'arg1=val1', '--arg', 'arg2=val2', '--http-method', 'POST' ] verifylist = [ ('driver', baremetal_fakes.baremetal_driver_name), ('method', baremetal_fakes.baremetal_driver_passthru_method), ('arg', ['arg1=val1', 'arg2=val2']), ('http_method', 'POST') ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) # Set expected values args = [ baremetal_fakes.baremetal_driver_name, baremetal_fakes.baremetal_driver_passthru_method, ] kwargs = { 'http_method': 'POST', 'args': {'arg1': 'val1', 'arg2': 'val2'} } (self.baremetal_mock.driver.vendor_passthru. assert_called_once_with(*args, **kwargs))
[docs] def test_baremetal_driver_passthru_call_no_arg(self): arglist = [] verifylist = [] self.assertRaises(oscutils.ParserException, self.check_parser, self.cmd, arglist, verifylist)
[docs]class TestPassthruListBaremetalDriver(TestBaremetalDriver):
[docs] def setUp(self): super(TestPassthruListBaremetalDriver, self).setUp() self.baremetal_mock.driver.get_vendor_passthru_methods.return_value = ( baremetal_fakes.BAREMETAL_DRIVER_PASSTHRU ) self.cmd = baremetal_driver.PassthruListBaremetalDriver(self.app, None)
[docs] def test_baremetal_driver_passthru_list(self): arglist = ['fakedrivername'] verifylist = [] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) args = ['fakedrivername'] (self.baremetal_mock.driver.get_vendor_passthru_methods. assert_called_with(*args)) collist = ( "Name", "Supported HTTP methods", "Async", "Description", "Response is attachment", ) self.assertEqual(collist, tuple(columns)) datalist = (('lookup', 'POST', 'false', '', 'false'),) self.assertEqual(datalist, tuple(data))
[docs] def test_baremetal_driver_passthru_list_no_arg(self): arglist = [] verifylist = [] self.assertRaises(oscutils.ParserException, self.check_parser, self.cmd, arglist, verifylist)
[docs]class TestShowBaremetalDriver(TestBaremetalDriver):
[docs] def setUp(self): super(TestShowBaremetalDriver, self).setUp() self.baremetal_mock.driver.get.return_value = ( baremetal_fakes.FakeBaremetalResource( None, copy.deepcopy(baremetal_fakes.BAREMETAL_DRIVER), loaded=True)) self.cmd = baremetal_driver.ShowBaremetalDriver(self.app, None)
[docs] def test_baremetal_driver_show(self): arglist = ['fakedrivername'] verifylist = [] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) args = ['fakedrivername'] self.baremetal_mock.driver.get.assert_called_with(*args) self.assertFalse(self.baremetal_mock.driver.properties.called) collist = ('hosts', 'name') self.assertEqual(collist, columns) datalist = ( ', '.join(baremetal_fakes.baremetal_driver_hosts), baremetal_fakes.baremetal_driver_name) self.assertEqual(datalist, tuple(data))
[docs] def test_baremetal_driver_show_no_arg(self): arglist = [] verifylist = [] self.assertRaises(oscutils.ParserException, self.check_parser, self.cmd, arglist, verifylist)

Project Source