mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
381 lines
12 KiB
Go
381 lines
12 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package integration
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
neturl "net/url"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/git/models/packages"
|
|
repo_model "github.com/hanzoai/git/models/repo"
|
|
"github.com/hanzoai/git/models/unittest"
|
|
user_model "github.com/hanzoai/git/models/user"
|
|
"github.com/hanzoai/git/modules/gitrepo"
|
|
"github.com/hanzoai/git/modules/json"
|
|
cargo_module "github.com/hanzoai/git/modules/packages/cargo"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
cargo_router "github.com/hanzoai/git/routers/api/packages/cargo"
|
|
cargo_service "github.com/hanzoai/git/services/packages/cargo"
|
|
"github.com/hanzoai/git/tests"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPackageCargo(t *testing.T) {
|
|
onGitRun(t, testPackageCargo)
|
|
}
|
|
|
|
func testPackageCargo(t *testing.T, _ *neturl.URL) {
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
packageName := "cargo-package"
|
|
packageVersion := "1.0.3"
|
|
packageDescription := "Package Description"
|
|
packageAuthor := "KN4CK3R"
|
|
packageHomepage := "https://gitea.io/"
|
|
packageLicense := "MIT"
|
|
|
|
createPackage := func(name, version string) io.Reader {
|
|
metadata := `{
|
|
"name":"` + name + `",
|
|
"vers":"` + version + `",
|
|
"description":"` + packageDescription + `",
|
|
"authors": ["` + packageAuthor + `"],
|
|
"deps":[
|
|
{
|
|
"name":"dep",
|
|
"version_req":"1.0",
|
|
"registry": "https://gitea.io/user/_cargo-index",
|
|
"kind": "normal",
|
|
"default_features": true
|
|
}
|
|
],
|
|
"homepage":"` + packageHomepage + `",
|
|
"license":"` + packageLicense + `"
|
|
}`
|
|
|
|
var buf bytes.Buffer
|
|
binary.Write(&buf, binary.LittleEndian, uint32(len(metadata)))
|
|
buf.WriteString(metadata)
|
|
binary.Write(&buf, binary.LittleEndian, uint32(4))
|
|
buf.WriteString("test")
|
|
return &buf
|
|
}
|
|
|
|
err := cargo_service.InitializeIndexRepository(t.Context(), user, user)
|
|
assert.NoError(t, err)
|
|
|
|
repo, err := repo_model.GetRepositoryByOwnerAndName(t.Context(), user.Name, cargo_service.IndexRepositoryName)
|
|
assert.NotNil(t, repo)
|
|
assert.NoError(t, err)
|
|
|
|
readGitContent := func(t *testing.T, path string) string {
|
|
gitRepo, err := gitrepo.OpenRepository(t.Context(), repo)
|
|
assert.NoError(t, err)
|
|
defer gitRepo.Close()
|
|
|
|
commit, err := gitRepo.GetBranchCommit(repo.DefaultBranch)
|
|
assert.NoError(t, err)
|
|
|
|
blob, err := commit.GetBlobByPath(t.Context(), gitRepo, path)
|
|
assert.NoError(t, err)
|
|
|
|
content, err := blob.GetBlobContent(1024)
|
|
assert.NoError(t, err)
|
|
|
|
return content
|
|
}
|
|
|
|
root := fmt.Sprintf("%sv1/packages/%s/cargo", setting.AppURL, user.Name)
|
|
url := root + "/api/v1/crates"
|
|
|
|
t.Run("Index", func(t *testing.T) {
|
|
t.Run("Git/Config", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
content := readGitContent(t, cargo_service.ConfigFileName)
|
|
|
|
var config cargo_service.Config
|
|
err := json.Unmarshal([]byte(content), &config)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, url, config.DownloadURL)
|
|
assert.Equal(t, root, config.APIURL)
|
|
})
|
|
|
|
t.Run("HTTP/Config", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "GET", root+"/"+cargo_service.ConfigFileName)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
var config cargo_service.Config
|
|
err := json.Unmarshal(resp.Body.Bytes(), &config)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, url, config.DownloadURL)
|
|
assert.Equal(t, root, config.APIURL)
|
|
})
|
|
})
|
|
|
|
t.Run("Upload", func(t *testing.T) {
|
|
t.Run("InvalidNameOrVersion", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
content := createPackage("0test", "1.0.0")
|
|
|
|
req := NewRequestWithBody(t, "PUT", url+"/new", content).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusBadRequest)
|
|
|
|
status := DecodeJSON(t, resp, &cargo_router.StatusResponse{})
|
|
assert.False(t, status.OK)
|
|
|
|
content = createPackage("test", "-1.0.0")
|
|
|
|
req = NewRequestWithBody(t, "PUT", url+"/new", content).
|
|
AddBasicAuth(user.Name)
|
|
resp = MakeRequest(t, req, http.StatusBadRequest)
|
|
|
|
status = DecodeJSON(t, resp, &cargo_router.StatusResponse{})
|
|
assert.False(t, status.OK)
|
|
})
|
|
|
|
t.Run("InvalidContent", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
metadata := `{"name":"test","vers":"1.0.0"}`
|
|
|
|
var buf bytes.Buffer
|
|
binary.Write(&buf, binary.LittleEndian, uint32(len(metadata)))
|
|
buf.WriteString(metadata)
|
|
binary.Write(&buf, binary.LittleEndian, uint32(4))
|
|
buf.WriteString("te")
|
|
|
|
req := NewRequestWithBody(t, "PUT", url+"/new", &buf).
|
|
AddBasicAuth(user.Name)
|
|
MakeRequest(t, req, http.StatusBadRequest)
|
|
})
|
|
|
|
t.Run("Valid", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequestWithBody(t, "PUT", url+"/new", createPackage(packageName, packageVersion))
|
|
MakeRequest(t, req, http.StatusUnauthorized)
|
|
|
|
req = NewRequestWithBody(t, "PUT", url+"/new", createPackage(packageName, packageVersion)).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
status := DecodeJSON(t, resp, &cargo_router.StatusResponse{})
|
|
assert.True(t, status.OK)
|
|
|
|
pvs, err := packages.GetVersionsByPackageType(t.Context(), user.ID, packages.TypeCargo)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, pvs, 1)
|
|
|
|
pd, err := packages.GetPackageDescriptor(t.Context(), pvs[0])
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, pd.SemVer)
|
|
assert.IsType(t, &cargo_module.Metadata{}, pd.Metadata)
|
|
assert.Equal(t, packageName, pd.Package.Name)
|
|
assert.Equal(t, packageVersion, pd.Version.Version)
|
|
|
|
pfs, err := packages.GetFilesByVersionID(t.Context(), pvs[0].ID)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, pfs, 1)
|
|
assert.Equal(t, fmt.Sprintf("%s-%s.crate", packageName, packageVersion), pfs[0].Name)
|
|
assert.True(t, pfs[0].IsLead)
|
|
|
|
pb, err := packages.GetBlobByID(t.Context(), pfs[0].BlobID)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 4, pb.Size)
|
|
|
|
req = NewRequestWithBody(t, "PUT", url+"/new", createPackage(packageName, packageVersion)).
|
|
AddBasicAuth(user.Name)
|
|
MakeRequest(t, req, http.StatusConflict)
|
|
|
|
t.Run("Index", func(t *testing.T) {
|
|
t.Run("Git", func(t *testing.T) {
|
|
t.Run("Entry", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
content := readGitContent(t, cargo_service.BuildPackagePath(packageName))
|
|
|
|
var entry cargo_service.IndexVersionEntry
|
|
err := json.Unmarshal([]byte(content), &entry)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, packageName, entry.Name)
|
|
assert.Equal(t, packageVersion, entry.Version)
|
|
assert.Equal(t, pb.HashSHA256, entry.FileChecksum)
|
|
assert.False(t, entry.Yanked)
|
|
assert.Len(t, entry.Dependencies, 1)
|
|
dep := entry.Dependencies[0]
|
|
assert.Equal(t, "dep", dep.Name)
|
|
assert.Equal(t, "1.0", dep.Req)
|
|
assert.Equal(t, "normal", dep.Kind)
|
|
assert.True(t, dep.DefaultFeatures)
|
|
assert.Empty(t, dep.Features)
|
|
assert.False(t, dep.Optional)
|
|
assert.Nil(t, dep.Target)
|
|
assert.NotNil(t, dep.Registry)
|
|
assert.Equal(t, "https://gitea.io/user/_cargo-index", *dep.Registry)
|
|
assert.Nil(t, dep.Package)
|
|
})
|
|
|
|
t.Run("Rebuild", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
err := cargo_service.RebuildIndex(t.Context(), user, user)
|
|
assert.NoError(t, err)
|
|
|
|
_ = readGitContent(t, cargo_service.BuildPackagePath(packageName))
|
|
})
|
|
})
|
|
|
|
t.Run("HTTP", func(t *testing.T) {
|
|
t.Run("Entry", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "GET", root+"/"+cargo_service.BuildPackagePath(packageName))
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
var entry cargo_service.IndexVersionEntry
|
|
err := json.Unmarshal(resp.Body.Bytes(), &entry)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, packageName, entry.Name)
|
|
assert.Equal(t, packageVersion, entry.Version)
|
|
assert.Equal(t, pb.HashSHA256, entry.FileChecksum)
|
|
assert.False(t, entry.Yanked)
|
|
assert.Len(t, entry.Dependencies, 1)
|
|
dep := entry.Dependencies[0]
|
|
assert.Equal(t, "dep", dep.Name)
|
|
assert.Equal(t, "1.0", dep.Req)
|
|
assert.Equal(t, "normal", dep.Kind)
|
|
assert.True(t, dep.DefaultFeatures)
|
|
assert.Empty(t, dep.Features)
|
|
assert.False(t, dep.Optional)
|
|
assert.Nil(t, dep.Target)
|
|
assert.NotNil(t, dep.Registry)
|
|
assert.Equal(t, "https://gitea.io/user/_cargo-index", *dep.Registry)
|
|
assert.Nil(t, dep.Package)
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
t.Run("Download", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
pv, err := packages.GetVersionByNameAndVersion(t.Context(), user.ID, packages.TypeCargo, packageName, packageVersion)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, pv.DownloadCount)
|
|
|
|
pfs, err := packages.GetFilesByVersionID(t.Context(), pv.ID)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, pfs, 1)
|
|
|
|
req := NewRequest(t, "GET", fmt.Sprintf("%s/%s/%s/download", url, neturl.PathEscape(packageName), neturl.PathEscape(pv.Version))).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
assert.Equal(t, "test", resp.Body.String())
|
|
|
|
pv, err = packages.GetVersionByNameAndVersion(t.Context(), user.ID, packages.TypeCargo, packageName, packageVersion)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, pv.DownloadCount)
|
|
})
|
|
|
|
t.Run("Search", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
cases := []struct {
|
|
Query string
|
|
Page int
|
|
PerPage int
|
|
ExpectedTotal int64
|
|
ExpectedResults int
|
|
}{
|
|
{"", 0, 0, 1, 1},
|
|
{"", 1, 10, 1, 1},
|
|
{"cargo", 1, 0, 1, 1},
|
|
{"cargo", 1, 10, 1, 1},
|
|
{"cargo", 2, 10, 1, 0},
|
|
{"test", 0, 10, 0, 0},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
req := NewRequest(t, "GET", fmt.Sprintf("%s?q=%s&page=%d&per_page=%d", url, c.Query, c.Page, c.PerPage)).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
result := DecodeJSON(t, resp, &cargo_router.SearchResult{})
|
|
|
|
assert.Equal(t, c.ExpectedTotal, result.Meta.Total, "case %d: unexpected total hits", i)
|
|
assert.Len(t, result.Crates, c.ExpectedResults, "case %d: unexpected result count", i)
|
|
}
|
|
})
|
|
|
|
t.Run("Yank", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "DELETE", fmt.Sprintf("%s/%s/%s/yank", url, neturl.PathEscape(packageName), neturl.PathEscape(packageVersion))).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
status := DecodeJSON(t, resp, &cargo_router.StatusResponse{})
|
|
assert.True(t, status.OK)
|
|
|
|
content := readGitContent(t, cargo_service.BuildPackagePath(packageName))
|
|
|
|
var entry cargo_service.IndexVersionEntry
|
|
err := json.Unmarshal([]byte(content), &entry)
|
|
assert.NoError(t, err)
|
|
|
|
assert.True(t, entry.Yanked)
|
|
})
|
|
|
|
t.Run("Unyank", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "PUT", fmt.Sprintf("%s/%s/%s/unyank", url, neturl.PathEscape(packageName), neturl.PathEscape(packageVersion))).
|
|
AddBasicAuth(user.Name)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
status := DecodeJSON(t, resp, &cargo_router.StatusResponse{})
|
|
assert.True(t, status.OK)
|
|
|
|
content := readGitContent(t, cargo_service.BuildPackagePath(packageName))
|
|
|
|
var entry cargo_service.IndexVersionEntry
|
|
err := json.Unmarshal([]byte(content), &entry)
|
|
assert.NoError(t, err)
|
|
|
|
assert.False(t, entry.Yanked)
|
|
})
|
|
|
|
t.Run("ListOwners", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "GET", fmt.Sprintf("%s/%s/owners", url, neturl.PathEscape(packageName)))
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
owners := DecodeJSON(t, resp, &cargo_router.Owners{})
|
|
|
|
assert.Len(t, owners.Users, 1)
|
|
assert.Equal(t, user.ID, owners.Users[0].ID)
|
|
assert.Equal(t, user.Name, owners.Users[0].Login)
|
|
assert.Equal(t, user.DisplayName(), owners.Users[0].Name)
|
|
})
|
|
}
|