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]>
171 lines
5.2 KiB
Go
171 lines
5.2 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package integration
|
|
|
|
import (
|
|
"archive/tar"
|
|
"bytes"
|
|
"compress/gzip"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
auth_model "github.com/hanzoai/git/models/auth"
|
|
"github.com/hanzoai/git/models/packages"
|
|
"github.com/hanzoai/git/models/unittest"
|
|
user_model "github.com/hanzoai/git/models/user"
|
|
"github.com/hanzoai/git/modules/json"
|
|
vagrant_module "github.com/hanzoai/git/modules/packages/vagrant"
|
|
"github.com/hanzoai/git/tests"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPackageVagrant(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
token := "Bearer " + getUserToken(t, user.Name, auth_model.AccessTokenScopeWritePackage)
|
|
|
|
packageName := "test_package"
|
|
packageVersion := "1.0.1"
|
|
packageDescription := "Test Description"
|
|
packageProvider := "virtualbox"
|
|
|
|
filename := packageProvider + ".box"
|
|
|
|
infoContent, _ := json.Marshal(map[string]string{
|
|
"description": packageDescription,
|
|
})
|
|
|
|
var buf bytes.Buffer
|
|
zw := gzip.NewWriter(&buf)
|
|
archive := tar.NewWriter(zw)
|
|
archive.WriteHeader(&tar.Header{
|
|
Name: "info.json",
|
|
Mode: 0o600,
|
|
Size: int64(len(infoContent)),
|
|
})
|
|
archive.Write(infoContent)
|
|
archive.Close()
|
|
zw.Close()
|
|
content := buf.Bytes()
|
|
|
|
root := fmt.Sprintf("/v1/packages/%s/vagrant", user.Name)
|
|
|
|
t.Run("Authenticate", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
authenticateURL := root + "/authenticate"
|
|
|
|
req := NewRequest(t, "GET", authenticateURL)
|
|
MakeRequest(t, req, http.StatusUnauthorized)
|
|
|
|
req = NewRequest(t, "GET", authenticateURL).
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusOK)
|
|
})
|
|
|
|
boxURL := fmt.Sprintf("%s/%s", root, packageName)
|
|
|
|
t.Run("Upload", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "HEAD", boxURL)
|
|
MakeRequest(t, req, http.StatusNotFound)
|
|
|
|
uploadURL := fmt.Sprintf("%s/%s/%s", boxURL, packageVersion, filename)
|
|
|
|
req = NewRequestWithBody(t, "PUT", uploadURL, bytes.NewReader(content))
|
|
MakeRequest(t, req, http.StatusUnauthorized)
|
|
|
|
req = NewRequestWithBody(t, "PUT", uploadURL, bytes.NewReader(content)).
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusCreated)
|
|
|
|
req = NewRequest(t, "HEAD", boxURL)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
assert.True(t, strings.HasPrefix(resp.Header().Get("Content-Type"), "application/json"))
|
|
|
|
pvs, err := packages.GetVersionsByPackageType(t.Context(), user.ID, packages.TypeVagrant)
|
|
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, &vagrant_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, filename, pfs[0].Name)
|
|
assert.True(t, pfs[0].IsLead)
|
|
|
|
pb, err := packages.GetBlobByID(t.Context(), pfs[0].BlobID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(len(content)), pb.Size)
|
|
|
|
req = NewRequestWithBody(t, "PUT", uploadURL, bytes.NewReader(content)).
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusConflict)
|
|
})
|
|
|
|
t.Run("Download", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "GET", fmt.Sprintf("%s/%s/%s", boxURL, packageVersion, filename))
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
assert.Equal(t, content, resp.Body.Bytes())
|
|
})
|
|
|
|
t.Run("EnumeratePackageVersions", func(t *testing.T) {
|
|
defer tests.PrintCurrentTest(t)()
|
|
|
|
req := NewRequest(t, "GET", boxURL)
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
|
|
type providerData struct {
|
|
Name string `json:"name"`
|
|
URL string `json:"url"`
|
|
Checksum string `json:"checksum"`
|
|
ChecksumType string `json:"checksum_type"`
|
|
}
|
|
|
|
type versionMetadata struct {
|
|
Version string `json:"version"`
|
|
Status string `json:"status"`
|
|
DescriptionHTML string `json:"description_html,omitempty"`
|
|
DescriptionMarkdown string `json:"description_markdown,omitempty"`
|
|
Providers []*providerData `json:"providers"`
|
|
}
|
|
|
|
type packageMetadata struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description,omitempty"`
|
|
ShortDescription string `json:"short_description,omitempty"`
|
|
Versions []*versionMetadata `json:"versions"`
|
|
}
|
|
|
|
result := DecodeJSON(t, resp, &packageMetadata{})
|
|
|
|
assert.Equal(t, packageName, result.Name)
|
|
assert.Equal(t, packageDescription, result.Description)
|
|
assert.Len(t, result.Versions, 1)
|
|
version := result.Versions[0]
|
|
assert.Equal(t, packageVersion, version.Version)
|
|
assert.Equal(t, "active", version.Status)
|
|
assert.Len(t, version.Providers, 1)
|
|
provider := version.Providers[0]
|
|
assert.Equal(t, packageProvider, provider.Name)
|
|
assert.Equal(t, "sha512", provider.ChecksumType)
|
|
assert.Equal(t, "259bebd6160acad695016d22a45812e26f187aaf78e71a4c23ee3201528346293f991af3468a8c6c5d2a21d7d9e1bdc1bf79b87110b2fddfcc5a0d45963c7c30", provider.Checksum)
|
|
})
|
|
}
|