Слияние кода завершено, страница обновится автоматически
package main
import (
"bytes"
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"net/http/httptest"
"os"
"testing"
)
func TestGetAuthed(t *testing.T) {
res, err := api("GET", "/user/repo/objects/"+contentOid, contentMediaType, testUser, testPass, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
by, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("expected response to contain content, got error: %s", err)
}
if string(by) != content {
t.Fatalf("expected content to be `content`, got: %s", string(by))
}
}
func TestGetAuthedWithRange(t *testing.T) {
req, err := http.NewRequest("GET", lfsServer.URL+"/user/repo/objects/"+contentOid, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
req.SetBasicAuth(testUser, testPass)
req.Header.Set("Accept", contentMediaType)
fromByte := 5
req.Header.Set("Range", fmt.Sprintf("bytes=%d-", fromByte))
res, err := http.DefaultClient.Do(req)
if err != nil {
t.Fatalf("response error: %s", err)
}
if res.StatusCode != 206 {
t.Fatalf("expected status 206, got %d", res.StatusCode)
}
if cr := res.Header.Get("Content-Range"); len(cr) > 0 {
expected := fmt.Sprintf("bytes %d-%d/%d", fromByte, len(content)-1, len(content)-fromByte)
if cr != expected {
t.Fatalf("expected Content-Range header of %q, got %q", expected, cr)
}
} else {
t.Fatalf("missing Content-Range header in response")
}
by, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("expected response to contain content, got error: %s", err)
}
if string(by) != content[fromByte:] {
t.Fatalf("expected content to be `content`, got: %s", string(by))
}
}
func TestGetUnAuthed(t *testing.T) {
res, err := api("GET", "/user/repo/objects/"+contentOid, contentMediaType, "", "", nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestGetBadAuth(t *testing.T) {
res, err := api("GET", "/user/repo/objects/"+contentOid, contentMediaType, testUser, testPass+"123", nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestGetMetaAuthed(t *testing.T) {
res, err := api("GET", "/bilbo/repo/objects/"+contentOid, metaMediaType, testUser, testPass, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
var meta Representation
dec := json.NewDecoder(res.Body)
dec.Decode(&meta)
if meta.Oid != contentOid {
t.Fatalf("expected to see oid `%s` in meta, got: `%s`", contentOid, meta.Oid)
}
if meta.Size != contentSize {
t.Fatalf("expected to see a size of `%d`, got: `%d`", contentSize, meta.Size)
}
download := meta.Actions["download"]
if download.Href != "http://localhost:8080/bilbo/repo/objects/"+contentOid {
t.Fatalf("expected download link, got %s", download.Href)
}
}
func TestGetMetaUnAuthed(t *testing.T) {
res, err := api("GET", "/user/repo/objects/"+contentOid, metaMediaType, "", "", nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestPostAuthedNewObject(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"oid":"%s", "size":1234}`, nonExistingOid))
res, err := api("POST", "/bilbo/repo/objects", metaMediaType, testUser, testPass, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 202 {
t.Fatalf("expected status 202, got %d", res.StatusCode)
}
var meta Representation
dec := json.NewDecoder(res.Body)
dec.Decode(&meta)
if meta.Oid != nonExistingOid {
t.Fatalf("expected to see oid `%s` in meta, got: `%s`", nonExistingOid, meta.Oid)
}
if meta.Size != 1234 {
t.Fatalf("expected to see a size of `1234`, got: `%d`", meta.Size)
}
if download, ok := meta.Actions["download"]; ok {
t.Fatalf("expected POST to not contain a download link, got %s", download.Href)
}
upload, ok := meta.Actions["upload"]
if !ok {
t.Fatal("expected upload link to be present")
}
if upload.Href != "http://localhost:8080/bilbo/repo/objects/"+nonExistingOid {
t.Fatalf("expected upload link, got %s", upload.Href)
}
}
func TestPostAuthedExistingObject(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"oid":"%s", "size":%d}`, contentOid, contentSize))
res, err := api("POST", "/bilbo/repo/objects", metaMediaType, testUser, testPass, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
var meta Representation
dec := json.NewDecoder(res.Body)
dec.Decode(&meta)
if meta.Oid != contentOid {
t.Fatalf("expected to see oid `%s` in meta, got: `%s`", contentOid, meta.Oid)
}
if meta.Size != contentSize {
t.Fatalf("expected to see a size of `%d`, got: `%d`", contentSize, meta.Size)
}
download := meta.Actions["download"]
if download.Href != "http://localhost:8080/bilbo/repo/objects/"+contentOid {
t.Fatalf("expected download link, got %s", download.Href)
}
upload, ok := meta.Actions["upload"]
if !ok {
t.Fatalf("expected upload link to be present")
}
if upload.Href != "http://localhost:8080/bilbo/repo/objects/"+contentOid {
t.Fatalf("expected upload link, got %s", upload.Href)
}
}
func TestPostUnAuthed(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"oid":"%s", "size":%d}`, contentOid, contentSize))
res, err := api("POST", "/bilbo/readonly/objects", metaMediaType, "", "", buf)
if err != nil {
t.Fatalf("response error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestPut(t *testing.T) {
req, err := http.NewRequest("PUT", lfsServer.URL+"/user/repo/objects/"+contentOid, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
req.SetBasicAuth(testUser, testPass)
req.Header.Set("Accept", contentMediaType)
req.Header.Set("Content-Type", "application/octet-stream")
req.Body = ioutil.NopCloser(bytes.NewBuffer([]byte(content)))
res, err := http.DefaultClient.Do(req)
if err != nil {
t.Fatalf("response error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
r, err := testContentStore.Get(&MetaObject{Oid: contentOid}, 0)
if err != nil {
t.Fatalf("error retreiving from content store: %s", err)
} else {
defer r.Close()
}
c, err := ioutil.ReadAll(r)
if err != nil {
t.Fatalf("error reading content: %s", err)
}
if string(c) != content {
t.Fatalf("expected content, got `%s`", string(c))
}
}
func TestMediaTypesRequired(t *testing.T) {
m := []string{"GET", "PUT", "POST", "HEAD"}
for _, method := range m {
res, err := api(method, "/user/repo/objects/"+contentOid, "", testUser, testPass, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 404 {
t.Fatalf("expected status 404, got %d", res.StatusCode)
}
}
}
func TestMediaTypesParsed(t *testing.T) {
accept := contentMediaType + "; charset=utf-8"
res, err := api("GET", "/user/repo/objects/"+contentOid, accept, testUser, testPass, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
}
func TestLocksList(t *testing.T) {
res, err := api("GET", "/user/repo/locks", metaMediaType, testUser, testPass, nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
body, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("expected response to contain content, got error: %s", err)
}
var list LockList
if err := json.Unmarshal(body, &list); err != nil {
t.Fatalf("expected response body to be LockList, got error: %s", err)
}
if len(list.Locks) != 1 {
t.Errorf("expected returned lock count to match, got: %d", len(list.Locks))
}
if list.Locks[0].Id != lockId {
t.Errorf("expected lockId to match, got: %s", list.Locks[0].Id)
}
}
func TestLocksListUnAuthed(t *testing.T) {
res, err := api("GET", "/user/repo/locks", metaMediaType, "", "", nil)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestLocksVerify(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"cursor": "", "limit": 0}`))
res, err := api("POST", "/user/repo/locks/verify", metaMediaType, testUser, testPass, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
body, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("expected response to contain content, got error: %s", err)
}
var list VerifiableLockList
if err := json.Unmarshal(body, &list); err != nil {
t.Fatalf("expected response body to be VerifiableLockList, got error: %s", err)
}
}
func TestLocksVerifyUnAuthed(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"cursor": "", "limit": 0}`))
res, err := api("POST", "/user/repo/locks/verify", metaMediaType, "", "", buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestLock(t *testing.T) {
path := "TestLock"
lock, err := createLock(testUser, testPass, path)
if err != nil {
t.Fatalf("create lock error: %s", err)
}
if lock == nil {
t.Errorf("expected lock to be created, got: %s", lock)
}
if lock.Owner.Name != testUser {
t.Errorf("expected lock owner to be match, got: %s", lock.Owner.Name)
}
if lock.Path != path {
t.Errorf("expected lock path to be match, got: %s", lock.Path)
}
}
func TestLockExists(t *testing.T) {
l, err := createLock(testUser, testPass, "TestLockExists")
if err != nil {
t.Fatalf("create lock error: %s", err)
}
buf := bytes.NewBufferString(fmt.Sprintf(`{"path":"%s"}`, l.Path))
res, err := api("POST", "/user/repo/locks", metaMediaType, testUser, testPass, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 409 {
t.Fatalf("expected status 409, got %d", res.StatusCode)
}
}
func TestLockUnAuthed(t *testing.T) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"path":"%s"}`, "TestLockUnAuthed"))
res, err := api("POST", "/user/repo/locks", metaMediaType, "", "", buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestUnlock(t *testing.T) {
l, err := createLock(testUser, testPass, "TestUnlock")
if err != nil {
t.Fatalf("create lock error: %s", err)
}
buf := bytes.NewBufferString(fmt.Sprintf(`{"force": %t}`, false))
res, err := api("POST", "/user/repo/locks/"+l.Id+"/unlock", metaMediaType, testUser, testPass, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
body, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("expected response to contain content, got error: %s", err)
}
var unlockResponse UnlockResponse
if err := json.Unmarshal(body, &unlockResponse); err != nil {
t.Fatalf("expected response body to be UnlockResponse, got error: %s", err)
}
lock := unlockResponse.Lock
if lock == nil || lock.Id != l.Id {
t.Errorf("expected deleted lock to be returned, got: %s", lock)
}
}
func TestUnLockUnAuthed(t *testing.T) {
l, err := createLock(testUser, testPass, "TestUnLockUnAuthed")
if err != nil {
t.Fatalf("create lock error: %s", err)
}
buf := bytes.NewBufferString(fmt.Sprintf(`{"force": %t}`, false))
res, err := api("POST", "/user/repo/locks/"+l.Id+"/unlock", metaMediaType, "", "", buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 401 {
t.Fatalf("expected status 401, got %d", res.StatusCode)
}
}
func TestUnlockNotOwner(t *testing.T) {
l, err := createLock(testUser, testPass, "TestUnlockNotOwner")
if err != nil {
t.Fatalf("create lock error: %s", err)
}
buf := bytes.NewBufferString(fmt.Sprintf(`{"force": %t}`, false))
res, err := api("POST", "/user/repo/locks/"+l.Id+"/unlock", metaMediaType, testUser1, testPass1, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 403 {
t.Fatalf("expected status 403, got %d", res.StatusCode)
}
}
func TestUnlockNotOwnerForce(t *testing.T) {
l, err := createLock(testUser, testPass, "TestUnlockNotOwnerForce")
if err != nil {
t.Fatalf("create lock error: %s", err)
}
buf := bytes.NewBufferString(fmt.Sprintf(`{"force": %t}`, true))
res, err := api("POST", "/user/repo/locks/"+l.Id+"/unlock", metaMediaType, testUser1, testPass1, buf)
if err != nil {
t.Fatalf("request error: %s", err)
}
if res.StatusCode != 200 {
t.Fatalf("expected status 200, got %d", res.StatusCode)
}
}
func createLock(username, password, path string) (*Lock, error) {
buf := bytes.NewBufferString(fmt.Sprintf(`{"path":"%s"}`, path))
res, err := api("POST", "/user/repo/locks", metaMediaType, username, password, buf)
if err != nil {
return nil, fmt.Errorf("request error: %s", err)
}
if res.StatusCode != 201 {
return nil, fmt.Errorf("expected status 201, got %d", res.StatusCode)
}
body, err := ioutil.ReadAll(res.Body)
if err != nil {
return nil, fmt.Errorf("expected response to contain content, got error: %s", err)
}
var lockResponse LockResponse
if err := json.Unmarshal(body, &lockResponse); err != nil {
return nil, fmt.Errorf("expected response body to be LockResponse, got error: %s", err)
}
return lockResponse.Lock, nil
}
// simple http client for making api request
func api(method, path, accept, username, password string, body *bytes.Buffer) (*http.Response, error) {
req, err := http.NewRequest(method, lfsServer.URL+path, nil)
if err != nil {
return nil, err
}
if accept != "" {
req.Header.Set("Accept", accept)
}
if username != "" || password != "" {
req.SetBasicAuth(username, password)
}
if (method == "POST" || method == "PUT") && body != nil {
req.Body = ioutil.NopCloser(body)
}
return http.DefaultClient.Do(req)
}
var (
lfsServer *httptest.Server
testMetaStore *MetaStore
testContentStore *ContentStore
)
const (
testUser = "bilbo"
testPass = "baggins"
testUser1 = "bilbo1"
testPass1 = "baggins1"
testRepo = "repo"
content = "this is my content"
contentSize = int64(len(content))
contentOid = "f97e1b2936a56511b3b6efc99011758e4700d60fb1674d31445d1ee40b663f24"
nonExistingOid = "aec070645fe53ee3b3763059376134f058cc337247c978add178b6ccdfb0019f"
lockId = "3cfec93346f7ff337c60f2da50cd86740715e2f6"
nonExistingLockId = "f310c1555a2485e2e5229ea015a94c9d590763d3"
lockPath = "this/is/lock/path"
)
func TestMain(m *testing.M) {
os.Remove("lfs-test.db")
var err error
testMetaStore, err = NewMetaStore("lfs-test.db")
if err != nil {
fmt.Printf("Error creating meta store: %s", err)
os.Exit(1)
}
testContentStore, err = NewContentStore("lfs-content-test")
if err != nil {
fmt.Printf("Error creating content store: %s", err)
os.Exit(1)
}
if err := seedMetaStore(); err != nil {
fmt.Printf("Error seeding meta store: %s", err)
os.Exit(1)
}
if err := seedContentStore(); err != nil {
fmt.Printf("Error seeding content store: %s", err)
os.Exit(1)
}
app := NewApp(testContentStore, testMetaStore)
lfsServer = httptest.NewServer(app)
logger = NewKVLogger(ioutil.Discard)
ret := m.Run()
lfsServer.Close()
testMetaStore.Close()
os.Remove("lfs-test.db")
os.RemoveAll("lfs-content-test")
os.Exit(ret)
}
func seedMetaStore() error {
if err := testMetaStore.AddUser(testUser, testPass); err != nil {
return err
}
if err := testMetaStore.AddUser(testUser1, testPass1); err != nil {
return err
}
rv := &RequestVars{Oid: contentOid, Size: contentSize}
if _, err := testMetaStore.Put(rv); err != nil {
return err
}
lock := NewTestLock(lockId, lockPath, testUser)
if err := testMetaStore.AddLocks(testRepo, lock); err != nil {
return err
}
return nil
}
func seedContentStore() error {
meta := &MetaObject{Oid: contentOid, Size: contentSize}
buf := bytes.NewBuffer([]byte(content))
if err := testContentStore.Put(meta, buf); err != nil {
return err
}
return nil
}
Вы можете оставить комментарий после Вход в систему
Неприемлемый контент может быть отображен здесь и не будет показан на странице. Вы можете проверить и изменить его с помощью соответствующей функции редактирования.
Если вы подтверждаете, что содержание не содержит непристойной лексики/перенаправления на рекламу/насилия/вульгарной порнографии/нарушений/пиратства/ложного/незначительного или незаконного контента, связанного с национальными законами и предписаниями, вы можете нажать «Отправить» для подачи апелляции, и мы обработаем ее как можно скорее.
Опубликовать ( 0 )