From aa3b7c341323aa5e6fbab4a324263f36f04ec79b Mon Sep 17 00:00:00 2001
From: Sridhar Ratnakumar <sridharr@activestate.com>
Date: Sat, 13 Oct 2012 12:02:38 -0700
Subject: [PATCH] test: add test for "-n 0"

---
 tail_test.go | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/tail_test.go b/tail_test.go
index a23307b..98d92b3 100644
--- a/tail_test.go
+++ b/tail_test.go
@@ -34,8 +34,8 @@ func TestMissingFile(t *testing.T) {
 	tail.Stop()
 }
 
-func TestLocationMinusOne(t *testing.T) {
-	fix := NewFixture("simple", t)
+func TestLocationFull(t *testing.T) {
+	fix := NewFixture("location-full", t)
 	fix.CreateFile("test.txt", "hello\nworld\n")
 	tail := fix.StartTail("test.txt", Config{Follow: true, Location: -1})
 	go fix.VerifyTail(tail, []string{"hello", "world"})
@@ -47,6 +47,22 @@ func TestLocationMinusOne(t *testing.T) {
 	tail.Stop()
 }
 
+func TestLocationEnd(t *testing.T) {
+	fix := NewFixture("location-end", t)
+	fix.CreateFile("test.txt", "hello\nworld\n")
+	tail := fix.StartTail("test.txt", Config{Follow: true, Location: 0})
+	go fix.VerifyTail(tail, []string{"more", "data"})
+
+	<-time.After(100 * time.Millisecond)
+	fix.AppendFile("test.txt", "more\ndata\n")
+
+	// Delete after a reasonable delay, to give tail sufficient time
+	// to read all lines.
+	<-time.After(250 * time.Millisecond)
+	fix.RemoveFile("test.txt")
+	tail.Stop()
+}
+
 type Fixture struct {
 	Name string
 	path string
@@ -63,7 +79,7 @@ func NewFixture(name string, t *testing.T) Fixture {
 }
 
 func (fix Fixture) CreateFile(name string, contents string) {
-	err := ioutil.WriteFile(fix.path+"/"+name, []byte(contents), 0777)
+	err := ioutil.WriteFile(fix.path+"/"+name, []byte(contents), 0600)
 	if err != nil {
 		fix.t.Fatal(err)
 	}
@@ -76,6 +92,18 @@ func (fix Fixture) RemoveFile(name string) {
 	}
 }
 
+func (fix Fixture) AppendFile(name string, contents string) {
+	f, err := os.OpenFile(fix.path+"/"+name, os.O_APPEND|os.O_WRONLY, 0600)
+	if err != nil {
+		fix.t.Fatal(err)
+	}
+	defer f.Close()
+	_, err = f.WriteString(contents)
+	if err != nil {
+		fix.t.Fatal(err)
+	}
+}
+
 func (fix Fixture) StartTail(name string, config Config) *Tail {
 	tail, err := TailFile(fix.path+"/"+name, config)
 	if err != nil {