예제 #1
0
파일: test.py 프로젝트: g2p/tranquil
	def model_test(self):
		self.clear(Choice)
		self.clear(Poll)
		sess = Session()
		green = Poll(question='Do you like green eggs and ham?',pub_date=datetime.datetime.today())
		green.choice_set.append( Choice( choice='Yes', votes=0 ) )
		green.choice_set.append( Choice( choice='No', votes=0 ) )
		sess.save( green )
		sess.commit()
		sess = Session()
		polls = sess.query(Poll).all()
		self.assertEqual( len( polls ), 1 )
		self.assertEqual( polls[0].question, 'Do you like green eggs and ham?' )
		self.assertEqual( len( polls[0].choice_set ), 2 )
		self.assert_( polls[0].choice_set[0].choice in ['Yes', 'No'] )
		self.assert_( polls[0].choice_set[1].choice in ['Yes', 'No'] )
예제 #2
0
파일: test.py 프로젝트: g2p/tranquil
	def self_ref_test(self):
		self.clear(SelfRef)
		sess = Session()
		root = SelfRef(name='root')
		sess.save( root )
		c1 = SelfRef(name='c1',parent=root)
		sess.save( c1 )
		sess.commit()
		sess = Session()
		root = sess.query(SelfRef).filter(SelfRef.c.id==1).one()
		self.assertEqual( root.name, 'root' )
		self.assertEqual( len( root.selfref_set ), 1 )
		self.assertEqual( root.parent, None )
		child = root.selfref_set[0]
		self.assertEqual( child.name, 'c1' )
		self.assertEqual( child.parent, root )
		self.assertEqual( child.selfref_set, [] )
예제 #3
0
파일: test.py 프로젝트: g2p/tranquil
	def clear(self,klass):
		sess = Session()
		for k in sess.query(klass).all():
			sess.delete(k)
		sess.commit()