mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-08-06 10:43:56 +00:00
add ut for utils WriteLine WriteBytesLine
This commit is contained in:
parent
3783821553
commit
761473cd44
@ -17,10 +17,12 @@ limitations under the License.
|
|||||||
package util
|
package util
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
v1 "k8s.io/api/core/v1"
|
v1 "k8s.io/api/core/v1"
|
||||||
@ -929,3 +931,67 @@ func TestGetClusterIPByFamily(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestWriteLine(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
words []string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "write no word",
|
||||||
|
words: []string{},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "write one word",
|
||||||
|
words: []string{"test1"},
|
||||||
|
expected: "test1\n",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "write multi word",
|
||||||
|
words: []string{"test1", "test2", "test3"},
|
||||||
|
expected: "test1 test2 test3\n",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
testBuffer := bytes.NewBuffer(nil)
|
||||||
|
for _, testCase := range testCases {
|
||||||
|
t.Run(testCase.name, func(t *testing.T) {
|
||||||
|
testBuffer.Reset()
|
||||||
|
WriteLine(testBuffer, testCase.words...)
|
||||||
|
if !strings.EqualFold(testBuffer.String(), testCase.expected) {
|
||||||
|
t.Fatalf("write word is %v\n expected: %s, got: %s", testCase.words, testCase.expected, testBuffer.String())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWriteBytesLine(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
bytes []byte
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "empty bytes",
|
||||||
|
bytes: []byte{},
|
||||||
|
expected: "\n",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "test bytes",
|
||||||
|
bytes: []byte("test write bytes line"),
|
||||||
|
expected: "test write bytes line\n",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
testBuffer := bytes.NewBuffer(nil)
|
||||||
|
for _, testCase := range testCases {
|
||||||
|
t.Run(testCase.name, func(t *testing.T) {
|
||||||
|
testBuffer.Reset()
|
||||||
|
WriteBytesLine(testBuffer, testCase.bytes)
|
||||||
|
if !strings.EqualFold(testBuffer.String(), testCase.expected) {
|
||||||
|
t.Fatalf("write word is %v\n expected: %s, got: %s", testCase.bytes, testCase.expected, testBuffer.String())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user