mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-03 17:24:22 +00:00 
			
		
		
		
	Attempt to handle unready PR in tests (#13305)
Signed-off-by: Andrew Thornton <art27@cantab.net>
This commit is contained in:
		@@ -5,14 +5,17 @@
 | 
			
		||||
package integrations
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"testing"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/models"
 | 
			
		||||
	"code.gitea.io/gitea/modules/auth"
 | 
			
		||||
	"code.gitea.io/gitea/modules/queue"
 | 
			
		||||
	api "code.gitea.io/gitea/modules/structs"
 | 
			
		||||
 | 
			
		||||
	"github.com/stretchr/testify/assert"
 | 
			
		||||
@@ -225,11 +228,25 @@ func doAPIMergePullRequest(ctx APITestContext, owner, repo string, index int64)
 | 
			
		||||
			Do:                string(models.MergeStyleMerge),
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		if ctx.ExpectedCode != 0 {
 | 
			
		||||
			ctx.Session.MakeRequest(t, req, ctx.ExpectedCode)
 | 
			
		||||
			return
 | 
			
		||||
		resp := ctx.Session.MakeRequest(t, req, NoExpectedStatus)
 | 
			
		||||
 | 
			
		||||
		if resp.Code == http.StatusMethodNotAllowed {
 | 
			
		||||
			err := api.APIError{}
 | 
			
		||||
			DecodeJSON(t, resp, &err)
 | 
			
		||||
			assert.EqualValues(t, "Please try again later", err.Message)
 | 
			
		||||
			queue.GetManager().FlushAll(context.Background(), 5*time.Second)
 | 
			
		||||
			resp = ctx.Session.MakeRequest(t, req, NoExpectedStatus)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		expected := ctx.ExpectedCode
 | 
			
		||||
		if expected == 0 {
 | 
			
		||||
			expected = 200
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if !assert.EqualValues(t, expected, resp.Code,
 | 
			
		||||
			"Request: %s %s", req.Method, req.URL.String()) {
 | 
			
		||||
			logUnexpectedResponse(t, resp)
 | 
			
		||||
		}
 | 
			
		||||
		ctx.Session.MakeRequest(t, req, 200)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user