diff options
Diffstat (limited to 'models/consistency_test.go')
| -rw-r--r-- | models/consistency_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/models/consistency_test.go b/models/consistency_test.go index 25acb467b6..8332b5d761 100644 --- a/models/consistency_test.go +++ b/models/consistency_test.go @@ -14,10 +14,10 @@ import ( func TestDeleteOrphanedObjects(t *testing.T) { assert.NoError(t, db.PrepareTestDatabase()) - countBefore, err := db.DefaultContext().Engine().Count(&PullRequest{}) + countBefore, err := db.GetEngine(db.DefaultContext).Count(&PullRequest{}) assert.NoError(t, err) - _, err = db.DefaultContext().Engine().Insert(&PullRequest{IssueID: 1000}, &PullRequest{IssueID: 1001}, &PullRequest{IssueID: 1003}) + _, err = db.GetEngine(db.DefaultContext).Insert(&PullRequest{IssueID: 1000}, &PullRequest{IssueID: 1001}, &PullRequest{IssueID: 1003}) assert.NoError(t, err) orphaned, err := CountOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id") @@ -27,7 +27,7 @@ func TestDeleteOrphanedObjects(t *testing.T) { err = DeleteOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id") assert.NoError(t, err) - countAfter, err := db.DefaultContext().Engine().Count(&PullRequest{}) + countAfter, err := db.GetEngine(db.DefaultContext).Count(&PullRequest{}) assert.NoError(t, err) assert.EqualValues(t, countBefore, countAfter) } |