lines_test.go

package strings

import "testing"

func TestTrimShiftLines(t *testing.T) {
	tests := []struct {
		name string
		s    string
		want string
	}{
		{"empty", "", ""},
		{"whitespace", "   ", ""},
		{"empty lines", "\n\n    \n ", "\n\n\n"},
		{"one line no trim", "abc\n", "abc\n"},
		{"one line with trim", "   abc\n", "abc\n"},
		{"one line with end trim", "   abc   \n", "abc\n"},
		{"two lines", "   abc   \n  def", " abc\ndef"},
		{"three lines", "   abc   \n  def\n   ghi", " abc\ndef\n ghi"},
		{"three lines no match", "\t  abc   \n  def\n   ghi", "\t  abc\n  def\n   ghi"},
		{"three lines no match 2", "   abc   \n\t def\n   ghi", "   abc\n\t def\n   ghi"},
		{"three lines no match 3", "  abc   \n\t  def\n   ghi", "  abc\n\t  def\n   ghi"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := TrimShiftLines(tt.s); got != tt.want {
				t.Errorf("TrimShiftLines() = %v, want %v", got, tt.want)
			}
		})
	}
}