def Test_DAGStructureMultipleUsers(self):
#tags for the first user...
u1 = MyUser(user="******", key_name="Bill")
u1.put()
u2 = MyUser(user="******", key_name="Ted")
u2.put()
# Bill's tags
t1 = TagVertex(parent = u1, key_name="Tag1")
t1.ttlCount = 1
t1.put()
t2 = TagVertex(parent = u1, key_name="Tag2")
t2.ttlCount = 2
t2.put()
t3 = TagVertex(parent = u1, key_name="Tag3")
t3.ttlCount = 3
t3.put()
t4 = TagVertex(parent = u1, key_name="Tag4")
t4.ttlCount = 4
t4.put()
t5 = TagVertex(parent = u1, key_name="Tag5")
t5.ttlCount = 5
t5.put()
# Ted's tags
tt1 = TagVertex(parent = u2, key_name="Tag1")
tt1.ttlCount = 11
tt1.put()
tt2 = TagVertex(parent = u2, key_name="Tag2")
tt2.ttlCount = 12
tt2.put()
tt3 = TagVertex(parent = u2, key_name="Tag3")
tt3.ttlCount = 13
tt3.put()
tt4 = TagVertex(parent = u2, key_name="Tag4")
tt4.ttlCount = 14
tt4.put()
tt5 = TagVertex(parent = u2, key_name="Tag5")
tt5.ttlCount = 15
tt5.put()
#extract Bill's tags
q1 = TagVertex.all()
q1.ancestor(u1.key())
results = q1.fetch(10)
check.ok_(len(results) == 5)
for v in results:
check.ok_(v.ttlCount > 0 and v.ttlCount < 11)
#endfor
q2 = TagVertex.all()
q2.ancestor(u2.key())
results = q2.fetch(10)
check.ok_(len(results) == 5)
for v in results:
check.ok_(v.ttlCount > 10 and v.ttlCount < 16)
def tearDown(self):
_current_session = None
self.testbed.deactivate()
allEdges = []
edgeQuery = TagEdge.all()
for edge in edgeQuery:
allEdges.append(edge.key())
#end for
db.delete(allEdges)
allTags = []
tagQuery = TagVertex.all()
for tag in tagQuery:
allTags.append(tag.key())
#end for
db.delete(allTags)
allUsers = []
userQuery = MyUser.all()
for user in userQuery:
allUsers.append(user.key())
#end for
db.delete(allUsers)
def tearDown(self):
_current_session = None
self.testbed.deactivate()
allEdges = []
edgeQuery = TagEdge.all()
for edge in edgeQuery:
allEdges.append(edge.key())
#end for
db.delete(allEdges)
allTags = []
tagQuery = TagVertex.all()
for tag in tagQuery:
allTags.append(tag.key())
#end for
db.delete(allTags)
allUsers = []
userQuery = MyUser.all()
for user in userQuery:
allUsers.append(user.key())
#end for
db.delete(allUsers)
def setUp(self):
#check that the db contents really have been deleted
tagQuery = TagVertex.all()
tags = tagQuery.fetch(100)
check.ok_(len(tags) == 0)
edgeQuery = TagEdge.all()
edges = edgeQuery.fetch(100)
check.ok_(len(edges) == 0)
userQuery = MyUser.all()
users = userQuery.fetch(100)
check.ok_(len(users) == 0)
def tearDown(self):
allEdges = []
edgeQuery = TagEdge.all()
for edge in edgeQuery:
allEdges.append(edge.key())
#end for
db.delete(allEdges)
allTags = []
tagQuery = TagVertex.all()
for tag in tagQuery:
allTags.append(tag.key())
#end for
db.delete(allTags)
allUsers = []
userQuery = MyUser.all()
for user in userQuery:
allUsers.append(user.key())
#end for
db.delete(allUsers)
