Ethan Koenig
7 years ago
committed by
Lunny Xiao
5 changed files with 123 additions and 72 deletions
@ -0,0 +1,4 @@ |
|||
- |
|||
id: 1 |
|||
user_id: 4 |
|||
follow_id: 2 |
@ -0,0 +1,71 @@ |
|||
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|||
// Use of this source code is governed by a MIT-style
|
|||
// license that can be found in the LICENSE file.
|
|||
|
|||
package models |
|||
|
|||
// Follow represents relations of user and his/her followers.
|
|||
type Follow struct { |
|||
ID int64 `xorm:"pk autoincr"` |
|||
UserID int64 `xorm:"UNIQUE(follow)"` |
|||
FollowID int64 `xorm:"UNIQUE(follow)"` |
|||
} |
|||
|
|||
// IsFollowing returns true if user is following followID.
|
|||
func IsFollowing(userID, followID int64) bool { |
|||
has, _ := x.Get(&Follow{UserID: userID, FollowID: followID}) |
|||
return has |
|||
} |
|||
|
|||
// FollowUser marks someone be another's follower.
|
|||
func FollowUser(userID, followID int64) (err error) { |
|||
if userID == followID || IsFollowing(userID, followID) { |
|||
return nil |
|||
} |
|||
|
|||
sess := x.NewSession() |
|||
defer sessionRelease(sess) |
|||
if err = sess.Begin(); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Insert(&Follow{UserID: userID, FollowID: followID}); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Exec("UPDATE `user` SET num_followers = num_followers + 1 WHERE id = ?", followID); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Exec("UPDATE `user` SET num_following = num_following + 1 WHERE id = ?", userID); err != nil { |
|||
return err |
|||
} |
|||
return sess.Commit() |
|||
} |
|||
|
|||
// UnfollowUser unmarks someone as another's follower.
|
|||
func UnfollowUser(userID, followID int64) (err error) { |
|||
if userID == followID || !IsFollowing(userID, followID) { |
|||
return nil |
|||
} |
|||
|
|||
sess := x.NewSession() |
|||
defer sessionRelease(sess) |
|||
if err = sess.Begin(); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Delete(&Follow{UserID: userID, FollowID: followID}); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Exec("UPDATE `user` SET num_followers = num_followers - 1 WHERE id = ?", followID); err != nil { |
|||
return err |
|||
} |
|||
|
|||
if _, err = sess.Exec("UPDATE `user` SET num_following = num_following - 1 WHERE id = ?", userID); err != nil { |
|||
return err |
|||
} |
|||
return sess.Commit() |
|||
} |
|||
|
@ -0,0 +1,45 @@ |
|||
package models |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func TestIsFollowing(t *testing.T) { |
|||
assert.NoError(t, PrepareTestDatabase()) |
|||
assert.True(t, IsFollowing(4, 2)) |
|||
assert.False(t, IsFollowing(2, 4)) |
|||
assert.False(t, IsFollowing(5, NonexistentID)) |
|||
assert.False(t, IsFollowing(NonexistentID, 5)) |
|||
assert.False(t, IsFollowing(NonexistentID, NonexistentID)) |
|||
} |
|||
|
|||
func TestFollowUser(t *testing.T) { |
|||
assert.NoError(t, PrepareTestDatabase()) |
|||
|
|||
testSuccess := func(followerID, followedID int64) { |
|||
assert.NoError(t, FollowUser(followerID, followedID)) |
|||
AssertExistsAndLoadBean(t, &Follow{UserID: followerID, FollowID: followedID}) |
|||
} |
|||
testSuccess(4, 2) |
|||
testSuccess(5, 2) |
|||
|
|||
assert.NoError(t, FollowUser(2, 2)) |
|||
|
|||
CheckConsistencyFor(t, &User{}) |
|||
} |
|||
|
|||
func TestUnfollowUser(t *testing.T) { |
|||
assert.NoError(t, PrepareTestDatabase()) |
|||
|
|||
testSuccess := func(followerID, followedID int64) { |
|||
assert.NoError(t, UnfollowUser(followerID, followedID)) |
|||
AssertNotExistsBean(t, &Follow{UserID: followerID, FollowID: followedID}) |
|||
} |
|||
testSuccess(4, 2) |
|||
testSuccess(5, 2) |
|||
testSuccess(2, 2) |
|||
|
|||
CheckConsistencyFor(t, &User{}) |
|||
} |
Loading…
Reference in new issue