15 func TestReadAllOrWarnFail(t *testing.T) {
19 // The special file /proc/self/mem can be opened for
20 // reading, but reading from byte 0 returns an error.
21 f, err := os.Open("/proc/self/mem")
23 t.Fatalf("Opening /proc/self/mem: %s", err)
25 if x, err := ReadAllOrWarn(f); err == nil {
26 t.Fatalf("Expected error, got %v", x)
29 if msg, err := rcv.ReadBytes('\n'); err != nil {
31 } else if matched, err := regexp.MatchString("^crunchstat: .*error.*", string(msg)); err != nil || !matched {
32 t.Fatalf("Expected error message about unreadable file, got \"%s\"", msg)
36 func TestReadAllOrWarnSuccess(t *testing.T) {
37 f, err := os.Open("./crunchstat_test.go")
39 t.Fatalf("Opening ./crunchstat_test.go: %s", err)
41 data, err := ReadAllOrWarn(f)
43 t.Fatalf("got error %s", err)
45 if matched, err := regexp.MatchString("^package main\n", string(data)); err != nil || !matched {
46 t.Fatalf("data failed regexp: %s", err)
50 // Test that CopyPipeToChildLog works even on lines longer than
51 // bufio.MaxScanTokenSize.
52 func TestCopyPipeToChildLogLongLines(t *testing.T) {
56 control := make(chan bool)
57 pipeIn, pipeOut := io.Pipe()
58 go CopyPipeToChildLog(pipeIn, control)
60 sentBytes := make([]byte, bufio.MaxScanTokenSize + (1 << 22))
62 pipeOut.Write([]byte("before\n"))
64 for i := range sentBytes {
65 // Some bytes that aren't newlines:
66 sentBytes[i] = byte((rand.Int() & 0xff) | 0x80)
68 sentBytes[len(sentBytes)-1] = '\n'
69 pipeOut.Write(sentBytes)
71 pipeOut.Write([]byte("after"))
75 if before, err := rcv.ReadBytes('\n'); err != nil || string(before) != "before\n" {
76 t.Fatalf("\"before\n\" not received (got \"%s\", %s)", before, err)
79 receivedString, err := rcv.ReadBytes('\n')
83 receivedBytes := []byte(receivedString)
84 if bytes.Compare(receivedBytes, sentBytes) != 0 {
85 t.Fatalf("sent %d bytes, got %d different bytes", len(sentBytes)+1, len(receivedBytes))
88 if after, err := rcv.ReadBytes('\n'); err != nil || string(after) != "after\n" {
89 t.Fatal("\"after\n\" not received (got \"%s\", %s)", after, err)
93 case <-time.After(time.Second):
100 func captureLogs() (*bufio.Reader) {
101 // Send childLog to our bufio reader instead of stderr
102 stderrIn, stderrOut := io.Pipe()
103 childLog = log.New(stderrOut, "", 0)
104 statLog = log.New(stderrOut, "crunchstat: ", 0)
105 return bufio.NewReader(stderrIn)
108 func uncaptureLogs() {
109 childLog = log.New(os.Stderr, "", 0)
110 statLog = log.New(os.Stderr, "crunchstat: ", 0)