mirror of https://github.com/go-gitea/gitea.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
210 lines
9.0 KiB
210 lines
9.0 KiB
// Copyright 2021 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package integration |
|
|
|
import ( |
|
"bytes" |
|
"fmt" |
|
"mime/multipart" |
|
"net/http" |
|
"testing" |
|
|
|
auth_model "code.gitea.io/gitea/models/auth" |
|
issues_model "code.gitea.io/gitea/models/issues" |
|
repo_model "code.gitea.io/gitea/models/repo" |
|
"code.gitea.io/gitea/models/unittest" |
|
user_model "code.gitea.io/gitea/models/user" |
|
api "code.gitea.io/gitea/modules/structs" |
|
"code.gitea.io/gitea/services/convert" |
|
"code.gitea.io/gitea/tests" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestAPIGetCommentAttachment(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: 2}) |
|
assert.NoError(t, comment.LoadIssue(t.Context())) |
|
assert.NoError(t, comment.LoadAttachments(t.Context())) |
|
attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: comment.Attachments[0].ID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: comment.Issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
t.Run("UnrelatedCommentID", func(t *testing.T) { |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 4}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
token := getUserToken(t, repoOwner.Name, auth_model.AccessTokenScopeWriteIssue) |
|
req := NewRequestf(t, "GET", "/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", repoOwner.Name, repo.Name, comment.ID, attachment.ID). |
|
AddTokenAuth(token) |
|
MakeRequest(t, req, http.StatusNotFound) |
|
}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeReadIssue) |
|
req := NewRequestf(t, "GET", "/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", repoOwner.Name, repo.Name, comment.ID, attachment.ID). |
|
AddTokenAuth(token) |
|
session.MakeRequest(t, req, http.StatusOK) |
|
req = NewRequestf(t, "GET", "/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", repoOwner.Name, repo.Name, comment.ID, attachment.ID). |
|
AddTokenAuth(token) |
|
resp := session.MakeRequest(t, req, http.StatusOK) |
|
|
|
var apiAttachment api.Attachment |
|
DecodeJSON(t, resp, &apiAttachment) |
|
|
|
expect := convert.ToAPIAttachment(repo, attachment) |
|
assert.Equal(t, expect.ID, apiAttachment.ID) |
|
assert.Equal(t, expect.Name, apiAttachment.Name) |
|
assert.Equal(t, expect.UUID, apiAttachment.UUID) |
|
assert.Equal(t, expect.Created.Unix(), apiAttachment.Created.Unix()) |
|
assert.Equal(t, expect.DownloadURL, apiAttachment.DownloadURL) |
|
} |
|
|
|
func TestAPIListCommentAttachments(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: 2}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeReadIssue) |
|
req := NewRequestf(t, "GET", "/api/v1/repos/%s/%s/issues/comments/%d/assets", repoOwner.Name, repo.Name, comment.ID). |
|
AddTokenAuth(token) |
|
resp := session.MakeRequest(t, req, http.StatusOK) |
|
|
|
var apiAttachments []*api.Attachment |
|
DecodeJSON(t, resp, &apiAttachments) |
|
expectedCount := unittest.GetCount(t, &repo_model.Attachment{CommentID: comment.ID}) |
|
assert.Len(t, apiAttachments, expectedCount) |
|
|
|
unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: apiAttachments[0].ID, CommentID: comment.ID}) |
|
} |
|
|
|
func TestAPICreateCommentAttachment(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: 2}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteIssue) |
|
|
|
body := &bytes.Buffer{} |
|
|
|
// Setup multi-part |
|
writer := multipart.NewWriter(body) |
|
part, err := writer.CreateFormFile("attachment", "image.png") |
|
assert.NoError(t, err) |
|
_, err = part.Write(testGeneratePngBytes()) |
|
assert.NoError(t, err) |
|
err = writer.Close() |
|
assert.NoError(t, err) |
|
|
|
req := NewRequestWithBody(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/issues/comments/%d/assets", repoOwner.Name, repo.Name, comment.ID), body). |
|
AddTokenAuth(token). |
|
SetHeader("Content-Type", writer.FormDataContentType()) |
|
resp := session.MakeRequest(t, req, http.StatusCreated) |
|
|
|
apiAttachment := new(api.Attachment) |
|
DecodeJSON(t, resp, &apiAttachment) |
|
|
|
unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: apiAttachment.ID, CommentID: comment.ID}) |
|
} |
|
|
|
func TestAPICreateCommentAttachmentWithUnallowedFile(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: 2}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteIssue) |
|
|
|
filename := "file.bad" |
|
body := &bytes.Buffer{} |
|
|
|
// Setup multi-part. |
|
writer := multipart.NewWriter(body) |
|
_, err := writer.CreateFormFile("attachment", filename) |
|
assert.NoError(t, err) |
|
err = writer.Close() |
|
assert.NoError(t, err) |
|
|
|
req := NewRequestWithBody(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/issues/comments/%d/assets", repoOwner.Name, repo.Name, comment.ID), body). |
|
AddTokenAuth(token). |
|
SetHeader("Content-Type", writer.FormDataContentType()) |
|
|
|
session.MakeRequest(t, req, http.StatusUnprocessableEntity) |
|
} |
|
|
|
func TestAPIEditCommentAttachment(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
const newAttachmentName = "newAttachmentName.txt" |
|
|
|
attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: 6}) |
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: attachment.CommentID}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteIssue) |
|
urlStr := fmt.Sprintf("/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", |
|
repoOwner.Name, repo.Name, comment.ID, attachment.ID) |
|
req := NewRequestWithValues(t, "PATCH", urlStr, map[string]string{ |
|
"name": newAttachmentName, |
|
}).AddTokenAuth(token) |
|
resp := session.MakeRequest(t, req, http.StatusCreated) |
|
apiAttachment := new(api.Attachment) |
|
DecodeJSON(t, resp, &apiAttachment) |
|
|
|
unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: apiAttachment.ID, CommentID: comment.ID, Name: apiAttachment.Name}) |
|
} |
|
|
|
func TestAPIEditCommentAttachmentWithUnallowedFile(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: 6}) |
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: attachment.CommentID}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteIssue) |
|
|
|
filename := "file.bad" |
|
urlStr := fmt.Sprintf("/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", |
|
repoOwner.Name, repo.Name, comment.ID, attachment.ID) |
|
req := NewRequestWithValues(t, "PATCH", urlStr, map[string]string{ |
|
"name": filename, |
|
}).AddTokenAuth(token) |
|
|
|
session.MakeRequest(t, req, http.StatusUnprocessableEntity) |
|
} |
|
|
|
func TestAPIDeleteCommentAttachment(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: 6}) |
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: attachment.CommentID}) |
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: comment.IssueID}) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID}) |
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteIssue) |
|
|
|
req := NewRequest(t, "DELETE", fmt.Sprintf("/api/v1/repos/%s/%s/issues/comments/%d/assets/%d", repoOwner.Name, repo.Name, comment.ID, attachment.ID)). |
|
AddTokenAuth(token) |
|
session.MakeRequest(t, req, http.StatusNoContent) |
|
|
|
unittest.AssertNotExistsBean(t, &repo_model.Attachment{ID: attachment.ID, CommentID: comment.ID}) |
|
}
|
|
|