def testSetAttrIncrementsVersion(self):
curver = clusto.get_latest_version_number()
d = clusto.drivers.Driver('d1')
self.assertEqual(curver + 1, clusto.get_latest_version_number())
Attribute(d.entity, 'cat', 'baz')
clusto.begin_transaction()
SESSION.clusto_description = "TEST"
#Attribute(d.entity, 'foo', 'cat')
#SESSION.add(Attribute(d.entity, 'coo', 'daa'))
Attribute.query().all()
a = Attribute(d.entity, 'foo', 'bar')
clusto.commit()
#clusto.rollback_transaction()
#self.assertEqual(['bar', 'bar'], d.attr_values('foo'))
self.assert_('bar' in d.attr_values('foo'))
self.assertEqual(curver + 3, clusto.get_latest_version_number())
def testSetAttrIncrementsVersion(self):
curver = clusto.get_latest_version_number()
d = clusto.drivers.Driver('d1')
self.assertEqual(curver + 1, clusto.get_latest_version_number())
Attribute(d.entity, 'cat', 'baz')
clusto.begin_transaction()
SESSION.clusto_description = "TEST"
#Attribute(d.entity, 'foo', 'cat')
#SESSION.add(Attribute(d.entity, 'coo', 'daa'))
Attribute.query().all()
a = Attribute(d.entity, 'foo', 'bar')
clusto.commit()
#clusto.rollback_transaction()
#self.assertEqual(['bar', 'bar'], d.attr_values('foo'))
self.assert_('bar' in d.attr_values('foo'))
self.assertEqual(curver + 3, clusto.get_latest_version_number())
def testEmptyCommits(self):
server = clusto.drivers.BasicServer('s1')
curver = clusto.get_latest_version_number()
server.attrs()
self.assertEqual(curver, clusto.get_latest_version_number())
try:
clusto.begin_transaction()
SESSION.clusto_description = "TEST"
Entity.query().all()
clusto.commit()
except:
clusto.rollback_transaction()
self.assertEqual(curver, clusto.get_latest_version_number())
self.assertEqual([], server.attr_values('foo'))
def testEmptyCommits(self):
server = clusto.drivers.BasicServer('s1')
curver = clusto.get_latest_version_number()
server.attrs()
self.assertEqual(curver, clusto.get_latest_version_number())
try:
clusto.begin_transaction()
SESSION.clusto_description = "TEST"
Entity.query().all()
clusto.commit()
except:
clusto.rollback_transaction()
self.assertEqual(curver, clusto.get_latest_version_number())
self.assertEqual([], server.attr_values('foo'))
def testConcurrentThreads(self):
DB = testbase.DB
if DB.startswith('sqlite'):
return
conf = ConfigParser.ConfigParser()
conf.add_section('clusto')
conf.set('clusto', 'dsn', testbase.DB)
clusto.connect(conf, echo=testbase.ECHO)
clusto.init_clusto()
firstver = clusto.get_latest_version_number()
threadcount = 5
threads = []
barrier = barrier_creator(threadcount)
for i in range(threadcount):
threads.append(ClustoWorkThread(DB, testbase.ECHO, barrier))
for i in threads:
i.start()
for i in threads:
i.join()
self.assertEqual(clusto.get_latest_version_number(),
threadcount + firstver)
def testConcurrentThreads(self):
DB = testbase.DB
if DB.startswith('sqlite'):
return
conf = ConfigParser.ConfigParser()
conf.add_section('clusto')
conf.set('clusto', 'dsn', testbase.DB)
clusto.connect(conf, echo=testbase.ECHO)
clusto.init_clusto()
firstver = clusto.get_latest_version_number()
threadcount = 5
threads = []
barrier = barrier_creator(threadcount)
for i in range(threadcount):
threads.append(ClustoWorkThread(DB, testbase.ECHO,
barrier))
for i in threads:
i.start()
for i in threads:
i.join()
self.assertEqual(clusto.get_latest_version_number(),
threadcount+firstver)
def testVersionIncrementing(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
self.assertEqual(clusto.get_latest_version_number(), curver + 2)
def testVersionIncrementing(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
self.assertEqual(clusto.get_latest_version_number(), curver + 2)
def testVersionIncrementWithAttrs(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e1.add_attr('foo', 2)
self.assertEqual(clusto.get_latest_version_number(), curver + 3)
def testVersionIncrementWithAttrs(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e1.add_attr('foo', 2)
self.assertEqual(clusto.get_latest_version_number(), curver + 3)
def testEntityRename(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e1.add_attr('foo',1)
e1.add_attr('foo',2)
e1attrs = [a.to_tuple for a in e1.attrs]
midver = clusto.get_latest_version_number()
clusto.rename('e1', 't1')
postrenamever = clusto.get_latest_version_number()
t1 = clusto.get_by_name('t1')
self.assertEqual(sorted(e1attrs),
sorted([a.to_tuple for a in t1.entity.attrs]))
t1.del_attrs('foo', 2)
self.assertRaises(LookupError, clusto.get_by_name, 'e1')
self.assertEqual(sorted(t1.attrs('foo',1)),
sorted(t1.attrs()))
SESSION.clusto_version = midver
self.assertRaises(LookupError, clusto.get_by_name, 't1')
e = clusto.get_by_name('e1')
self.assertEqual(sorted(e1attrs),
sorted([a.to_tuple for a in e.attrs()]))
for a in e.attrs():
self.assertEqual(e.entity.deleted_at_version,
a.deleted_at_version)
SESSION.clusto_version = postrenamever
self.assertEqual(e.entity.deleted_at_version,
t1.entity.version)
for a in t1.attrs():
self.assertEqual(e.entity.deleted_at_version,
a.version)
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 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 testViewOldVersion(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e3 = Entity('e3')
self.assertEqual(Entity.query().filter(Entity.name.like('e%')).count(),
3)
SESSION.clusto_version = curver
self.assertEqual(Entity.query().filter(Entity.name.like('e%')).count(),
0)
SESSION.clusto_version = clusto.working_version()
self.assertEqual(Entity.query().filter(Entity.name.like('e%')).count(),
3)
SESSION.clusto_version = curver + 1
self.assertEqual(Entity.query().filter(Entity.name.like('e%')).count(),
1)
SESSION.clusto_version = curver + 2
self.assertEqual(sorted([e1,e2]),
Entity.query().filter(Entity.name.like('e%')).all())
def testViewOldVersion(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e3 = Entity('e3')
self.assertEqual(
Entity.query().filter(Entity.name.like(u'e%')).count(), 3)
SESSION.clusto_version = curver
self.assertEqual(
Entity.query().filter(Entity.name.like(u'e%')).count(), 0)
SESSION.clusto_version = clusto.working_version()
self.assertEqual(
Entity.query().filter(Entity.name.like(u'e%')).count(), 3)
SESSION.clusto_version = curver + 1
self.assertEqual(
Entity.query().filter(Entity.name.like(u'e%')).count(), 1)
SESSION.clusto_version = curver + 2
self.assertEqual(sorted([e1, e2]),
Entity.query().filter(Entity.name.like(u'e%')).all())
def testEntityRename(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e1.add_attr('foo', 1)
e1.add_attr('foo', 2)
e1attrs = [a.to_tuple for a in e1.attrs]
midver = clusto.get_latest_version_number()
clusto.rename('e1', 't1')
postrenamever = clusto.get_latest_version_number()
t1 = clusto.get_by_name('t1')
self.assertEqual(sorted(e1attrs),
sorted([a.to_tuple for a in t1.entity.attrs]))
t1.del_attrs('foo', 2)
self.assertRaises(LookupError, clusto.get_by_name, 'e1')
self.assertEqual(sorted(t1.attrs('foo', 1)), sorted(t1.attrs()))
SESSION.clusto_version = midver
self.assertRaises(LookupError, clusto.get_by_name, 't1')
e = clusto.get_by_name('e1')
self.assertEqual(sorted(e1attrs),
sorted([a.to_tuple for a in e.attrs()]))
for a in e.attrs():
self.assertEqual(e.entity.deleted_at_version, a.deleted_at_version)
SESSION.clusto_version = postrenamever
self.assertEqual(e.entity.deleted_at_version, t1.entity.version)
for a in t1.attrs():
self.assertEqual(e.entity.deleted_at_version, a.version)
def testCreateEntityObject(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
res = Entity.query().filter_by(name='e1')
self.assertEqual(res.count(),1)
e = res.all()[0]
self.assertEqual(e.name, 'e1')
def testCreateEntityObject(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
res = Entity.query().filter_by(name=u'e1')
self.assertEqual(res.count(),1)
e = res.all()[0]
self.assertEqual(e.name, 'e1')
def testDeleteVersion(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
etest = clusto.get_by_name('e1')
e1.delete()
self.assertRaises(LookupError, clusto.get_by_name, 'e1')
e1a = Entity('e1')
etest = clusto.get_by_name('e1')
self.assertEqual(etest.entity.version, curver + 3)
def testDeleteVersion(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
etest = clusto.get_by_name('e1')
e1.delete()
self.assertRaises(LookupError, clusto.get_by_name, 'e1')
e1a = Entity('e1')
etest = clusto.get_by_name('e1')
self.assertEqual(etest.entity.version, curver+3)
def testOldVersionsOfAttributes(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e1.add_attr('foo', 1)
e1.add_attr('foo2', 2)
e1.add_attr('foo3', 3)
SESSION.clusto_version = curver + 3
self.assertEqual(len(list(e1.attrs)), 1)
e = Entity.query().filter_by(name=u'e1').one()
SESSION.clusto_version = curver + 4
self.assertEqual(sorted([a.key for a in e.attrs]),
sorted(['foo', 'foo2']))
def testPoolRename(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
p1 = clusto.drivers.Pool('p1')
p1.insert(e1)
p1.insert(e2)
self.assertEqual(sorted([e1, e2]),
sorted((d.entity for d in p1.contents())))
clusto.rename('p1', 'p1renamed')
p1renamed = clusto.get_by_name('p1renamed')
self.assertEqual(sorted([e1, e2]),
sorted((d.entity for d in p1renamed.contents())))
def testOldVersionsOfAttributes(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
e1.add_attr('foo', 1)
e1.add_attr('foo2', 2)
e1.add_attr('foo3', 3)
SESSION.clusto_version = curver + 3
self.assertEqual(len(list(e1.attrs)), 1)
e = Entity.query().filter_by(name='e1').one()
SESSION.clusto_version = curver + 4
self.assertEqual(sorted([a.key for a in e.attrs]),
sorted(['foo', 'foo2']))
def testPoolRename(self):
curver = clusto.get_latest_version_number()
e1 = Entity('e1')
e2 = Entity('e2')
p1 = clusto.drivers.Pool('p1')
p1.insert(e1)
p1.insert(e2)
self.assertEqual(sorted([e1,e2]),
sorted((d.entity for d in p1.contents())))
clusto.rename('p1', 'p1renamed')
p1renamed = clusto.get_by_name('p1renamed')
self.assertEqual(sorted([e1,e2]),
sorted((d.entity for d in p1renamed.contents())))
def testGetFirstVersionNumber(self):
curver = clusto.get_latest_version_number()
self.assertEqual(curver, 2)
def testGetFirstVersionNumber(self):
curver = clusto.get_latest_version_number()
self.assertEqual(curver, 2)
