在Go中轻松安全地从一种数据类型转换为另一种数据类型
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

118 lines
3.0 KiB

  1. // Copyright © 2014 Steve Francia <spf@spf13.com>.
  2. //
  3. // Use of this source code is governed by an MIT-style
  4. // license that can be found in the LICENSE file.
  5. package cast
  6. import (
  7. "testing"
  8. "html/template"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. func TestToInt(t *testing.T) {
  12. var eight interface{} = 8
  13. assert.Equal(t, ToInt(8), 8)
  14. assert.Equal(t, ToInt(8.31), 8)
  15. assert.Equal(t, ToInt("8"), 8)
  16. assert.Equal(t, ToInt(true), 1)
  17. assert.Equal(t, ToInt(false), 0)
  18. assert.Equal(t, ToInt(eight), 8)
  19. }
  20. func TestToFloat64(t *testing.T) {
  21. var eight interface{} = 8
  22. assert.Equal(t, ToFloat64(8), 8.00)
  23. assert.Equal(t, ToFloat64(8.31), 8.31)
  24. assert.Equal(t, ToFloat64("8.31"), 8.31)
  25. assert.Equal(t, ToFloat64(eight), 8.0)
  26. }
  27. func TestToString(t *testing.T) {
  28. var foo interface{} = "one more time"
  29. assert.Equal(t, ToString(8), "8")
  30. assert.Equal(t, ToString(8.12), "8.12")
  31. assert.Equal(t, ToString([]byte("one time")), "one time")
  32. assert.Equal(t, ToString(template.HTML("one time")), "one time")
  33. assert.Equal(t, ToString(foo), "one more time")
  34. assert.Equal(t, ToString(nil), "")
  35. }
  36. type foo struct {
  37. val string
  38. }
  39. func (x foo) String() string {
  40. return x.val
  41. }
  42. func TestStringerToString(t *testing.T) {
  43. var x foo
  44. x.val = "bar"
  45. assert.Equal(t, "bar", ToString(x))
  46. }
  47. type fu struct {
  48. val string
  49. }
  50. func (x fu) Error() string {
  51. return x.val
  52. }
  53. func TestErrorToString(t *testing.T) {
  54. var x fu
  55. x.val = "bar"
  56. assert.Equal(t, "bar", ToString(x))
  57. }
  58. func TestMaps(t *testing.T) {
  59. var taxonomies = map[interface{}]interface{}{"tag": "tags", "group": "groups"}
  60. var stringMapBool = map[interface{}]interface{}{"v1": true, "v2": false}
  61. assert.Equal(t, ToStringMap(taxonomies), map[string]interface{}{"tag": "tags", "group": "groups"})
  62. assert.Equal(t, ToStringMapBool(stringMapBool), map[string]bool{"v1": true, "v2": false})
  63. }
  64. func TestSlices(t *testing.T) {
  65. assert.Equal(t, []string{"a", "b"}, ToStringSlice([]string{"a", "b"}))
  66. assert.Equal(t, []string{"1", "3"}, ToStringSlice([]interface{}{1, 3}))
  67. assert.Equal(t, []int{1, 3}, ToIntSlice([]int{1, 3}))
  68. assert.Equal(t, []int{1, 3}, ToIntSlice([]interface{}{1.2, 3.2}))
  69. assert.Equal(t, []int{2, 3}, ToIntSlice([]string{"2", "3"}))
  70. assert.Equal(t, []int{2, 3}, ToIntSlice([2]string{"2", "3"}))
  71. }
  72. func TestToBool(t *testing.T) {
  73. assert.Equal(t, ToBool(0), false)
  74. assert.Equal(t, ToBool(nil), false)
  75. assert.Equal(t, ToBool("false"), false)
  76. assert.Equal(t, ToBool("FALSE"), false)
  77. assert.Equal(t, ToBool("False"), false)
  78. assert.Equal(t, ToBool("f"), false)
  79. assert.Equal(t, ToBool("F"), false)
  80. assert.Equal(t, ToBool(false), false)
  81. assert.Equal(t, ToBool("foo"), false)
  82. assert.Equal(t, ToBool("true"), true)
  83. assert.Equal(t, ToBool("TRUE"), true)
  84. assert.Equal(t, ToBool("True"), true)
  85. assert.Equal(t, ToBool("t"), true)
  86. assert.Equal(t, ToBool("T"), true)
  87. assert.Equal(t, ToBool(1), true)
  88. assert.Equal(t, ToBool(true), true)
  89. assert.Equal(t, ToBool(-1), true)
  90. }
  91. func TestIndirectPointers(t *testing.T) {
  92. x := 13
  93. y := &x
  94. z := &y
  95. assert.Equal(t, ToInt(y), 13)
  96. assert.Equal(t, ToInt(z), 13)
  97. }