convert_test.go

package strutils_test

import (
	"fmt"
	"strings"
	"testing"
	"time"

	strutils "github.com/tekintian/strutils"
)

func TestStr2Int(t *testing.T) {
	cases := []struct {
		str      string
		expected int
	}{
		{str: "12.3", expected: 12},
		{str: "hi123你好", expected: 123},
		{str: "2,56.9", expected: 256},
		{str: "hello2.56aa", expected: 2},
		{str: "hello aa 89", expected: 89},
	}
	for _, v := range cases {
		t.Run(v.str, func(t *testing.T) {
			ret := strutils.Str2Int(v.str)
			if ret != v.expected {
				t.Fatalf("Str2Int %v expected %v got %v", v.str, v.expected, ret)
			}
		})
	}
}
func TestStr2Int64Slice(t *testing.T) {
	cases := []struct {
		str      string
		expected []int64
	}{
		{str: "12.3你好, 456", expected: []int64{12, 456}},
		{str: "hi123你好88", expected: []int64{123}},
		{str: "hi123, 你好88", expected: []int64{123, 88}},
		{str: "hi123,你好88", expected: []int64{123, 88}},
	}
	for _, v := range cases {
		t.Run(v.str, func(t *testing.T) {
			ret := strutils.Str2Int64Slice(v.str)
			// 这里判断2个切片是否相等,我们直接将他转换为字符串后比较; go版本大于1.21的话可以使用 slices包里面的Equal函数比较
			if fmt.Sprintf("%v", ret) != fmt.Sprintf("%v", v.expected) {
				t.Fatalf("Str2Int64Slice %v expected %v got %v", v.str, v.expected, ret)
			}
		})
	}
}
func TestStr2Float64(t *testing.T) {
	cases := []struct {
		str      string
		expected float64
	}{
		{str: "12.3", expected: 12.3},
		{str: "hi123你好", expected: 123},
		{str: "2,56.9", expected: 256.9},
		{str: "hello2.56aa", expected: 2.56},
		{str: "hello aa 89", expected: 89},
	}
	for _, v := range cases {
		t.Run(v.str, func(t *testing.T) {
			ret := strutils.Str2Float64(v.str)
			if ret != v.expected {
				t.Fatalf("Str2Int %v expected %v got %v", v.str, v.expected, ret)
			}
		})
	}
}

func TestGbkToUtf8(t *testing.T) {
	utf8Data, err := strutils.GbkToUtf8(Gb2312Data)
	if err != nil {
		t.Error(err)
	}
	t.Logf("utf8 data: %v", string(utf8Data))
}

func TestUtf8ToGbk(t *testing.T) {
	utf8Data, err := strutils.GbkToUtf8(Gb2312Data)
	if err != nil {
		t.Error(err)
	}
	gbkData, err := strutils.Utf8ToGbk(utf8Data)
	if err != nil {
		t.Error(err)
	}
	if fmt.Sprintf("%v", gbkData) != fmt.Sprintf("%v", Gb2312Data) {
		t.Fatal("Utf8ToGbk编码转换失败, got false, want true")
	} else {
		t.Log("ok")
	}
}

func TestAnyToStr(t *testing.T) {
	now := time.Now()
	testCases := []struct {
		input  interface{}
		output string
	}{
		{input: []byte("你好中国Hello world"), output: "你好中国Hello world"},
		{input: now, output: now.Format(time.RFC3339)},
		{input: [2]int{123, 456}, output: "123 456"},
		{input: []int{123, 456}, output: "123 456"},
		{input: map[string]interface{}{"aaa": 123, "bbb": 456}, output: "aaa:123 bbb:456"},
		{input: 123, output: "123"},
		{input: 88.9, output: "88.9"},
		{input: true, output: "true"},
	}
	for _, v := range testCases {
		str := strutils.AnyToStr(v.input)
		if !strings.HasPrefix(str, v.output) {
			t.Errorf("Testing failure , expected %v, got %v", v.output, str)
		}
	}

}

func TestStrToInt64(t *testing.T) {
	testCases := []struct {
		input  string
		output float64
	}{
		{input: "123", output: 123},
		{input: "1,2,3", output: 123},
		{input: "abc123", output: 123},
		{input: "12.888", output: 12.888},
	}
	for _, v := range testCases {
		ival := strutils.StrToFloat64(v.input)
		if ival != v.output {
			t.Fatalf("Expected output to be %v, got %v", v.output, ival)
		}
	}
}

func TestTimeToStr(t *testing.T) {
	ts, _ := time.Parse(time.DateTime, "2024-08-03 12:21:15") // ts对象
	// 测试用例
	testCases := []struct {
		input  interface{}
		output string
	}{
		{input: "2024年08月03日12:04:49", output: "2024-08-03T12:04:49Z"},
		{input: ts, output: "2024-08-03T12:21:15Z"},
		{input: 1722657237, output: "2024-08-03T11:53:57+08:00"},
		{input: 1722683285997, output: "2024-08-03T19:08:05+08:00"}, // 毫秒时间戳
	}
	for _, v := range testCases {
		ival := strutils.TimeToStr(v.input)
		if ival != v.output {
			t.Fatalf("Expected output to be %v, got %v", v.output, ival)
		}
	}
	// 时间戳, 指定输出字符串格式
	ival := strutils.TimeToStr(1722657237, time.DateTime)
	if ival != "2024-08-03 11:53:57" {
		t.Fatalf("Expected output to be 2024-08-03 11:53:57, got %v", ival)
	}
	// 指定输入tval字符串的格式,和最终输出字符串格式
	ival2 := strutils.TimeToStr("2024-08-03T12:21:15Z", time.DateTime, time.RFC3339)
	if ival2 != "2024-08-03 12:21:15" {
		t.Fatalf("Expected output to be 2024-08-03 12:21:15, got %v", ival2)
	}
}