Python User.take示例

示例#1

文件： tests.py 项目： Steve-Andrew-Lynch/dartmouth-major-planner

	def test_take(self):
		u1 = User(full_name = 'Alex', netid = 'd302304')
		u2 = User(full_name = 'Santiago', netid = 'd30st04')
		db.session.add(u1)
		db.session.add(u2)
		db.session.commit()

		d1 = Department(name = "Dept 1", abbr = "DEPT")
		db.session.add(d1)
		db.session.commit()

		c1 = Course(name = 'Course', number = "1", department = d1.id)
		db.session.add(c1)
		db.session.commit()

		t1 = Term(year = 2013, season = "F")
		db.session.add(t1)
		db.session.commit()

		h1 = Hour(period = "3A")
		db.session.add(h1)
		db.session.commit()

		o1 = Offering(course = c1.id, term = t1.id, hour = h1.id)
		db.session.add(o1)
		db.session.commit()

		u = u1.take(o1)
		db.session.add(u)
		db.session.commit()
		assert u1.take(o1) == None
		assert u1.is_taking(o1)