Question

让我们说我有烧瓶应用现在我想测试u = g.user.be_friend（用户），如果他们不是朋友，那么它们会让对方成为朋友但是我不知道我是怎么做的可以测试g.user，最重要的是g.user.be_friend你能帮忙吗？

view.py：

    user = Users.query.filter_by(name = name).first()
    if user is None:
        flash('User %s not found.' % name)
        return redirect(url_for('layout.user'))
    if user == g.user:
        flash('You can\'t Friend yourself!')
        return redirect(url_for('layout.user',page=1, sortby='normal'))
    u = g.user.be_friend(user)

mytest_app.py：

def test_friend_main(self):
        e1 = Users(name='admine1', email='admine2@gmail.com', age=25)
        e2 = Users(name='teste2', email='teste2@gmail.com', age=27)
        db.session.add_all([e1, e2])
        db.session.commit()
        with self.client:
            self.assertTrue(g.user =='admine1')

        e1.be_friend(e2)
        response=self.client.get('/friend/admine1', follow_redirects=True)
        self.assertTrue('You are now Friend with !',response.data)

Answer 1

您需要flask-login模块。

from flask.ext.login import current_user

@lm.user_loader   # you should have same function
def load_user(user_id):
   return User.get_user(user_id)

@flask_app.before_request
def before_request():
    g.user = current_user # save current user in request context.

可悲的是，但我没有找到任何简单的方法来获取测试代码中的当前g值。只有艰难的test_request_context。无论如何，您可以测试current_user，如果它传递给模板：

def test_something(self)
     self.assertEqual(self.get_context_variable('current_user').name, name_should_be)

但首先，您应该登录：

response = self.client.post('/login', data=dict(name='admine1', email='admine2@gmail.com'))

可以改进

User模型：class User(UserMixin, db.Model):

此外，在测试中，您不应直接执行e1.be_friend(e2)。您应该从生产代码中执行它。

Answer 2

我已经为我在本书中介绍的Flasky应用程序的“关注者”功能实施了单元测试。这是测试，您可以从中获取一些想法：

def test_follows(self):
    u1 = User(email='john@example.com', password='cat')
    u2 = User(email='susan@example.org', password='dog')
    db.session.add(u1)
    db.session.add(u2)
    db.session.commit()
    self.assertFalse(u1.is_following(u2))
    self.assertFalse(u1.is_followed_by(u2))
    timestamp_before = datetime.utcnow()
    u1.follow(u2)
    db.session.add(u1)
    db.session.commit()
    timestamp_after = datetime.utcnow()
    self.assertTrue(u1.is_following(u2))
    self.assertFalse(u1.is_followed_by(u2))
    self.assertTrue(u2.is_followed_by(u1))
    self.assertTrue(u1.followed.count() == 2)
    self.assertTrue(u2.followers.count() == 2)
    f = u1.followed.all()[-1]
    self.assertTrue(f.followed == u2)
    self.assertTrue(timestamp_before <= f.timestamp <= timestamp_after)
    f = u2.followers.all()[-1]
    self.assertTrue(f.follower == u1)
    u1.unfollow(u2)
    db.session.add(u1)
    db.session.commit()
    self.assertTrue(u1.followed.count() == 1)
    self.assertTrue(u2.followers.count() == 1)
    self.assertTrue(Follow.query.count() == 2)
    u2.follow(u1)
    db.session.add(u1)
    db.session.add(u2)
    db.session.commit()
    db.session.delete(u2)
    db.session.commit()
    self.assertTrue(Follow.query.count() == 1)

当您查看此代码时，请记住此应用程序会使用户自己成为关注者，例如，当用户跟随另一个用户时，跟随者计数是两个，而不是一个。

如何在Flask中测试u = g.user

2 个答案: