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.
103 lines
3.3 KiB
103 lines
3.3 KiB
// Copyright 2017 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package repo_test |
|
|
|
import ( |
|
"testing" |
|
|
|
repo_model "code.gitea.io/gitea/models/repo" |
|
"code.gitea.io/gitea/models/unittest" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestIncreaseDownloadCount(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
attachment, err := repo_model.GetAttachmentByUUID(t.Context(), "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11") |
|
assert.NoError(t, err) |
|
assert.Equal(t, int64(0), attachment.DownloadCount) |
|
|
|
// increase download count |
|
err = attachment.IncreaseDownloadCount(t.Context()) |
|
assert.NoError(t, err) |
|
|
|
attachment, err = repo_model.GetAttachmentByUUID(t.Context(), "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11") |
|
assert.NoError(t, err) |
|
assert.Equal(t, int64(1), attachment.DownloadCount) |
|
} |
|
|
|
func TestGetByCommentOrIssueID(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
// count of attachments from issue ID |
|
attachments, err := repo_model.GetAttachmentsByIssueID(t.Context(), 1) |
|
assert.NoError(t, err) |
|
assert.Len(t, attachments, 1) |
|
|
|
attachments, err = repo_model.GetAttachmentsByCommentID(t.Context(), 1) |
|
assert.NoError(t, err) |
|
assert.Len(t, attachments, 2) |
|
} |
|
|
|
func TestDeleteAttachments(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
count, err := repo_model.DeleteAttachmentsByIssue(t.Context(), 4, false) |
|
assert.NoError(t, err) |
|
assert.Equal(t, 2, count) |
|
|
|
count, err = repo_model.DeleteAttachmentsByComment(t.Context(), 2, false) |
|
assert.NoError(t, err) |
|
assert.Equal(t, 2, count) |
|
|
|
err = repo_model.DeleteAttachment(t.Context(), &repo_model.Attachment{ID: 8}, false) |
|
assert.NoError(t, err) |
|
|
|
attachment, err := repo_model.GetAttachmentByUUID(t.Context(), "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a18") |
|
assert.Error(t, err) |
|
assert.True(t, repo_model.IsErrAttachmentNotExist(err)) |
|
assert.Nil(t, attachment) |
|
} |
|
|
|
func TestGetAttachmentByID(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
attach, err := repo_model.GetAttachmentByID(t.Context(), 1) |
|
assert.NoError(t, err) |
|
assert.Equal(t, "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", attach.UUID) |
|
} |
|
|
|
func TestAttachment_DownloadURL(t *testing.T) { |
|
attach := &repo_model.Attachment{ |
|
UUID: "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", |
|
ID: 1, |
|
} |
|
assert.Equal(t, "https://try.gitea.io/attachments/a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", attach.DownloadURL()) |
|
} |
|
|
|
func TestUpdateAttachment(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
attach, err := repo_model.GetAttachmentByID(t.Context(), 1) |
|
assert.NoError(t, err) |
|
assert.Equal(t, "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", attach.UUID) |
|
|
|
attach.Name = "new_name" |
|
assert.NoError(t, repo_model.UpdateAttachment(t.Context(), attach)) |
|
|
|
unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{Name: "new_name"}) |
|
} |
|
|
|
func TestGetAttachmentsByUUIDs(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
attachList, err := repo_model.GetAttachmentsByUUIDs(t.Context(), []string{"a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a17", "not-existing-uuid"}) |
|
assert.NoError(t, err) |
|
assert.Len(t, attachList, 2) |
|
assert.Equal(t, "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", attachList[0].UUID) |
|
assert.Equal(t, "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a17", attachList[1].UUID) |
|
assert.Equal(t, int64(1), attachList[0].IssueID) |
|
assert.Equal(t, int64(5), attachList[1].IssueID) |
|
}
|
|
|