mirror of https://github.com/gogs/gogs.git
107 lines
3.6 KiB
Go
107 lines
3.6 KiB
Go
// Copyright 2020 The Gogs Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package lfs
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"gopkg.in/macaron.v1"
|
|
|
|
"gogs.io/gogs/internal/conf"
|
|
"gogs.io/gogs/internal/db"
|
|
"gogs.io/gogs/internal/lfsutil"
|
|
)
|
|
|
|
func Test_serveBatch(t *testing.T) {
|
|
conf.SetMockServer(t, conf.ServerOpts{
|
|
ExternalURL: "https://gogs.example.com/",
|
|
})
|
|
|
|
m := macaron.New()
|
|
m.Use(func(c *macaron.Context) {
|
|
c.Map(&db.User{Name: "owner"})
|
|
c.Map(&db.Repository{Name: "repo"})
|
|
})
|
|
m.Post("/", serveBatch)
|
|
|
|
tests := []struct {
|
|
name string
|
|
body string
|
|
mockLFSStore *db.MockLFSStore
|
|
expStatusCode int
|
|
expBody string
|
|
}{
|
|
{
|
|
name: "unrecognized operation",
|
|
body: `{"operation": "update"}`,
|
|
expStatusCode: http.StatusBadRequest,
|
|
expBody: `{"message":"Operation not recognized"}` + "\n",
|
|
},
|
|
{
|
|
name: "upload: contains invalid oid",
|
|
body: `{
|
|
"operation": "upload",
|
|
"objects": [
|
|
{"oid": "bad_oid", "size": 123},
|
|
{"oid": "ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f", "size": 123}
|
|
]}`,
|
|
expStatusCode: http.StatusOK,
|
|
expBody: `{"transfer":"basic","objects":[{"oid":"bad_oid","size":123,"actions":{"error":{"code":422,"message":"Object has invalid oid"}}},{"oid":"ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f","size":123,"actions":{"upload":{"href":"https://gogs.example.com/owner/repo.git/info/lfs/objects/basic/ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f"},"verify":{"href":"https://gogs.example.com/owner/repo.git/info/lfs/objects/basic/verify"}}}]}` + "\n",
|
|
},
|
|
{
|
|
name: "download: contains non-existent oid and mismatched size",
|
|
body: `{
|
|
"operation": "download",
|
|
"objects": [
|
|
{"oid": "bad_oid", "size": 123},
|
|
{"oid": "ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f", "size": 123},
|
|
{"oid": "5cac0a318669fadfee734fb340a5f5b70b428ac57a9f4b109cb6e150b2ba7e57", "size": 456}
|
|
]}`,
|
|
mockLFSStore: &db.MockLFSStore{
|
|
MockGetObjectsByOIDs: func(repoID int64, oids ...lfsutil.OID) ([]*db.LFSObject, error) {
|
|
return []*db.LFSObject{
|
|
{
|
|
OID: "ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f",
|
|
Size: 1234,
|
|
}, {
|
|
OID: "5cac0a318669fadfee734fb340a5f5b70b428ac57a9f4b109cb6e150b2ba7e57",
|
|
Size: 456,
|
|
},
|
|
}, nil
|
|
},
|
|
},
|
|
expStatusCode: http.StatusOK,
|
|
expBody: `{"transfer":"basic","objects":[{"oid":"bad_oid","size":123,"actions":{"error":{"code":404,"message":"Object does not exist"}}},{"oid":"ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f","size":123,"actions":{"error":{"code":422,"message":"Object size mismatch"}}},{"oid":"5cac0a318669fadfee734fb340a5f5b70b428ac57a9f4b109cb6e150b2ba7e57","size":456,"actions":{"download":{"href":"https://gogs.example.com/owner/repo.git/info/lfs/objects/basic/5cac0a318669fadfee734fb340a5f5b70b428ac57a9f4b109cb6e150b2ba7e57"}}}]}` + "\n",
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
db.SetMockLFSStore(t, test.mockLFSStore)
|
|
|
|
r, err := http.NewRequest("POST", "/", bytes.NewBufferString(test.body))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rr := httptest.NewRecorder()
|
|
m.ServeHTTP(rr, r)
|
|
|
|
resp := rr.Result()
|
|
assert.Equal(t, test.expStatusCode, resp.StatusCode)
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, test.expBody, string(body))
|
|
})
|
|
}
|
|
}
|