Fix test

go-gitea · Dec 6, 2019 · bd483e6 · bd483e6
1 parent 53a6f62
commit bd483e6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/services/pull/check_test.go b/services/pull/check_test.go
@@ -10,6 +10,8 @@ import (
 	"testing"
 	"time"
 
+	"code.gitea.io/gitea/models"
+
 	"github.com/stretchr/testify/assert"
 )
 
@@ -20,13 +22,13 @@ func TestPullRequest_AddToTaskQueue(t *testing.T) {
 	AddToTaskQueue(pr)
 
 	select {
-	case id := <-models.PullRequestQueue.Queue():
+	case id := <-pullRequestQueue.Queue():
 		assert.EqualValues(t, strconv.FormatInt(pr.ID, 10), id)
 	case <-time.After(time.Second):
 		assert.Fail(t, "Timeout: nothing was added to pullRequestQueue")
 	}
 
-	assert.True(t, models.PullRequestQueue.Exist(pr.ID))
-	pr = AssertExistsAndLoadBean(t, &models.PullRequest{ID: 1}).(*models.PullRequest)
+	assert.True(t, pullRequestQueue.Exist(pr.ID))
+	pr = models.AssertExistsAndLoadBean(t, &models.PullRequest{ID: 1}).(*models.PullRequest)
 	assert.Equal(t, models.PullRequestStatusChecking, pr.Status)
 }