diff --git a/models/packages/package_test.go b/models/packages/package_test.go index 3d3a2333ba..0bfe6a398b 100644 --- a/models/packages/package_test.go +++ b/models/packages/package_test.go @@ -27,7 +27,7 @@ func TestMain(m *testing.M) { func TestHasOwnerPackages(t *testing.T) { assert.NoError(t, unittest.PrepareTestDatabase()) - owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) + owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}).(*user_model.User) p, err := packages_model.TryInsertPackage(db.DefaultContext, &packages_model.Package{ OwnerID: owner.ID, diff --git a/models/user.go b/models/user.go index 49374014aa..fb9246070d 100644 --- a/models/user.go +++ b/models/user.go @@ -100,9 +100,9 @@ func DeleteUser(ctx context.Context, u *user_model.User) (err error) { // Delete Comments const batchSize = 50 - for start := 0; ; start += batchSize { + for { comments := make([]*issues_model.Comment, 0, batchSize) - if err = e.Where("type=? AND poster_id=?", issues_model.CommentTypeComment, u.ID).Limit(batchSize, start).Find(&comments); err != nil { + if err = e.Where("type=? AND poster_id=?", issues_model.CommentTypeComment, u.ID).Limit(batchSize, 0).Find(&comments); err != nil { return err } if len(comments) == 0 { @@ -200,7 +200,7 @@ func DeleteUser(ctx context.Context, u *user_model.User) (err error) { // ***** END: ExternalLoginUser ***** if _, err = e.ID(u.ID).Delete(new(user_model.User)); err != nil { - return fmt.Errorf("Delete: %v", err) + return fmt.Errorf("delete: %v", err) } return nil