def testParents(self):
class OtherDriver(Driver):
_clusto_type = 'otherdriver'
_driver_name = 'otherdriver'
p1 = Pool('toppool')
d1 = Pool('grandparent')
d1a = Pool('othergrandparent')
d1b = OtherDriver('anothergrandparent')
d2 = Driver('parent')
d3 = Driver('child')
d1b.add_attr('foo', 'someval')
p1.insert(d1a)
d1b.insert(d2)
d1a.insert(d2)
d1.insert(d2)
d2.insert(d3)
self.assertEqual(
sorted([d1, d1a, p1]),
sorted(d3.parents(clusto_types=[Pool], search_parents=True)))
self.assertEqual([d2], d3.parents())
self.assertEqual(['someval'],
d3.attr_values('foo', merge_container_attrs=True))
def testSetAttrs(self):
d1 = Driver('d1')
d1.set_attr('foo', 'bar')
self.assertEqual(d1.attr_items(), [(('foo', None, None), 'bar')])
d1.set_attr('foo', 'bar2')
self.assertEqual(d1.attr_items(), [(('foo', None, None), 'bar2')])
d1.add_attr('foo', 'bar3')
self.assertEqual(
sorted(d1.attr_items()),
sorted([(('foo', None, None), 'bar2'),
(('foo', None, None), 'bar3')]))
self.assertRaises(DriverException, d1.set_attr, 'foo', 'bar4')
d2 = Driver('d2')
d2.add_attr('a', number=0, subkey='foo', value='bar1')
d2.add_attr('a', number=1, subkey='foo', value='bar1')
d2.add_attr('a', number=2, subkey='foo', value='bar1')
d2.set_attr('a', 't1')
self.assertEqual(
sorted(d2.attr_items()),
sorted([(('a', 0, 'foo'), 'bar1'), (('a', 1, 'foo'), 'bar1'),
(('a', 2, 'foo'), 'bar1'), (('a', None, None), 't1')]))
def get_ip_manager(cls, ip):
"""return a valid ip manager for the given ip.
@param ip: the ip
@type ip: integer, string, or IPy object
@return: the appropriate IP manager from the clusto database
"""
ipman = None
if isinstance(ip, Attribute):
ipman = ip.entity
return Driver(ipman)
for ipmantest in clusto.get_entities(clusto_drivers=[cls]):
try:
ipmantest.ensure_type(ip)
except ResourceTypeException:
continue
ipman = Driver(ipmantest)
break
if not ipman:
raise ResourceException(u"No resource manager for %s exists." %
str(ip))
return ipman
def testDriverSets(self):
d1 = Driver('d1')
d2 = Driver('d2')
s = set([d1, d1, d2])
self.assertEquals(len(s), 2)
def testContents(self):
d1 = Driver('d1')
d2 = Driver('d2')
d1.insert(d2)
self.assertEqual(d1.contents(), [d2])
def testMultipleInserts(self):
d1 = Driver('d1')
d2 = Driver('d2')
d3 = Driver('d3')
d1.insert(d2)
self.assertRaises(TypeError, d3.insert, d2)
def testInsert(self):
d1 = Driver('d1')
d2 = Driver('d2')
d1.insert(d2)
clusto.flush()
d = clusto.get_by_name('d1')
self.assertEqual(d.attr_items(ignore_hidden=False),
[(('_contains', 0, None), d2)])
def testChildrenContents(self):
p1 = Pool('p1')
p2 = Pool('p2')
d1 = Driver('d1')
d2 = Driver('d2')
p1.insert(d1)
p2.insert(d2)
p2.insert(p1)
self.assertEqual(sorted([p1, d1, d2]),
sorted(p2.contents(search_children=True)))
def testGettingAttrValues(self):
d1 = Driver('d1')
d2 = Driver('d2')
d1.add_attr('foo', 'bar')
d1.add_attr('foo0', 'bar1')
d2.add_attr('d1', d1)
clusto.flush()
self.assertEqual(sorted(['bar', 'bar1']),
sorted(d1.attr_values('foo.*', regex=True)))
self.assertEqual([d1], d2.attr_values())
def testGettingAttrsMultipleTimes(self):
d1 = Driver('d1')
d2 = Driver('d2')
d1.add_attr('foo', 'bar')
d1.add_attr('foo0', 'bar1')
d2.add_attr('d1', d1)
clusto.flush()
d = clusto.get_by_name('d1')
self.assertEqual(len(d.references()), 1)
self.assertEqual(len(d.attrs()), 2)
def get_pools(cls, obj, allPools=True):
d = cls.ensure_driver(obj, "obj must be either an Entity or a Driver.")
pools = [
Driver(a.entity) for a in d.parents()
if isinstance(Driver(a.entity), Pool)
]
if allPools:
for i in pools:
pools.extend(Pool.get_pools(i, allPools=True))
return pools
def testRemove(self):
d1 = Driver('d1')
d2 = Driver('d2')
d1.insert(d2)
clusto.flush()
d = clusto.get_by_name('d1')
d.remove(d2)
clusto.flush()
self.assertEqual(d.attr_items(ignore_hidden=False), [])
def testGetByAttr(self):
d1 = Driver('d1')
d1.add_attr('foo', 1)
d2 = Driver('d2')
d2.add_attr('foo', 2)
d3 = Driver('d3')
d3.add_attr('bar', 3)
clusto.flush()
result = Driver.get_by_attr('foo', 2)
self.assertEqual(result, [d2])
def get_rack_and_u(cls, device):
"""
Get the rack and rackU for a given device.
returns a tuple of (rack, u-number)
"""
rack = set(device.parents(clusto_types=[cls]))
if len(rack) > 1:
raise Exception(
"%s is somehow in more than one rack, this will "
"likely need to be rectified manually. It currently "
"appears to be in racks %s" % (device.name, str(rack)))
if rack:
rack = rack.pop()
return {
'rack':
Driver(rack.entity),
'RU': [
x.number
for x in rack.content_attrs(value=device, subkey='ru')
]
}
else:
return None
def testMultipleGrandchildrenContents(self):
p1 = Pool('p1')
p2 = Pool('p2')
d1 = Driver('d1')
d2 = Driver('d2')
d3 = Driver('d3')
p2.insert(d1)
p2.insert(d2)
p2.insert(d3)
p1.insert(p2)
self.assertEqual(sorted([p2, d1, d2, d3]),
sorted(p1.contents(search_children=True)))
def testCreatingDriverWithUsedName(self):
d1 = Driver('d1')
self.assertRaises(NameException, Driver, 'd1')
d1.attrs()
def get_resource_manager(cls, resource_attr):
"""Return the resource manager for a given resource_attr"""
thing = Driver(resource_attr.entity)
return thing.attr_value(key=resource_attr.key,
subkey='manager',
number=resource_attr.number)
def testNumberedAttrsWithSubKeys(self):
d1 = Driver('d1')
d1.add_attr(key='foo', value='bar1', number=True, subkey='one')
d1.add_attr(key='foo', value='bar2', number=True, subkey='two')
self.assertEqual(d1.attr_items(), [(('foo', 0, 'one'), 'bar1'),
(('foo', 1, 'two'), 'bar2')])
def testAttributeGetValueAfterAdd(self):
d1 = Driver('d1')
d1.add_attr('foo', 2)
self.assertEqual(d1.attr_items('foo'), [(('foo', None, None), 2)])
d1.add_attr('bar', 3)
self.assertEqual(d1.attr_items('foo'), [(('foo', None, None), 2)])
self.assertEqual(d1.attr_items('bar'), [(('bar', None, None), 3)])
def testSubKeyAttrs(self):
d1 = Driver('d1')
d1.add_attr('foo', 'bar', subkey='subfoo')
d1.add_attr('foo', 'caz', subkey='subbar')
self.assertEqual(
sorted(d1.attr_key_tuples()),
sorted([('foo', None, 'subfoo'), ('foo', None, 'subbar')]))
def get_ip_managers(cls, ip):
"""return a list of valid ip managers for the given ip.
@param ip: the ip
@type ip: integer, string, or IPy object
@return: a list of IP managers from the clusto database
"""
ipman = None
if isinstance(ip, Attribute):
ipman = ip.entity
return Driver(ipman)
ipmanagers = []
for ipman in clusto.get_entities(clusto_drivers=[cls]):
try:
ipman.ensure_type(ip)
ipmanagers.append(Driver(ipman))
except ResourceTypeException:
continue
return ipmanagers
def testHasAttr(self):
d1 = Driver('d1')
d1.add_attr(key='foo', value='bar1', number=True, subkey='one')
d1.add_attr(key='foo', value='bar2', number=True, subkey='two')
d1.add_attr(key='foo', value='bar3', number=True, subkey='three')
d1.add_attr(key='foo', value='bar4', number=True, subkey='four')
self.assertFalse(d1.has_attr(key='foo', number=False))
self.assertTrue(d1.has_attr(key='foo', number=True))
self.assertTrue(d1.has_attr(key='foo', number=1, subkey='two'))
def testHiddenAttrs(self):
d1 = Driver('d1')
d1.add_attr(key='foo', value='bar1', number=True, subkey='one')
d1.add_attr(key='foo', value='bar2', number=True, subkey='two')
d1.add_attr(key='_foo', value='bar3', number=True, subkey='three')
d1.add_attr(key='_foo', value='bar4', number=True, subkey='four')
self.assertEqual(d1.attr_items(ignore_hidden=True),
[(('foo', 0, 'one'), 'bar1'),
(('foo', 1, 'two'), 'bar2')])
def testGettingAttrs(self):
d1 = Driver('d1')
d1.add_attr('foo', 'bar')
d1.add_attr('foo', 'bar1', number=0)
self.assertEqual(sorted(d1.attr_items()),
[(('foo', None, None), 'bar'),
(('foo', 0, None), 'bar1')])
self.assertEqual(d1.attr_items(number=True),
[(('foo', 0, None), 'bar1')])
def testSetAttrAlreadySet(self):
d1 = Driver('d1')
version = clusto.get_latest_version_number()
d1.set_attr(key='foo', value='bar1')
self.assertEqual(version + 1, clusto.get_latest_version_number())
d1.set_attr(key='foo', value='bar1')
self.assertEqual(version + 1, clusto.get_latest_version_number())
def testGettingAttrsWithSpecificValues(self):
d1 = Driver('d1')
d1.add_attr(key='foo', value='bar1', number=True, subkey='one')
d1.add_attr(key='foo', value='bar2', number=True, subkey='two')
d1.add_attr(key='foo', value='bar3', number=True, subkey='three')
d1.add_attr(key='foo', value='bar4', number=True, subkey='four')
self.assertEqual(list(d1.attr_items(value='bar3')),
[(('foo', 2, 'three'), 'bar3')])
self.assertEqual(list(d1.attr_items(value='bar1')),
[(('foo', 0, 'one'), 'bar1')])
def data(self):
d1 = Driver('d1')
d2 = Driver('d2')
d3 = Driver('d3')
d1.add_attr('_foo', 'bar1')
d1.add_attr('car', 'baz')
d1.add_attr('car', 'baz')
d1.add_attr('d', 'dee', number=True)
d1.add_attr('d', 'dee', number=True)
d1.add_attr('a', 1)
d1.add_attr('a', 1, subkey='t')
d1.add_attr('a', 1, subkey='g')
d1.add_attr('a', 1, subkey='z', number=4)
d1.add_attr('a', 1, subkey='z', number=5)
d1.add_attr('a', 1, subkey='z', number=6)
d1.set_attr('d2', d2)
d1.set_attr('d3', d3)
d2.set_attr('aaa', 1)
d2.set_attr('aab', 2)
d2.set_attr('aac', 3)
def testDelAttrs(self):
d1 = Driver('d1')
d1.add_attr(key='foo', value='bar1', number=True, subkey='one')
d1.add_attr(key='foo', value='bar2', number=True, subkey='two')
d1.add_attr(key='foo', value='bar3', number=True, subkey='three')
d1.add_attr(key='foo', value='bar4', number=True, subkey='four')
d1.del_attrs(key='foo', value='bar4')
self.assertEqual(list(d1.attr_items(value='bar4')), [])
self.assertEqual(list(d1.attr_items(value='bar3')),
[(('foo', 2, 'three'), 'bar3')])
d1.del_attrs(key='foo', subkey='three', number=2)
self.assertEqual(list(d1.attr_items(value='bar3')), [])
def testNumberedInserts(self):
d1 = Driver('d1')
d1.insert(Driver('d2'))
d1.insert(Driver('d3'))
d1.insert(Driver('d4'))
d1.insert(Driver('d5'))
d1.insert(Driver('d6'))
self.assertEqual(range(5),
[x.number for x in d1.attrs(ignore_hidden=False)])
def resources(cls, thing):
"""return a list of resources from the resource manager that is
associated with the given thing.
A resource is a resource attribute in a resource manager.
"""
attrs = [x for x in thing.attrs(cls._attr_name, subkey='manager')
if isinstance(Driver(x.value), cls)]
res = []
for attr in attrs:
t=thing.attrs(cls._attr_name, number=attr.number, subkey=None)
res.extend(t)
return res
