OpenStack Study: test_capabilities.py

**** CubicPower OpenStack Study ****

# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright 2014 Big Switch Networks, Inc.

# All Rights Reserved.

#

# 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.

#

# @author Kevin Benton

from contextlib import nested

import mock

from neutron.tests.unit.bigswitch import test_router_db

PLUGIN = 'neutron.plugins.bigswitch.plugin'

SERVERMANAGER = PLUGIN + '.servermanager'

SERVERPOOL = SERVERMANAGER + '.ServerPool'

SERVERRESTCALL = SERVERMANAGER + '.ServerProxy.rest_call'

**** CubicPower OpenStack Study ****

class CapabilitiesTests(test_router_db.RouterDBTestCase):

**** CubicPower OpenStack Study ****

    def test_floating_ip_capability(self):

        with nested(

            mock.patch(SERVERRESTCALL,

                       return_value=(200, None, None, '["floatingip"]')),

            mock.patch(SERVERPOOL + '.rest_create_floatingip',

                       return_value=(200, None, None, None)),

            mock.patch(SERVERPOOL + '.rest_delete_floatingip',

                       return_value=(200, None, None, None))

        ) as (mock_rest, mock_create, mock_delete):

            with self.floatingip_with_assoc() as fip:

                pass

            mock_create.assert_has_calls(

                [mock.call(fip['floatingip']['tenant_id'], fip['floatingip'])]

            )

            mock_delete.assert_has_calls(

                [mock.call(fip['floatingip']['tenant_id'],

                           fip['floatingip']['id'])]

            )

**** CubicPower OpenStack Study ****

    def test_floating_ip_capability_neg(self):

        with nested(

            mock.patch(SERVERRESTCALL,

                       return_value=(200, None, None, '[""]')),

            mock.patch(SERVERPOOL + '.rest_update_network',

                       return_value=(200, None, None, None))

        ) as (mock_rest, mock_netupdate):

            with self.floatingip_with_assoc() as fip:

                pass

            updates = [call[0][2]['floatingips']

                       for call in mock_netupdate.call_args_list]

            all_floats = [f['floating_ip_address']

                          for floats in updates for f in floats]

            self.assertIn(fip['floatingip']['floating_ip_address'], all_floats)