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()