123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package lfs
-
- import (
- "bytes"
- "context"
- "io"
- "net/http"
- "strings"
- "testing"
-
- "code.gitea.io/gitea/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, MediaType, 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(bytes.NewBufferString("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)}
- } else {
- 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(context.Background(), c.link)
- if len(c.expectederror) > 0 {
- assert.True(t, strings.Contains(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(context.Background(), c.link, p, bytes.NewBufferString("dummy"))
- if len(c.expectederror) > 0 {
- assert.True(t, strings.Contains(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(context.Background(), c.link, p)
- if len(c.expectederror) > 0 {
- assert.True(t, strings.Contains(err.Error(), c.expectederror), "case %d: '%s' should contain '%s'", n, err.Error(), c.expectederror)
- } else {
- assert.NoError(t, err, "case %d", n)
- }
- }
- })
- }
|