diff options
Diffstat (limited to 'models/user_follow_test.go')
| -rw-r--r-- | models/user_follow_test.go | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/models/user_follow_test.go b/models/user_follow_test.go index c7e56410f0..5ba922728a 100644 --- a/models/user_follow_test.go +++ b/models/user_follow_test.go @@ -7,7 +7,6 @@ package models import ( "testing" - "code.gitea.io/gitea/models/db" "code.gitea.io/gitea/models/unittest" "github.com/stretchr/testify/assert" ) @@ -16,9 +15,9 @@ func TestIsFollowing(t *testing.T) { assert.NoError(t, unittest.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)) + assert.False(t, IsFollowing(5, unittest.NonexistentID)) + assert.False(t, IsFollowing(unittest.NonexistentID, 5)) + assert.False(t, IsFollowing(unittest.NonexistentID, unittest.NonexistentID)) } func TestFollowUser(t *testing.T) { @@ -26,14 +25,14 @@ func TestFollowUser(t *testing.T) { testSuccess := func(followerID, followedID int64) { assert.NoError(t, FollowUser(followerID, followedID)) - db.AssertExistsAndLoadBean(t, &Follow{UserID: followerID, FollowID: followedID}) + unittest.AssertExistsAndLoadBean(t, &Follow{UserID: followerID, FollowID: followedID}) } testSuccess(4, 2) testSuccess(5, 2) assert.NoError(t, FollowUser(2, 2)) - CheckConsistencyFor(t, &User{}) + unittest.CheckConsistencyFor(t, &User{}) } func TestUnfollowUser(t *testing.T) { @@ -41,11 +40,11 @@ func TestUnfollowUser(t *testing.T) { testSuccess := func(followerID, followedID int64) { assert.NoError(t, UnfollowUser(followerID, followedID)) - db.AssertNotExistsBean(t, &Follow{UserID: followerID, FollowID: followedID}) + unittest.AssertNotExistsBean(t, &Follow{UserID: followerID, FollowID: followedID}) } testSuccess(4, 2) testSuccess(5, 2) testSuccess(2, 2) - CheckConsistencyFor(t, &User{}) + unittest.CheckConsistencyFor(t, &User{}) } |