eiblog/vendor/github.com/qiniu/api.v7/kodo/token_test.go

package kodo

import (
	"crypto/sha1"
	"encoding/base64"
	"io"
	"net/http"
	"testing"
)

func init() {

	if skipTest() {
		return
	}

	// 删除 可能存在的 key
	bucket.BatchDelete(nil, key)
}

func TestGetPrivateUrl(t *testing.T) {

	if skipTest() {
		return
	}

	// 上传一个文件用用于测试
	err := upFile("token.go", key)
	if err != nil {
		t.Fatal(err)
	}
	defer bucket.Delete(nil, key)

	baseUrl := MakeBaseUrl(domain, key)
	privateUrl := client.MakePrivateUrl(baseUrl, nil)

	resp, err := http.Get(privateUrl)
	if err != nil {
		t.Fatal("http.Get failed:", err)
	}
	defer resp.Body.Close()

	h := sha1.New()
	io.Copy(h, resp.Body)
	etagExpected := base64.URLEncoding.EncodeToString(h.Sum([]byte{'\x16'}))

	etag := resp.Header.Get("Etag")
	if etag[1:len(etag)-1] != etagExpected {
		t.Fatal("http.Get etag failed:", etag, etagExpected)
	}
}

func testClient_MakeUptokenBucket(t *testing.T) {
	c := New(0, nil)
	token := c.MakeUptoken(&PutPolicy{
		Scope:   "gosdk",
		Expires: 3600,
	})
	if token == "" {
		t.Fatal("nil token")
	}

	token, err := c.MakeUptokenWithSafe(&PutPolicy{
		Scope:   "NotExistBucket",
		Expires: 3600,
	})
	if err == nil {
		t.Fatal("make up token fail")
	}
}