def student_swapblock(student_id):
wants_json = request_wants_json()
try:
swapblock = api.get_student_swapblock(student_id, json=wants_json)
except api.DbNotFoundError:
abort(404)
return jsonify({'swapblock': swapblock})
def test_student_swapblock(self):
#initial setup
student = Student(name='test', umail_address='test', facebook_id='test', fb_auth_token='test',
fb_profile_link='test', fb_picture_link='test')
swap_block = Swapblock(student)
db.session.add(student)
db.session.add(swap_block)
db.session.commit()
test_block = api.get_student_swapblock(student.id)
self.assertEqual(test_block.student.name, swap_block.student.name)
self.assertRaises(api.DbNotFoundError, api.get_student_swapblock, 10000)
