utils_test.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. package utils
  2. import (
  3. "regexp"
  4. "testing"
  5. "time"
  6. "github.com/stretchr/testify/assert"
  7. )
  8. func TestInitStructWithDefaultTag(t *testing.T) {
  9. config := &struct {
  10. B bool `default:"true"`
  11. S string `default:"default string"`
  12. I int `default:"10"`
  13. T time.Duration `default:"100"`
  14. E int `default:""`
  15. }{}
  16. InitStructWithDefaultTag(config)
  17. assert.NotNil(t, config)
  18. assert.Equal(t, true, config.B)
  19. assert.Equal(t, "default string", config.S)
  20. assert.Equal(t, 10, config.I)
  21. assert.Equal(t, time.Duration(100), config.T)
  22. assert.Equal(t, 0, config.E)
  23. }
  24. func TestGetUUID(t *testing.T) {
  25. uuid := NewUUID()
  26. assert.Equal(t, 16, len(uuid))
  27. assert.Equal(t, 36, len(uuid.String()))
  28. uuidString := GetUUID()
  29. assert.Equal(t, 32, len(uuidString))
  30. }
  31. func TestGetMD5Base64(t *testing.T) {
  32. assert.Equal(t, "ERIHLmRX2uZmssDdxQnnxQ==",
  33. GetMD5Base64([]byte("That's all folks!!")))
  34. assert.Equal(t, "GsJRdI3kAbAnHo/0+3wWJw==",
  35. GetMD5Base64([]byte("中文也没啥问题")))
  36. }
  37. func TestGetTimeInFormatRFC2616(t *testing.T) {
  38. s := GetTimeInFormatRFC2616()
  39. assert.Equal(t, 29, len(s))
  40. re := regexp.MustCompile(`^[A-Z][a-z]{2}, [0-9]{2} [A-Z][a-z]{2} [0-9]{4} [0-9]{2}:[0-9]{2}:[0-9]{2} GMT$`)
  41. assert.True(t, re.MatchString(s))
  42. }
  43. func TestGetTimeInFormatISO8601(t *testing.T) {
  44. s := GetTimeInFormatISO8601()
  45. assert.Equal(t, 20, len(s))
  46. // 2006-01-02T15:04:05Z
  47. re := regexp.MustCompile(`^[0-9]{4}-[0-9]{2}-[0-9]{2}T[0-9]{2}:[0-9]{2}:[0-9]{2}Z$`)
  48. assert.True(t, re.MatchString(s))
  49. }
  50. func TestGetUrlFormedMap(t *testing.T) {
  51. m := make(map[string]string)
  52. m["key"] = "value"
  53. s := GetUrlFormedMap(m)
  54. assert.Equal(t, "key=value", s)
  55. m["key2"] = "http://domain/?key=value&key2=value2"
  56. s2 := GetUrlFormedMap(m)
  57. assert.Equal(t, "key=value&key2=http%3A%2F%2Fdomain%2F%3Fkey%3Dvalue%26key2%3Dvalue2", s2)
  58. }