22bfd0c83e
fix: format of date when MR was closed or merged (#367) refactor: Add Payload Validators + Middleware In Go Code (#366) fix: Add better checks for leaving comments (#369) fix: regex support for http credentials embedded in remote url (#372) fix: Comment on single line selects two lines (#371) This is a #PATCH release.
70 lines
2.3 KiB
Go
70 lines
2.3 KiB
Go
package app
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/xanzy/go-gitlab"
|
|
)
|
|
|
|
type fakeFileUploaderClient struct {
|
|
testBase
|
|
}
|
|
|
|
func (f fakeFileUploaderClient) UploadFile(pid interface{}, content io.Reader, filename string, options ...gitlab.RequestOptionFunc) (*gitlab.ProjectFile, *gitlab.Response, error) {
|
|
resp, err := f.handleGitlabError()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return &gitlab.ProjectFile{}, resp, nil
|
|
}
|
|
|
|
type fakeFileReader struct{}
|
|
|
|
func (f fakeFileReader) ReadFile(path string) (io.Reader, error) {
|
|
return &bytes.Reader{}, nil
|
|
}
|
|
|
|
func TestAttachmentHandler(t *testing.T) {
|
|
attachmentTestRequestData := AttachmentRequest{
|
|
FileName: "some_file_name",
|
|
FilePath: "some_file_path",
|
|
}
|
|
|
|
t.Run("Returns 200-status response after upload", func(t *testing.T) {
|
|
request := makeRequest(t, http.MethodPost, "/attachment", attachmentTestRequestData)
|
|
svc := middleware(
|
|
attachmentService{testProjectData, fakeFileReader{}, fakeFileUploaderClient{}},
|
|
withPayloadValidation(methodToPayload{http.MethodPost: &AttachmentRequest{}}),
|
|
withMethodCheck(http.MethodPost),
|
|
)
|
|
data := getSuccessData(t, svc, request)
|
|
assert(t, data.Message, "File uploaded successfully")
|
|
})
|
|
|
|
t.Run("Handles errors from Gitlab client", func(t *testing.T) {
|
|
request := makeRequest(t, http.MethodPost, "/attachment", attachmentTestRequestData)
|
|
svc := middleware(
|
|
attachmentService{testProjectData, fakeFileReader{}, fakeFileUploaderClient{testBase{errFromGitlab: true}}},
|
|
withPayloadValidation(methodToPayload{http.MethodPost: &AttachmentRequest{}}),
|
|
withMethodCheck(http.MethodPost),
|
|
)
|
|
data, _ := getFailData(t, svc, request)
|
|
checkErrorFromGitlab(t, data, "Could not upload some_file_name to Gitlab")
|
|
})
|
|
|
|
t.Run("Handles non-200s from Gitlab client", func(t *testing.T) {
|
|
request := makeRequest(t, http.MethodPost, "/attachment", attachmentTestRequestData)
|
|
svc := middleware(
|
|
attachmentService{testProjectData, fakeFileReader{}, fakeFileUploaderClient{testBase{status: http.StatusSeeOther}}},
|
|
withPayloadValidation(methodToPayload{http.MethodPost: &AttachmentRequest{}}),
|
|
withMethodCheck(http.MethodPost),
|
|
)
|
|
data, _ := getFailData(t, svc, request)
|
|
checkNon200(t, data, "Could not upload some_file_name to Gitlab", "/attachment")
|
|
})
|
|
}
|