cloudreve/pkg/serializer/upload_test.go

package serializer

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestDecodeUploadPolicy(t *testing.T) {
	asserts := assert.New(t)

	testCases := []struct {
		input       string
		expectError bool
		expectNil   bool
		expectRes   *UploadPolicy
	}{
		{
			"错误的base64字符",
			true,
			true,
			&UploadPolicy{},
		},
		{
			"6ZSZ6K+v55qESlNPTuWtl+espg==",
			true,
			true,
			&UploadPolicy{},
		},
		{
			"e30=",
			false,
			false,
			&UploadPolicy{},
		},
		{
			"eyJjYWxsYmFja191cmwiOiJ0ZXN0In0=",
			false,
			false,
			&UploadPolicy{CallbackURL: "test"},
		},
	}

	for _, testCase := range testCases {
		res, err := DecodeUploadPolicy(testCase.input)
		if testCase.expectError {
			asserts.Error(err)
		}
		if testCase.expectNil {
			asserts.Nil(res)
		}
		if !testCase.expectNil {
			asserts.Equal(testCase.expectRes, res)
		}
	}
}

func TestUploadPolicy_EncodeUploadPolicy(t *testing.T) {
	asserts := assert.New(t)
	testPolicy := UploadPolicy{}
	res, err := testPolicy.EncodeUploadPolicy()
	asserts.NoError(err)
	asserts.NotEmpty(res)
}