mirror of
https://github.com/halejohn/Cloudreve.git
synced 2026-01-26 09:34:57 +08:00
Fix: rand seeding error / Test: conf/version and util
This commit is contained in:
27
pkg/conf/version_test.go
Normal file
27
pkg/conf/version_test.go
Normal file
@@ -0,0 +1,27 @@
|
||||
package conf
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestWriteVersionLock(t *testing.T) {
|
||||
asserts := assert.New(t)
|
||||
|
||||
// 清理残余文件
|
||||
if _, err := os.Stat("version.lock"); !os.IsNotExist(err) {
|
||||
err = os.Remove("version.lock")
|
||||
asserts.NoError(err)
|
||||
}
|
||||
|
||||
err := WriteVersionLock()
|
||||
defer func() { err = os.Remove("version.lock") }()
|
||||
writtenVersion, err := ioutil.ReadFile("version.lock")
|
||||
|
||||
// 写入的版本应与当前版本相同
|
||||
asserts.NoError(err)
|
||||
asserts.Equal(string(writtenVersion), BackendVersion)
|
||||
|
||||
}
|
||||
@@ -2,14 +2,12 @@ package util
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"time"
|
||||
)
|
||||
|
||||
// RandStringRunes 返回随机字符串
|
||||
func RandStringRunes(n int) string {
|
||||
var letterRunes = []rune("1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
|
||||
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
b := make([]rune, n)
|
||||
for i := range b {
|
||||
b[i] = letterRunes[rand.Intn(len(letterRunes))]
|
||||
|
||||
27
pkg/util/common_test.go
Normal file
27
pkg/util/common_test.go
Normal file
@@ -0,0 +1,27 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestRandStringRunes(t *testing.T) {
|
||||
asserts := assert.New(t)
|
||||
|
||||
// 0 长度字符
|
||||
randStr := RandStringRunes(0)
|
||||
asserts.Len(randStr, 0)
|
||||
|
||||
// 16 长度字符
|
||||
randStr = RandStringRunes(16)
|
||||
asserts.Len(randStr, 16)
|
||||
|
||||
// 32 长度字符
|
||||
randStr = RandStringRunes(32)
|
||||
asserts.Len(randStr, 32)
|
||||
|
||||
//相同长度字符
|
||||
sameLenStr1 := RandStringRunes(32)
|
||||
sameLenStr2 := RandStringRunes(32)
|
||||
asserts.NotEqual(sameLenStr1, sameLenStr2)
|
||||
}
|
||||
Reference in New Issue
Block a user