mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-11-10 04:05:42 +01:00
4de909747b
Some checks are pending
/ release (push) Waiting to run
testing / test-remote-cacher (map[image:redis:7.2 port:6379]) (push) Blocked by required conditions
testing / test-remote-cacher (map[image:registry.redict.io/redict:7.3.0-scratch port:6379]) (push) Blocked by required conditions
testing / test-mysql (push) Blocked by required conditions
testing / test-pgsql (push) Blocked by required conditions
testing / backend-checks (push) Waiting to run
testing / frontend-checks (push) Waiting to run
testing / test-unit (push) Blocked by required conditions
testing / test-remote-cacher (map[image:docker.io/valkey/valkey:7.2.5-alpine3.19 port:6379]) (push) Blocked by required conditions
testing / test-remote-cacher (map[image:ghcr.io/microsoft/garnet-alpine:1.0.14 port:6379]) (push) Blocked by required conditions
testing / test-sqlite (push) Blocked by required conditions
testing / security-check (push) Blocked by required conditions
go-require lint is ignored for now Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/4535 Reviewed-by: Gusted <gusted@noreply.codeberg.org> Co-authored-by: TheFox0x7 <thefox0x7@gmail.com> Co-committed-by: TheFox0x7 <thefox0x7@gmail.com>
65 lines
2.3 KiB
Go
65 lines
2.3 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package repo_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
"code.gitea.io/gitea/modules/git"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestRepoArchiveDownloadCount(t *testing.T) {
|
|
require.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
release, err := repo_model.GetReleaseByID(db.DefaultContext, 1)
|
|
require.NoError(t, err)
|
|
|
|
// We have no count, so it should return 0
|
|
downloadCount, err := repo_model.GetArchiveDownloadCount(db.DefaultContext, release.RepoID, release.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(0), downloadCount.Zip)
|
|
assert.Equal(t, int64(0), downloadCount.TarGz)
|
|
|
|
// Set the TarGz counter to 1
|
|
err = repo_model.CountArchiveDownload(db.DefaultContext, release.RepoID, release.ID, git.TARGZ)
|
|
require.NoError(t, err)
|
|
|
|
downloadCount, err = repo_model.GetArchiveDownloadCountForTagName(db.DefaultContext, release.RepoID, release.TagName)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(0), downloadCount.Zip)
|
|
assert.Equal(t, int64(1), downloadCount.TarGz)
|
|
|
|
// Set the TarGz counter to 2
|
|
err = repo_model.CountArchiveDownload(db.DefaultContext, release.RepoID, release.ID, git.TARGZ)
|
|
require.NoError(t, err)
|
|
|
|
downloadCount, err = repo_model.GetArchiveDownloadCountForTagName(db.DefaultContext, release.RepoID, release.TagName)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(0), downloadCount.Zip)
|
|
assert.Equal(t, int64(2), downloadCount.TarGz)
|
|
|
|
// Set the Zip counter to 1
|
|
err = repo_model.CountArchiveDownload(db.DefaultContext, release.RepoID, release.ID, git.ZIP)
|
|
require.NoError(t, err)
|
|
|
|
downloadCount, err = repo_model.GetArchiveDownloadCountForTagName(db.DefaultContext, release.RepoID, release.TagName)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(1), downloadCount.Zip)
|
|
assert.Equal(t, int64(2), downloadCount.TarGz)
|
|
|
|
// Delete the count
|
|
err = repo_model.DeleteArchiveDownloadCountForRelease(db.DefaultContext, release.ID)
|
|
require.NoError(t, err)
|
|
|
|
downloadCount, err = repo_model.GetArchiveDownloadCountForTagName(db.DefaultContext, release.RepoID, release.TagName)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(0), downloadCount.Zip)
|
|
assert.Equal(t, int64(0), downloadCount.TarGz)
|
|
}
|