// Copyright (C) 2023  Shanhu Tech Inc.
//
// This program is free software: you can redistribute it and/or modify it
// under the terms of the GNU Affero General Public License as published by the
// Free Software Foundation, either version 3 of the License, or (at your
// option) any later version.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License
// for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package strutil

import (
	"testing"
)

func TestCountLines(t *testing.T) {
	for _, test := range []struct {
		bs   []byte
		want int
	}{
		{nil, 0},
		{make([]byte, 0), 0},
		{[]byte("abcd"), 1},
		{[]byte("\n"), 1},
		{[]byte(" \n"), 1},
		{[]byte("\n\n"), 2},
		{[]byte("\n\nabc"), 3},
	} {
		got := CountLines(test.bs)
		if test.want != got {
			t.Errorf(
				"CountLines(%q): got %d, want %d",
				string(test.bs), got, test.want,
			)
		}
	}
}
