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)