fifo_test.go

Documentation: os

		 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  //go:build darwin || dragonfly || freebsd || linux || netbsd || openbsd
		 6  // +build darwin dragonfly freebsd linux netbsd openbsd
		 7  
		 8  package os_test
		 9  
		10  import (
		11  	"bufio"
		12  	"bytes"
		13  	"fmt"
		14  	"io"
		15  	"os"
		16  	"path/filepath"
		17  	"runtime"
		18  	"sync"
		19  	"syscall"
		20  	"testing"
		21  	"time"
		22  )
		23  
		24  // Issue 24164.
		25  func TestFifoEOF(t *testing.T) {
		26  	switch runtime.GOOS {
		27  	case "android":
		28  		t.Skip("skipping on Android; mkfifo syscall not available")
		29  	}
		30  
		31  	dir := t.TempDir()
		32  	fifoName := filepath.Join(dir, "fifo")
		33  	if err := syscall.Mkfifo(fifoName, 0600); err != nil {
		34  		t.Fatal(err)
		35  	}
		36  
		37  	var wg sync.WaitGroup
		38  	wg.Add(1)
		39  	go func() {
		40  		defer wg.Done()
		41  
		42  		w, err := os.OpenFile(fifoName, os.O_WRONLY, 0)
		43  		if err != nil {
		44  			t.Error(err)
		45  			return
		46  		}
		47  
		48  		defer func() {
		49  			if err := w.Close(); err != nil {
		50  				t.Errorf("error closing writer: %v", err)
		51  			}
		52  		}()
		53  
		54  		for i := 0; i < 3; i++ {
		55  			time.Sleep(10 * time.Millisecond)
		56  			_, err := fmt.Fprintf(w, "line %d\n", i)
		57  			if err != nil {
		58  				t.Errorf("error writing to fifo: %v", err)
		59  				return
		60  			}
		61  		}
		62  		time.Sleep(10 * time.Millisecond)
		63  	}()
		64  
		65  	defer wg.Wait()
		66  
		67  	r, err := os.Open(fifoName)
		68  	if err != nil {
		69  		t.Fatal(err)
		70  	}
		71  
		72  	done := make(chan bool)
		73  	go func() {
		74  		defer close(done)
		75  
		76  		defer func() {
		77  			if err := r.Close(); err != nil {
		78  				t.Errorf("error closing reader: %v", err)
		79  			}
		80  		}()
		81  
		82  		rbuf := bufio.NewReader(r)
		83  		for {
		84  			b, err := rbuf.ReadBytes('\n')
		85  			if err == io.EOF {
		86  				break
		87  			}
		88  			if err != nil {
		89  				t.Error(err)
		90  				return
		91  			}
		92  			t.Logf("%s\n", bytes.TrimSpace(b))
		93  		}
		94  	}()
		95  
		96  	select {
		97  	case <-done:
		98  		// Test succeeded.
		99  	case <-time.After(time.Second):
	 100  		t.Error("timed out waiting for read")
	 101  		// Close the reader to force the read to complete.
	 102  		r.Close()
	 103  	}
	 104  }
	 105
View as plain text