libgo/go/os/exec/internal_test.go

   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 }