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]>
172 lines
4.3 KiB
Go
172 lines
4.3 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package lfs
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/git/modules/json"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBasicTransferAdapterName(t *testing.T) {
|
|
a := &BasicTransferAdapter{}
|
|
|
|
assert.Equal(t, "basic", a.Name())
|
|
}
|
|
|
|
func TestBasicTransferAdapter(t *testing.T) {
|
|
p := Pointer{Oid: "b5a2c96250612366ea272ffac6d9744aaf4b45aacd96aa7cfcb931ee3b558259", Size: 5}
|
|
|
|
roundTripHandler := func(req *http.Request) *http.Response {
|
|
assert.Equal(t, AcceptHeader, req.Header.Get("Accept"))
|
|
assert.Equal(t, "test-value", req.Header.Get("test-header"))
|
|
|
|
url := req.URL.String()
|
|
if strings.Contains(url, "download-request") {
|
|
assert.Equal(t, "GET", req.Method)
|
|
|
|
return &http.Response{StatusCode: http.StatusOK, Body: io.NopCloser(strings.NewReader("dummy"))}
|
|
} else if strings.Contains(url, "upload-request") {
|
|
assert.Equal(t, "PUT", req.Method)
|
|
assert.Equal(t, "application/octet-stream", req.Header.Get("Content-Type"))
|
|
|
|
b, err := io.ReadAll(req.Body)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "dummy", string(b))
|
|
|
|
return &http.Response{StatusCode: http.StatusOK}
|
|
} else if strings.Contains(url, "verify-request") {
|
|
assert.Equal(t, "POST", req.Method)
|
|
assert.Equal(t, MediaType, req.Header.Get("Content-Type"))
|
|
|
|
var vp Pointer
|
|
err := json.NewDecoder(req.Body).Decode(&vp)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, p.Oid, vp.Oid)
|
|
assert.Equal(t, p.Size, vp.Size)
|
|
|
|
return &http.Response{StatusCode: http.StatusOK}
|
|
} else if strings.Contains(url, "error-response") {
|
|
er := &ErrorResponse{
|
|
Message: "Object not found",
|
|
}
|
|
payload := new(bytes.Buffer)
|
|
json.NewEncoder(payload).Encode(er)
|
|
|
|
return &http.Response{StatusCode: http.StatusNotFound, Body: io.NopCloser(payload)}
|
|
}
|
|
t.Errorf("Unknown test case: %s", url)
|
|
return nil
|
|
}
|
|
|
|
hc := &http.Client{Transport: RoundTripFunc(roundTripHandler)}
|
|
a := &BasicTransferAdapter{hc}
|
|
|
|
t.Run("Download", func(t *testing.T) {
|
|
cases := []struct {
|
|
link *Link
|
|
expectederror string
|
|
}{
|
|
// case 0
|
|
{
|
|
link: &Link{
|
|
Href: "https://download-request.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "",
|
|
},
|
|
// case 1
|
|
{
|
|
link: &Link{
|
|
Href: "https://error-response.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "Object not found",
|
|
},
|
|
}
|
|
|
|
for n, c := range cases {
|
|
_, err := a.Download(t.Context(), c.link)
|
|
if len(c.expectederror) > 0 {
|
|
assert.Contains(t, err.Error(), c.expectederror, "case %d: '%s' should contain '%s'", n, err.Error(), c.expectederror)
|
|
} else {
|
|
assert.NoError(t, err, "case %d", n)
|
|
}
|
|
}
|
|
})
|
|
|
|
t.Run("Upload", func(t *testing.T) {
|
|
cases := []struct {
|
|
link *Link
|
|
expectederror string
|
|
}{
|
|
// case 0
|
|
{
|
|
link: &Link{
|
|
Href: "https://upload-request.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "",
|
|
},
|
|
// case 1
|
|
{
|
|
link: &Link{
|
|
Href: "https://error-response.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "Object not found",
|
|
},
|
|
}
|
|
|
|
for n, c := range cases {
|
|
err := a.Upload(t.Context(), c.link, p, strings.NewReader("dummy"))
|
|
if len(c.expectederror) > 0 {
|
|
assert.Contains(t, err.Error(), c.expectederror, "case %d: '%s' should contain '%s'", n, err.Error(), c.expectederror)
|
|
} else {
|
|
assert.NoError(t, err, "case %d", n)
|
|
}
|
|
}
|
|
})
|
|
|
|
t.Run("Verify", func(t *testing.T) {
|
|
cases := []struct {
|
|
link *Link
|
|
expectederror string
|
|
}{
|
|
// case 0
|
|
{
|
|
link: &Link{
|
|
Href: "https://verify-request.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "",
|
|
},
|
|
// case 1
|
|
{
|
|
link: &Link{
|
|
Href: "https://error-response.io",
|
|
Header: map[string]string{"test-header": "test-value"},
|
|
},
|
|
expectederror: "Object not found",
|
|
},
|
|
}
|
|
|
|
for n, c := range cases {
|
|
err := a.Verify(t.Context(), c.link, p)
|
|
if len(c.expectederror) > 0 {
|
|
assert.Contains(t, err.Error(), c.expectederror, "case %d: '%s' should contain '%s'", n, err.Error(), c.expectederror)
|
|
} else {
|
|
assert.NoError(t, err, "case %d", n)
|
|
}
|
|
}
|
|
})
|
|
}
|