diff --git a/pkg/hashid/hash_test.go b/pkg/hashid/hash_test.go new file mode 100644 index 0000000..5e8716c --- /dev/null +++ b/pkg/hashid/hash_test.go @@ -0,0 +1,32 @@ +package hashid + +import ( + "github.com/stretchr/testify/assert" + "testing" +) + +func TestHashEncode(t *testing.T) { + asserts := assert.New(t) + + { + res, err := HashEncode([]int{1, 2, 3}) + asserts.NoError(err) + asserts.NotEmpty(res) + } + + { + res, err := HashEncode([]int{}) + asserts.Error(err) + asserts.Empty(res) + } + +} + +func TestHashID(t *testing.T) { + asserts := assert.New(t) + + { + res := HashID(1, ShareID) + asserts.NotEmpty(res) + } +} diff --git a/service/share/manage.go b/service/share/manage.go index ebe483b..f2f3607 100644 --- a/service/share/manage.go +++ b/service/share/manage.go @@ -51,7 +51,7 @@ func (service *ShareCreateService) Create(c *gin.Context) serializer.Response { IsDir: service.IsDir, UserID: user.ID, SourceID: service.SourceID, - Score: service.RemainDownloads, + Score: service.Score, } // 如果开启了自动过期 @@ -78,4 +78,5 @@ func (service *ShareCreateService) Create(c *gin.Context) serializer.Response { Code: 0, Data: shareURL.String(), } + }