Платформа ЦРНП "Мирокод" для разработки проектов
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.
50 lines
1.3 KiB
50 lines
1.3 KiB
// Copyright 2020 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 |
|
|
|
import ( |
|
"testing" |
|
|
|
"code.gitea.io/gitea/models/db" |
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestIsFollowing(t *testing.T) { |
|
assert.NoError(t, db.PrepareTestDatabase()) |
|
assert.True(t, IsFollowing(4, 2)) |
|
assert.False(t, IsFollowing(2, 4)) |
|
assert.False(t, IsFollowing(5, db.NonexistentID)) |
|
assert.False(t, IsFollowing(db.NonexistentID, 5)) |
|
assert.False(t, IsFollowing(db.NonexistentID, db.NonexistentID)) |
|
} |
|
|
|
func TestFollowUser(t *testing.T) { |
|
assert.NoError(t, db.PrepareTestDatabase()) |
|
|
|
testSuccess := func(followerID, followedID int64) { |
|
assert.NoError(t, FollowUser(followerID, followedID)) |
|
db.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, db.PrepareTestDatabase()) |
|
|
|
testSuccess := func(followerID, followedID int64) { |
|
assert.NoError(t, UnfollowUser(followerID, followedID)) |
|
db.AssertNotExistsBean(t, &Follow{UserID: followerID, FollowID: followedID}) |
|
} |
|
testSuccess(4, 2) |
|
testSuccess(5, 2) |
|
testSuccess(2, 2) |
|
|
|
CheckConsistencyFor(t, &User{}) |
|
}
|
|
|