mirror of
https://github.com/tailscale/tailscale.git
synced 2025-01-10 18:13:41 +00:00
33 lines
696 B
Go
33 lines
696 B
Go
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
|
||
|
|
package lineiter
|
||
|
|
|
||
|
|
import (
|
||
|
|
"slices"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBytesLines(t *testing.T) {
|
||
|
|
var got []string
|
||
|
|
for line := range Bytes([]byte("foo\n\nbar\nbaz")) {
|
||
|
|
got = append(got, string(line))
|
||
|
|
}
|
||
|
|
want := []string{"foo", "", "bar", "baz"}
|
||
|
|
if !slices.Equal(got, want) {
|
||
|
|
t.Errorf("got %q; want %q", got, want)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReader(t *testing.T) {
|
||
|
|
var got []string
|
||
|
|
for line := range Reader(strings.NewReader("foo\n\nbar\nbaz")) {
|
||
|
|
got = append(got, string(line.MustValue()))
|
||
|
|
}
|
||
|
|
want := []string{"foo", "", "bar", "baz"}
|
||
|
|
if !slices.Equal(got, want) {
|
||
|
|
t.Errorf("got %q; want %q", got, want)
|
||
|
|
}
|
||
|
|
}
|