package xfs

import (
	"fmt"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"github.com/stretchr/testify/assert"
	"github.com/yaoapp/gou"
	"github.com/yaoapp/kun/any"
	"github.com/yaoapp/yao/share"
)

func init() {
	share.App = share.AppInfo{
		Storage: share.AppStorage{
			Default: "oss",
			OSS: &share.AppStorageOSS{
				Endpoint:    "oss-cn-hangzhou.aliyuncs.com",
				ID:          os.Getenv("OSS_TEST_ID"),
				Secret:      os.Getenv("OSS_TEST_SECRET"),
				RoleArn:     "acs:ram::31524094:role/ramosstest",
				SessionName: "SessionTest",
			},
		},
	}
}
func TestProcessGetToken(t *testing.T) {

	args := []interface{}{"oss"}
	process := gou.NewProcess("xiang.fs.GetToken", args...)
	response := processGetToken(process)
	assert.NotNil(t, response)
	res := any.Of(response).Map()
	assert.True(t, res.Has("AccessKeyId"))
	assert.True(t, res.Has("AccessKeySecret"))
	assert.True(t, res.Has("Expiration"))
	assert.True(t, res.Has("SecurityToken"))
	assert.True(t, res.Has("Endpoint"))

	// 使用token
	client, err := oss.New(
		res.Get("Endpoint").(string),
		res.Get("AccessKeyId").(string),
		res.Get("AccessKeySecret").(string),
		oss.SecurityToken(res.Get("SecurityToken").(string)),
	)
	assert.Nil(t, err)

	bucket, err := client.Bucket("image-appcook")
	assert.Nil(t, err)

	// 上传字符串。
	now := fmt.Sprintf("%d", time.Now().UnixNano())
	err = bucket.PutObject("xiang/unit-test."+now+".txt", strings.NewReader(now))
	assert.Nil(t, err)
}