mirror of
				https://github.com/cloudreve/cloudreve.git
				synced 2025-10-31 00:27:31 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			43 lines
		
	
	
		
			920 B
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			43 lines
		
	
	
		
			920 B
		
	
	
	
		
			Go
		
	
	
	
	
	
| package qiniu
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	model "github.com/HFO4/cloudreve/models"
 | |
| 	"github.com/HFO4/cloudreve/pkg/cache"
 | |
| 	"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 	"testing"
 | |
| )
 | |
| 
 | |
| func TestHandler_Token(t *testing.T) {
 | |
| 	asserts := assert.New(t)
 | |
| 	handler := Handler{
 | |
| 		Policy: &model.Policy{
 | |
| 			MaxSize: 10,
 | |
| 			OptionsSerialized: model.PolicyOption{
 | |
| 				MimeType: "ss",
 | |
| 			},
 | |
| 			AccessKey: "ak",
 | |
| 			SecretKey: "sk",
 | |
| 			Server:    "http://test.com",
 | |
| 		},
 | |
| 	}
 | |
| 	ctx := context.Background()
 | |
| 
 | |
| 	// 成功
 | |
| 	{
 | |
| 		cache.Set("setting_siteURL", "http://test.cloudreve.org", 0)
 | |
| 		ctx = context.WithValue(ctx, fsctx.SavePathCtx, "/123")
 | |
| 		_, err := handler.Token(ctx, 10, "123")
 | |
| 		asserts.NoError(err)
 | |
| 	}
 | |
| 
 | |
| 	// 上下文无存储路径
 | |
| 	{
 | |
| 		ctx = context.Background()
 | |
| 		cache.Set("setting_siteURL", "http://test.cloudreve.org", 0)
 | |
| 		_, err := handler.Token(ctx, 10, "123")
 | |
| 		asserts.Error(err)
 | |
| 	}
 | |
| }
 | 
