gh-actions-cache/service/actions_cache_test.go

package service

import (
	"errors"
	"net/url"
	"testing"

	"github.com/actions/gh-actions-cache/internal"
	"github.com/actions/gh-actions-cache/types"
	"github.com/cli/go-gh/pkg/api"
	"github.com/stretchr/testify/assert"
	"gopkg.in/h2non/gock.v1"
)

const VERSION string = "1.0.0"

func TestGetCacheUsage_CorrectRepo(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Get("/repos/testOrg/testRepo/actions/cache/usage").
		Reply(200).
		JSON(`{
			"full_name": "testOrg/testRepo",
			"active_caches_size_in_bytes": 291205,
			"active_caches_count": 12
		}`)

	repo, err := internal.GetRepo("testOrg/testRepo")
	assert.Nil(t, err)

	artifactCache, err := NewArtifactCache(repo, "list", VERSION)
	assert.Nil(t, err)
	totalCacheSize, err := artifactCache.GetCacheUsage()

	assert.Equal(t, totalCacheSize, float64(291205))
	assert.Nil(t, err)
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}

func TestGetCacheUsage_IncorrectRepo(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Get("/repos/testOrg/wrongRepo/actions/cache/usage").
		Reply(404).
		JSON(`{
			"message": "Not Found",
			"documentation_url": "https://docs.github.com/rest/reference/actions#get-github-actions-cache-usage-for-a-repository"
		}`)

	repo, err := internal.GetRepo("testOrg/wrongRepo")
	assert.Nil(t, err)

	artifactCache, err := NewArtifactCache(repo, "list", VERSION)
	assert.Nil(t, err)
	totalCacheSize, err := artifactCache.GetCacheUsage()
	var httpError api.HTTPError
	errors.As(err, &httpError)

	assert.NotNil(t, err)
	assert.Equal(t, httpError.StatusCode, 404)
	assert.Equal(t, httpError.Message, "Not Found")
	assert.Equal(t, totalCacheSize, float64(-1))
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}

func TestListCaches_Success(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Get("/repos/testOrg/testRepo/actions/caches").
		Reply(200).
		JSON(`{
			"total_count": 1,
			"actions_caches": [
				{
					"id": 29,
					"ref": "refs/heads/master",
					"key": "Linux-build-cache-node-modules-3fd22dd3a926d576e2562e8b76a5ff157cd3b986f3d44195acfe7efa6bc05919-8",
					"version": "7fcda33c1e1d849a13bcc06f49b9ab64efc01ca9dabe4d7a8d0d387feef4fc88",
					"last_accessed_at": "2022-06-22T20:32:45.550000000Z",
					"created_at": "2022-06-22T20:32:45.550000000Z",
					"size_in_bytes": 2432967
				}]
			}`)

	repo, err := internal.GetRepo("testOrg/testRepo")
	assert.Nil(t, err)

	f := types.ListOptions{BaseOptions: types.BaseOptions{Repo: "testOrg/testRepo"}, Limit: 30}
	queryParams := url.Values{}
	f.GenerateQueryParams(queryParams)

	artifactCache, err := NewArtifactCache(repo, "list", VERSION)
	assert.Nil(t, err)
	listCacheResponse, err := artifactCache.ListCaches(queryParams)

	assert.Nil(t, err)
	assert.NotNil(t, listCacheResponse)
	assert.Equal(t, listCacheResponse.TotalCount, 1)
	assert.Equal(t, len(listCacheResponse.ActionsCaches), 1)
	assert.Equal(t, listCacheResponse.ActionsCaches[0].Id, 29)
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}

func TestListCaches_Failure(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Get("/repos/testOrg/testRepo/actions/caches").
		Reply(404).
		JSON(`{
			"message": "Not Found",
			"documentation_url": "https://docs.github.com/rest/reference/actions#get-github-actions-cache-list-for-a-repository"
		}`)

	repo, err := internal.GetRepo("testOrg/testRepo")
	assert.Nil(t, err)

	f := types.ListOptions{BaseOptions: types.BaseOptions{Repo: "testOrg/testRepo"}, Limit: 30}
	queryParams := url.Values{}
	f.GenerateQueryParams(queryParams)

	artifactCache, err := NewArtifactCache(repo, "list", VERSION)
	assert.Nil(t, err)
	listCacheResponse, err := artifactCache.ListCaches(queryParams)
	var httpError api.HTTPError
	errors.As(err, &httpError)

	assert.NotNil(t, err)
	assert.Equal(t, httpError.StatusCode, 404)
	assert.Equal(t, httpError.Message, "Not Found")
	assert.Equal(t, listCacheResponse, types.ListApiResponse{})
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}

func TestDeleteCaches_Success(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Delete("/repos/testOrg/testRepo/actions/caches").
		Reply(200).
		JSON(`{
			"total_count": 1,
			"actions_caches": [
				{
					"id": 29,
					"ref": "refs/heads/master",
					"key": "Linux-build-cache-node-modules-3fd22dd3a926d576e2562e8b76a5ff157cd3b986f3d44195acfe7efa6bc05919-8",
					"version": "7fcda33c1e1d849a13bcc06f49b9ab64efc01ca9dabe4d7a8d0d387feef4fc88",
					"last_accessed_at": "2022-06-22T20:32:45.550000000Z",
					"created_at": "2022-06-22T20:32:45.550000000Z",
					"size_in_bytes": 2432967
				}]
			}`)

	repo, err := internal.GetRepo("testOrg/testRepo")
	assert.Nil(t, err)

	f := types.DeleteOptions{BaseOptions: types.BaseOptions{Repo: "testOrg/testRepo"}}
	queryParams := url.Values{}
	f.GenerateBaseQueryParams(queryParams)

	artifactCache, err := NewArtifactCache(repo, "delete", VERSION)
	assert.Nil(t, err)
	deletedCache, err := artifactCache.DeleteCaches(queryParams)

	assert.Nil(t, err)
	assert.Equal(t, deletedCache, 1)
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}

func TestDeleteCaches_Failure(t *testing.T) {
	t.Cleanup(gock.Off)

	gock.New("https://api.github.com").
		Delete("/repos/testOrg/testRepo/actions/caches").
		Reply(404).
		JSON(`{
			"message": "Not Found",
			"documentation_url": "https://docs.github.com/rest/reference/actions#get-github-actions-cache-list-for-a-repository"
		}`)

	repo, err := internal.GetRepo("testOrg/testRepo")
	assert.Nil(t, err)

	f := types.DeleteOptions{BaseOptions: types.BaseOptions{Repo: "testOrg/testRepo"}}
	queryParams := url.Values{}
	f.GenerateBaseQueryParams(queryParams)

	artifactCache, err := NewArtifactCache(repo, "delete", VERSION)
	assert.Nil(t, err)
	deletedCache, err := artifactCache.DeleteCaches(queryParams)

	assert.NotNil(t, err)
	assert.Equal(t, deletedCache, 0)
	assert.True(t, gock.IsDone(), internal.PrintPendingMocks(gock.Pending()))
}