Files
python-blazarclient/blazarclient/tests/test_plugin.py
Mark Powers f2277b5f7a Add openstackclient support
This adds support for the recommended CLI using the OpenStackClient,
without modifying the existing Blazar shell CLI.

The existing shell command classes are used, by introducing a check in
the base comand class to use either the client passed by Blazar shell,
or the client using the osc_lib client_manager.

The argument --physical-reservation is also removed for the create lease
command when using the OpenStack client.

Implements: blueprint openstackclient-support
Change-Id: I97a7b91f0d05efc887307ac167e5c368276d4f81
2021-08-05 10:24:28 -05:00

37 lines
1.2 KiB
Python

# 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.
from unittest import mock
from blazarclient.osc import plugin
from blazarclient import tests
class ReservationPluginTests(tests.TestCase):
@mock.patch("blazarclient.v1.client.Client")
def test_make_client(self, mock_client):
instance = mock.Mock()
instance._api_version = {"reservation": "1"}
endpoint = "blazar_endpoint"
instance.get_endpoint_for_service_type = mock.Mock(
return_value=endpoint
)
plugin.make_client(instance)
mock_client.assert_called_with(
"1",
session=instance.session,
endpoint_override=endpoint
)