使用Flask app.test_client进行单元测试时出现SQLAlchemy DetachedInstanceError

时间:2019-06-16 23:22:32

标签: python flask sqlalchemy python-unittest

我的目标是能够使用self.client.post(url, ...通过Flask应用程序test_client测试路由。我遇到的问题是,无论何时使用该请求发出请求,我都会从SQLAlchemy中遇到DetachedInstanceError,说不会发生刷新或延迟加载操作。

这是我的设置:

class BaseTestCase(unittest.TestCase):

   def setUp(self):
      self.postgresql = Postgresql()
      config = {
         'FLASK_ENV': 'test',
         'SQLALCHEMY_DATABASE_URI': self.postgresql.url()
      }
      with patch.dict('os.environ', config):
         self.app = create_app()
         self.client = self.app.test_client()
         ctx = self.app.app_context()
         ctx.push()
         db.create_all()
         self.assertEqual(app.testing, True)
      reset_fake_data()

   def tearDown(self):
      db.session.remove()
      self.postgresql.stop()

   def post(self, url, data={}, headers={}):
      headers = {'Content-Type': 'application/json', **headers}
      return self.client.post(url, headers=headers, data=json.dumps(data))

   def post_with_auth_token(self, url, data={}, headers={}):
      headers = {'Authorization': f'Bearer {self.auth_token}'}
      return self.post(url, headers=headers, data=data)

然后在另一个文件中,我用response = self.post('/api/users/register', data=data)调用它并收到错误消息:

sqlalchemy.orm.exc.DetachedInstanceError: Parent instance <User at 0x10ea43f98> is not bound to a Session; lazy load operation of attribute 'courses' cannot proceed (Background on this error at: http://sqlalche.me/e/bhk3)

下面是路线。我创建了一个用户对象,然后尝试向其添加课程,但是显然,当我尝试执行user.courses.append时,它无法延迟加载课程。我已经考虑过禁用expire_on_commit属性,但是我不确定仅在测试情况下或者是否有更好的方法可以这样做。

我读到,从延迟加载到急切加载也可以解决问题,但是我宁愿让事情延迟加载,而且当我将关系设置为lazy='subquery'时,似乎也没有解决问题。

@users.route('/register', methods=['POST'])
def register():
   data = request.get_json()
   try:
      user = User.from_data(**data)
   except UnprocessableEntity:
      return jsonify(msg='Missing parameters'), 422
   except Conflict:
      return jsonify(msg='Email already exists'), 409
   except Unauthorized:
      return jsonify(msg='Unauthorized'), 401
   if data.get('invitation_token'):
      invite = CourseInvite.query.filter_by(token=data['invitation_token']).first_or_404('Invite not found')
      if invite.email and invite.email != user.email:
         return jsonify(msg='Email mismatch'), 401
      elif invite.used:
         return jsonify(msg='Invite already used'), 401
      user.courses.append(CourseEnrollment(course_id=invite.course.id))
      invite.used = True
   db.session.add(user)
   db.session.commit()
   return jsonify(msg='User successfully created'), 201

0 个答案:

没有答案