052b7fe901
Adds gateway as an optional paramter to the ip block creation Change-Id: Ie6c80a0712b8dd6f48840efab7af2e18b6cedbf9
55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2011 OpenStack LLC.
|
|
# 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.
|
|
|
|
import mock
|
|
|
|
from melange.client import ipam_client
|
|
from melange.client import tests
|
|
|
|
|
|
class TestFactory(tests.BaseTest):
|
|
|
|
def test_factory_gives_client(self):
|
|
factory = ipam_client.Factory("host", "8080")
|
|
|
|
self.assertEquals(ipam_client.IpBlockClient, type(factory.ip_block))
|
|
|
|
def test_factory_raises_attribute_error_for_non_existent_client(self):
|
|
factory = ipam_client.Factory("host", "8080")
|
|
|
|
self.assertRaisesExcMessage(AttributeError,
|
|
"Factory has no attribute "
|
|
"non_existent_client",
|
|
lambda: factory.non_existent_client)
|
|
|
|
|
|
class TestIpBlockCreate(tests.BaseTest):
|
|
|
|
def test_create_ip_block_with_gateway(self):
|
|
factory = ipam_client.Factory("host", "8080")
|
|
client = ipam_client.IpBlockClient(factory._client,
|
|
factory._auth_client, 'fake')
|
|
with mock.patch('melange.client.ipam_client.Resource.create') as patch:
|
|
client.create(type='public', cidr='10.0.0.0/24',
|
|
gateway='10.0.0.1')
|
|
patch.assert_called_with(type='public',
|
|
cidr='10.0.0.0/24',
|
|
network_id=None,
|
|
policy_id=None,
|
|
gateway='10.0.0.1')
|