def test_get_resource_by_attr(self):
class TestClass(object):
def __init__(self, val):
self.val = val
obj1 = TestClass('foo')
obj2 = TestClass(1)
obj3 = TestClass({})
arr = [obj1, obj2, obj3]
self.assertIs(
base.CloudDriver()._get_resource_by_attr(lambda: arr, 'val',
'foo'), obj1)
self.assertIs(
base.CloudDriver()._get_resource_by_attr(lambda: arr, 'val', 1),
obj2)
self.assertIs(
base.CloudDriver()._get_resource_by_attr(lambda: arr, 'val', {}),
obj3)
self.assertRaises(AttributeError,
base.CloudDriver()._get_resource_by_attr,
lambda: arr, 'blah', {})
self.assertRaises(IndexError,
base.CloudDriver()._get_resource_by_attr,
lambda: arr, 'val', 1234)
def test_delete_security_group_rule(self):
sg = models.SecurityGroup(name='webapp')
sgr = models.SecurityGroupRule(security_group=sg,
source_ip='0.0.0.0/0',
from_port=443,
to_port=443,
protocol='tcp')
base.CloudDriver().delete_security_group_rule(sgr)
def test_delete_security_group(self):
sg = models.SecurityGroup(name='webapp')
base.CloudDriver().delete_security_group(sg)
def test_init_sets_cluster(self):
driver = base.CloudDriver(cluster='https://example.com/testcluster')
self.assertIs(type(driver.cluster), aasemble.client.Cluster)
def setUp(self):
super(CloudDriverTests, self).setUp()
self.driver = base.CloudDriver()