27 lines
640 B
Go
27 lines
640 B
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGenerateToken(t *testing.T) {
|
|
m := GenerateToken()
|
|
assert.Equal(t, 8, len(m))
|
|
}
|
|
|
|
func TestInvalidDownloadToken(t *testing.T) {
|
|
assert.True(t, InvalidDownloadToken("xxx"))
|
|
assert.True(t, InvalidDownloadToken("1ud3qfpu"))
|
|
DownloadToken.Set("1ud3qfpu", "1ud3qfpu")
|
|
assert.False(t, InvalidDownloadToken("1ud3qfpu"))
|
|
}
|
|
|
|
func TestInvalidPreviewToken(t *testing.T) {
|
|
assert.True(t, InvalidPreviewToken("xxx"))
|
|
assert.True(t, InvalidPreviewToken("2ud3qfpu"))
|
|
PreviewToken.Set("2ud3qfpu", "2ud3qfpu")
|
|
assert.False(t, InvalidPreviewToken("2ud3qfpu"))
|
|
}
|