Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.1 KiB
45 lines
1.1 KiB
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{}) |
|
}
|
|
|