Move some pull request functions from models to services (#9266)

* Move some pull request functions from models to services

* Fix test

1 files changed, 34 insertions, 0 deletions

diff --git a/services/pull/check_test.go b/services/pull/check_test.go
new file mode 100644
index 0000000000..48a7774a61
--- /dev/null
+++ b/services/pull/check_test.go
@@ -0,0 +1,34 @@
+// Copyright 2019 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 pull
+
+import (
+	"strconv"
+	"testing"
+	"time"
+
+	"code.gitea.io/gitea/models"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPullRequest_AddToTaskQueue(t *testing.T) {
+	assert.NoError(t, models.PrepareTestDatabase())
+
+	pr := models.AssertExistsAndLoadBean(t, &models.PullRequest{ID: 1}).(*models.PullRequest)
+	AddToTaskQueue(pr)
+
+	select {
+	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, pullRequestQueue.Exist(pr.ID))
+	pr = models.AssertExistsAndLoadBean(t, &models.PullRequest{ID: 1}).(*models.PullRequest)
+	assert.Equal(t, models.PullRequestStatusChecking, pr.Status)
+}