def test_default_id(self):
a = models.Account()
vtype = models.VolumeType('vtype')
v = models.Volume(account=a, volume_type_name=vtype.name)
self.db.add_all([a, vtype, v])
self.db.commit()
self.assert_(v.id)
def test_name(self):
a = models.Account()
vtype = models.VolumeType('vtype')
name = 'mylittlevolume'
v = models.Volume(account=a, volume_type_name=vtype.name, name=name)
self.db.add_all([a, vtype, v])
self.db.commit()
self.assert_(v.id)
self.assertEquals(v.name, name)
def test_create(self):
vt = models.VolumeType('foo')
expected = {
'name': 'foo',
'status': 'ACTIVE',
'min_size': 1,
'max_size': 1024,
}
self.db.add(vt)
self.db.commit()
for k, v in expected.items():
attr = getattr(vt, k)
err_msg = '%s != %s (%s)' % (v, attr, k)
self.assertEquals(v, attr, err_msg)
def test_active_backup_count(self):
vt = models.VolumeType('lunr')
a = models.Account()
n = models.Node('lunr',
10,
volume_type=vt,
hostname='127.0.0.1',
port=8080)
v = models.Volume(account=a, size=1, volume_type=vt, node=n)
b1 = models.Backup(volume=v, status='AVAILABLE')
b2 = models.Backup(volume=v, status='AVAILABLE')
b3 = models.Backup(volume=v, status='NOTAVAILABLE')
b4 = models.Backup(volume=v, status='SOMETHING')
b5 = models.Backup(volume=v, status='AVAILABLE')
b6 = models.Backup(volume=v, status='AUDITING')
b7 = models.Backup(volume=v, status='DELETED')
self.db.add_all([a, n, v, b1, b2, b3, b4, b5, b6, b7])
self.db.commit()
self.db.refresh(v)
self.assertEquals(5, v.active_backup_count())
def setUp(self):
ModelTest.setUp(self)
self.volume_type = models.VolumeType('lunr')
self.default_volume_type = models.VolumeType('vtype')
self.db.add_all([self.volume_type, self.default_volume_type])
self.db.commit()
