internal_test.go

Documentation: os/exec

		 1  // Copyright 2015 The Go Authors. All rights reserved.
		 2  // Use of this source code is governed by a BSD-style
		 3  // license that can be found in the LICENSE file.
		 4  
		 5  package exec
		 6  
		 7  import (
		 8  	"io"
		 9  	"testing"
		10  )
		11  
		12  func TestPrefixSuffixSaver(t *testing.T) {
		13  	tests := []struct {
		14  		N			int
		15  		writes []string
		16  		want	 string
		17  	}{
		18  		{
		19  			N:			2,
		20  			writes: nil,
		21  			want:	 "",
		22  		},
		23  		{
		24  			N:			2,
		25  			writes: []string{"a"},
		26  			want:	 "a",
		27  		},
		28  		{
		29  			N:			2,
		30  			writes: []string{"abc", "d"},
		31  			want:	 "abcd",
		32  		},
		33  		{
		34  			N:			2,
		35  			writes: []string{"abc", "d", "e"},
		36  			want:	 "ab\n... omitting 1 bytes ...\nde",
		37  		},
		38  		{
		39  			N:			2,
		40  			writes: []string{"ab______________________yz"},
		41  			want:	 "ab\n... omitting 22 bytes ...\nyz",
		42  		},
		43  		{
		44  			N:			2,
		45  			writes: []string{"ab_______________________y", "z"},
		46  			want:	 "ab\n... omitting 23 bytes ...\nyz",
		47  		},
		48  	}
		49  	for i, tt := range tests {
		50  		w := &prefixSuffixSaver{N: tt.N}
		51  		for _, s := range tt.writes {
		52  			n, err := io.WriteString(w, s)
		53  			if err != nil || n != len(s) {
		54  				t.Errorf("%d. WriteString(%q) = %v, %v; want %v, %v", i, s, n, err, len(s), nil)
		55  			}
		56  		}
		57  		if got := string(w.Bytes()); got != tt.want {
		58  			t.Errorf("%d. Bytes = %q; want %q", i, got, tt.want)
		59  		}
		60  	}
		61  }
		62
View as plain text