gitea/tests/integration/api_repo_avatar_test.go

// Copyright 2018 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package integration

import (
	"encoding/base64"
	"fmt"
	"net/http"
	"os"
	"path/filepath"
	"testing"

	auth_model "gitea.dev/models/auth"
	repo_model "gitea.dev/models/repo"
	"gitea.dev/models/unittest"
	user_model "gitea.dev/models/user"
	"gitea.dev/modules/setting"
	api "gitea.dev/modules/structs"
	"gitea.dev/tests"

	"github.com/stretchr/testify/assert"
)

func TestAPIUpdateRepoAvatar(t *testing.T) {
	defer tests.PrepareTestEnv(t)()

	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
	user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
	token := getUserToken(t, user2.LowerName, auth_model.AccessTokenScopeWriteRepository)

	t.Run("Delete", func(t *testing.T) {
		defer tests.PrintCurrentTest(t)()

		req := NewRequest(t, "DELETE", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name)).
			AddTokenAuth(token)
		MakeRequest(t, req, http.StatusNoContent)
	})

	avatar, err := os.ReadFile(filepath.Join(setting.GetGiteaTestSourceRoot(), "tests/integration/avatar.png"))
	assert.NoError(t, err)
	if err != nil {
		assert.FailNow(t, "Unable to open avatar.png")
	}

	for _, avatarCase := range []struct {
		image  string
		status int
	}{
		// Test what happens if you use a valid image
		{base64.StdEncoding.EncodeToString(avatar), http.StatusNoContent},
		// Test what happens if you don't have a valid Base64 string
		{"Invalid", http.StatusBadRequest},
		// Test what happens if you use a file that is not an image
		{base64.StdEncoding.EncodeToString([]byte("This is not an image")), http.StatusInternalServerError},
	} {
		req := NewRequestWithJSON(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name), &api.UpdateRepoAvatarOption{Image: avatarCase.image}).
			AddTokenAuth(token)
		MakeRequest(t, req, avatarCase.status)
	}
}