runtimes/google/lib/follow/notify_reader_test.go

// +build darwin freebsd linux netbsd openbsd windows

package follow

import (
	"io/ioutil"
	"os"
	"testing"
	"time"
)

// TestNotifyReadPartial tests partial reads with the fsnotify-based fsReader
func TestNotifyReadPartial(t *testing.T) {
	// Create the test file.
	testFile, err := ioutil.TempFile(os.TempDir(), "follow.reader.notify.partial")
	if err != nil {
		t.Fatalf("ioutil.TempFile() failed: %v", err)
	}
	testFile.Close()
	testFileName := testFile.Name()
	defer os.Remove(testFileName)

	// Create the fsnotify-based fsWatcher.
	watcher, err := newFSNotifyWatcher(testFileName)
	if err != nil {
		t.Fatalf("newCustomFSWatcher() failed: %v", err)
	}

	timeout := time.Second
	if err := testReadPartial(testFileName, watcher, timeout); err != nil {
		t.Fatal("testReadPartial() failed: %v", err)
	}
}

// TestNotifyReadFull tests full reads with the fsnotify-based fsReader
func TestNotifyReadFull(t *testing.T) {
	// Create the test file.
	testFile, err := ioutil.TempFile(os.TempDir(), "follow.reader.notify.full")
	if err != nil {
		t.Fatalf("ioutil.TempFile() failed: %v", err)
	}
	testFile.Close()
	testFileName := testFile.Name()
	defer os.Remove(testFileName)

	// Create the fsnotify-based fsWatcher.
	watcher, err := newFSNotifyWatcher(testFileName)
	if err != nil {
		t.Fatalf("newCustomFSWatcher() failed: %v", err)
	}

	timeout := time.Second
	if err := testReadFull(testFileName, watcher, timeout); err != nil {
		t.Fatal("testReadFull() failed: %v", err)
	}
}

// TestNotifyClose tests close with the fsnotify-based fsReader
func TestNotifyClose(t *testing.T) {
	// Create the test file.
	testFile, err := ioutil.TempFile(os.TempDir(), "follow.reader.notify.close")
	if err != nil {
		t.Fatalf("ioutil.TempFile() failed: %v", err)
	}
	testFile.Close()
	testFileName := testFile.Name()
	defer os.Remove(testFileName)

	// Create the fsnotify-based fsWatcher.
	watcher, err := newFSNotifyWatcher(testFileName)
	if err != nil {
		t.Fatalf("newCustomFSWatcher() failed: %v", err)
	}

	timeout := time.Second
	if err := testClose(testFileName, watcher, timeout); err != nil {
		t.Fatal("testClose() failed: %v", err)
	}
}