irc/helpers_test.go at master · jakebailey/irc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package irc

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestStringFieldsHelper(t *testing.T) {
	tests := []struct {
		s        string
		sep      byte
		expected []string
	}{
		{
			s:   "",
			sep: ' ',
		},
		{
			s:        "A",
			sep:      ' ',
			expected: []string{"A"},
		},
		{
			s:        "A B C",
			sep:      ' ',
			expected: []string{"A", "B", "C"},
		},
		{
			s:        "A B C",
			sep:      '-',
			expected: []string{"A B C"},
		},
		{
			s:        "A-B-C",
			sep:      '-',
			expected: []string{"A", "B", "C"},
		},
		{
			s:        "A  B   C    ",
			sep:      ' ',
			expected: []string{"A", "B", "C"},
		},
		{
			s:   "       ",
			sep: ' ',
		},
		{
			s:        "A ",
			sep:      ' ',
			expected: []string{"A"},
		},
	}

	for _, test := range tests {
		builtin := strings.FieldsFunc(test.s, func(r rune) bool {
			return r == rune(test.sep)
		})
		if len(builtin) == 0 {
			builtin = nil
		}
		assert.Equal(t, test.expected, builtin, "expected should match string.Fields")

		got := stringFields(test.s, test.sep)
		assert.Equal(t, test.expected, got, "expected result with sep `%s`", string(test.sep))
	}
}

func TestFastTrimHelper(t *testing.T) {
	tests := []string{
		"",
		"A",
		" A",
		"  A",
		" A ",
		"  A ",
		"  A  ",
		" A  ",
		"A ",
		"A   ",
		"foo bar",
		"   foo bar ",
		"ab ",
		" ab ",
		"ab",
		" ab",
		"a-b-c ",
	}

	for _, test := range tests {
		expected := strings.Trim(test, " ")
		actual := fastTrim(test, ' ')

		assert.Equal(t, expected, actual, "test=`%s`", test)
	}
}