コード例 #1
0
 def test_favorite_follows(self):
     u1 = User.create(email='*****@*****.**', password='******')
     u2 = User.create(email='*****@*****.**', password='******')
     favor = Favorite.create(title='美术', user=u2)
     db.session.add_all([u1, u2, favor])
     db.session.commit()
     self.assertFalse(u1.is_following_favorite(favor))
     self.assertFalse(favor.is_followed_by(u1))
     timestamp_before = datetime.datetime.utcnow()
     u1.follow(favor)
     db.session.add(u1)
     db.session.commit()
     timestamp_after = datetime.datetime.utcnow()
     self.assertTrue(u1.is_following_favorite(favor))
     self.assertTrue(favor.is_followed_by(u1))
     self.assertTrue(u1.followed_favorites.count() == 1)
     self.assertTrue(favor.followers.count() == 1)
     f = u1.followed_favorites.all()[0]
     self.assertTrue(f.followed == favor)
     self.assertTrue(timestamp_before <= f.timestamp <= timestamp_after)
     u1.unfollow(favor)
     db.session.add(u1)
     db.session.commit()
     self.assertFalse(u1.is_following_favorite(favor))
     self.assertFalse(favor.is_followed_by(u1))
     self.assertTrue(u1.followed_favorites.count() == 0)
     self.assertTrue(favor.followers.count() == 0)
     self.assertTrue(FollowFavorite.query.count() == 0)
コード例 #2
0
ファイル: favorite.py プロジェクト: giligiliduang/new_zhidao
def create_favorite(username):
    s = search()
    if s:
        return s
    user = User.query.filter_by(username=username).first()
    if current_user != user and not current_user.can(Permission.ADMINISTER):
        flash('没有权限')
        return redirect(url_for('.user', username=user.username))
    form = FavoriteForm()
    if form.validate_on_submit():
        Favorite.create(title=form.title.data,
                        description=form.description.data,
                        public=form.public.data,
                        user=current_user._get_current_object())

        flash('新建了{}收藏夹'.format(form.title.data), category='info')
        return redirect(url_for('.user', username=user.username))
    return render_template('favorite/create_favorite.html', form=form)  # 渲染表单
コード例 #3
0
ファイル: favorite.py プロジェクト: damnever/2L
 def post(self, post_id):
     username = self.current_user
     f = yield gen.maybe_future(Favorite.get_by_user_post(username, post_id))
     if f:
         raise exceptions.PostAlreadyFavorited()
     else:
         yield gen.maybe_future(Favorite.create(username, post_id))
         count = yield gen.maybe_future(Favorite.count_by_post(post_id))
         # Update gold.
         update_gold.apply_async(("post_be_favorite", post_id))
         raise gen.Return({"count": count})
コード例 #4
0
    def test_collect_question(self):
        q = Question.create(title='wdffd')
        f = Favorite.create(title='数学')

        self.assertListEqual(f.questions.all(), [])
        self.assertTrue(f.questions.count() == 0)
        self.assertTrue(q.favorites.count() == 0)
        f.collect(q)
        self.assertListEqual([i.question for i in f.questions.all()], [q])
        self.assertTrue(f.questions.count() == 1)
        self.assertTrue(q.favorites.count() == 1)
        self.assertTrue(f.has_collect_question(q))
        f.uncollect(q)
        self.assertTrue(f.questions.count() == 0)
        self.assertTrue(q.favorites.count() == 0)
        self.assertFalse(f.has_collect_question(q))
コード例 #5
0
    def test_collect_post(self):
        p = Post.create(title='haha')
        f = Favorite.create(title='数学')

        self.assertListEqual(f.posts.all(), [])
        self.assertTrue(f.posts.count() == 0)
        self.assertTrue(p.favorites.count() == 0)
        f.collect(p)
        self.assertListEqual([i.post for i in f.posts.all()], [p])
        self.assertTrue(f.posts.count() == 1)
        self.assertTrue(p.favorites.count() == 1)
        self.assertTrue(f.has_collect_post(p))
        f.uncollect(p)
        self.assertTrue(f.posts.count() == 0)
        self.assertTrue(p.favorites.count() == 0)
        self.assertFalse(f.has_collect_post(p))
コード例 #6
0
    def test_collect_answer(self):
        q = Question.create(title='dwwf')
        a = Answer.create(body='dwwdf', question=q)
        f = Favorite.create(title='数学')

        self.assertListEqual(f.answers.all(), [])
        self.assertTrue(f.answers.count() == 0)
        self.assertTrue(a.favorites.count() == 0)
        f.collect(a)
        self.assertListEqual([i.answer for i in f.answers.all()], [a])
        self.assertTrue(f.answers.count() == 1)
        self.assertTrue(a.favorites.count() == 1)
        self.assertTrue(f.has_collect_answer(a))
        f.uncollect(a)
        self.assertTrue(f.answers.count() == 0)
        self.assertTrue(a.favorites.count() == 0)
        self.assertFalse(f.has_collect_answer(a))