go-kit
/
tail
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

This Change includes: 

- Fixes Flaky test, some tests do not require call to tail.Stop()
- Changes inadequate names
This commit is contained in:
Nino Khodabandeh 2016-04-26 11:30:20 -07:00
parent 04356b27e1
commit 308ca46680
1 changed files with 130 additions and 126 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

254
tail_test.go
View File

@ -50,38 +50,38 @@ func TestMustExist(t *testing.T) {
tail.Cleanup() tail.Cleanup()
} }
func TestWaitsForFileToExist(_t *testing.T) { func TestWaitsForFileToExist(t *testing.T) {
t := NewTailTest("waits-for-file-to-exist", _t) tailTest := NewTailTest("waits-for-file-to-exist", t)
tail := t.StartTail("test.txt", Config{}) tail := tailTest.StartTail("test.txt", Config{})
go t.VerifyTailOutput(tail, []string{"hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"hello", "world"})
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestWaitsForFileToExistRelativePath(_t *testing.T) { func TestWaitsForFileToExistRelativePath(t *testing.T) {
t := NewTailTest("waits-for-file-to-exist-relative", _t) tailTest := NewTailTest("waits-for-file-to-exist-relative", t)
oldWD, err := os.Getwd() oldWD, err := os.Getwd()
if err != nil { if err != nil {
t.Fatal(err) tailTest.Fatal(err)
} }
os.Chdir(t.path) os.Chdir(tailTest.path)
defer os.Chdir(oldWD) defer os.Chdir(oldWD)
tail, err := TailFile("test.txt", Config{}) tail, err := TailFile("test.txt", Config{})
if err != nil { if err != nil {
t.Fatal(err) tailTest.Fatal(err)
} }
go t.VerifyTailOutput(tail, []string{"hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"hello", "world"})
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
if err := ioutil.WriteFile("test.txt", []byte("hello\nworld\n"), 0600); err != nil { if err := ioutil.WriteFile("test.txt", []byte("hello\nworld\n"), 0600); err != nil {
t.Fatal(err) tailTest.Fatal(err)
} }
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestStop(t *testing.T) { func TestStop(t *testing.T) {
@ -95,130 +95,130 @@ func TestStop(t *testing.T) {
tail.Cleanup() tail.Cleanup()
} }
func TestStopAtEOF(_t *testing.T) { func TestStopAtEOF(t *testing.T) {
t := NewTailTest("maxlinesize", _t) tailTest := NewTailTest("maxlinesize", t)
t.CreateFile("test.txt", "hello\nthere\nworld\n") tailTest.CreateFile("test.txt", "hello\nthere\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: nil}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: nil})
// read "hello" // read "hello"
<-tail.Lines <-tail.Lines
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.VerifyTailOutput(tail, []string{"there", "world"}) tailTest.VerifyTailOutput(tail, []string{"there", "world"})
tail.StopAtEOF() tail.StopAtEOF()
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func MaxLineSizeT(_t *testing.T, follow bool, fileContent string, expected []string) { func MaxLineSizeT(t *testing.T, follow bool, fileContent string, expected []string) {
t := NewTailTest("maxlinesize", _t) tailTest := NewTailTest("maxlinesize", t)
t.CreateFile("test.txt", fileContent) tailTest.CreateFile("test.txt", fileContent)
tail := t.StartTail("test.txt", Config{Follow: follow, Location: nil, MaxLineSize: 3}) tail := tailTest.StartTail("test.txt", Config{Follow: follow, Location: nil, MaxLineSize: 3})
go t.VerifyTailOutput(tail, expected) go tailTest.VerifyTailOutput(tail, expected)
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestMaxLineSizeFollow(_t *testing.T) { func TestMaxLineSizeFollow(t *testing.T) {
// As last file line does not end with newline, it will not be present in tail's output // As last file line does not end with newline, it will not be present in tail's output
MaxLineSizeT(_t, true, "hello\nworld\nfin\nhe", []string{"hel", "lo", "wor", "ld", "fin"}) MaxLineSizeT(t, true, "hello\nworld\nfin\nhe", []string{"hel", "lo", "wor", "ld", "fin"})
} }
func TestMaxLineSizeNoFollow(_t *testing.T) { func TestMaxLineSizeNoFollow(t *testing.T) {
MaxLineSizeT(_t, false, "hello\nworld\nfin\nhe", []string{"hel", "lo", "wor", "ld", "fin", "he"}) MaxLineSizeT(t, false, "hello\nworld\nfin\nhe", []string{"hel", "lo", "wor", "ld", "fin", "he"})
} }
func TestOver4096ByteLine(_t *testing.T) { func TestOver4096ByteLine(t *testing.T) {
t := NewTailTest("Over4096ByteLine", _t) tailTest := NewTailTest("Over4096ByteLine", t)
testString := strings.Repeat("a", 4097) testString := strings.Repeat("a", 4097)
t.CreateFile("test.txt", "test\n"+testString+"\nhello\nworld\n") tailTest.CreateFile("test.txt", "test\n"+testString+"\nhello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: nil}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: nil})
go t.VerifyTailOutput(tail, []string{"test", testString, "hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"test", testString, "hello", "world"})
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestOver4096ByteLineWithSetMaxLineSize(_t *testing.T) { func TestOver4096ByteLineWithSetMaxLineSize(t *testing.T) {
t := NewTailTest("Over4096ByteLineMaxLineSize", _t) tailTest := NewTailTest("Over4096ByteLineMaxLineSize", t)
testString := strings.Repeat("a", 4097) testString := strings.Repeat("a", 4097)
t.CreateFile("test.txt", "test\n"+testString+"\nhello\nworld\n") tailTest.CreateFile("test.txt", "test\n"+testString+"\nhello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: nil, MaxLineSize: 4097}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: nil, MaxLineSize: 4097})
go t.VerifyTailOutput(tail, []string{"test", testString, "hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"test", testString, "hello", "world"})
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestLocationFull(_t *testing.T) { func TestLocationFull(t *testing.T) {
t := NewTailTest("location-full", _t) tailTest := NewTailTest("location-full", t)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: nil}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: nil})
go t.VerifyTailOutput(tail, []string{"hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"hello", "world"})
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestLocationFullDontFollow(_t *testing.T) { func TestLocationFullDontFollow(t *testing.T) {
t := NewTailTest("location-full-dontfollow", _t) tailTest := NewTailTest("location-full-dontfollow", t)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: false, Location: nil}) tail := tailTest.StartTail("test.txt", Config{Follow: false, Location: nil})
go t.VerifyTailOutput(tail, []string{"hello", "world"}) go tailTest.VerifyTailOutput(tail, []string{"hello", "world"})
// Add more data only after reasonable delay. // Add more data only after reasonable delay.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.AppendFile("test.txt", "more\ndata\n") tailTest.AppendFile("test.txt", "more\ndata\n")
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestLocationEnd(_t *testing.T) { func TestLocationEnd(t *testing.T) {
t := NewTailTest("location-end", _t) tailTest := NewTailTest("location-end", t)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: &SeekInfo{0, os.SEEK_END}}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: &SeekInfo{0, os.SEEK_END}})
go t.VerifyTailOutput(tail, []string{"more", "data"}) go tailTest.VerifyTailOutput(tail, []string{"more", "data"})
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.AppendFile("test.txt", "more\ndata\n") tailTest.AppendFile("test.txt", "more\ndata\n")
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestLocationMiddle(_t *testing.T) { func TestLocationMiddle(t *testing.T) {
// Test reading from middle. // Test reading from middle.
t := NewTailTest("location-middle", _t) tailTest := NewTailTest("location-middle", t)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
tail := t.StartTail("test.txt", Config{Follow: true, Location: &SeekInfo{-6, os.SEEK_END}}) tail := tailTest.StartTail("test.txt", Config{Follow: true, Location: &SeekInfo{-6, os.SEEK_END}})
go t.VerifyTailOutput(tail, []string{"world", "more", "data"}) go tailTest.VerifyTailOutput(tail, []string{"world", "more", "data"})
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.AppendFile("test.txt", "more\ndata\n") tailTest.AppendFile("test.txt", "more\ndata\n")
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func _TestReOpen(_t *testing.T, poll bool) { func reOpen(t *testing.T, poll bool) {
var name string var name string
var delay time.Duration var delay time.Duration
if poll { if poll {
@ -228,102 +228,102 @@ func _TestReOpen(_t *testing.T, poll bool) {
name = "reopen-inotify" name = "reopen-inotify"
delay = 100 * time.Millisecond delay = 100 * time.Millisecond
} }
t := NewTailTest(name, _t) tailTest := NewTailTest(name, t)
t.CreateFile("test.txt", "hello\nworld\n") tailTest.CreateFile("test.txt", "hello\nworld\n")
tail := t.StartTail( tail := tailTest.StartTail(
"test.txt", "test.txt",
Config{Follow: true, ReOpen: true, Poll: poll}) Config{Follow: true, ReOpen: true, Poll: poll})
go t.VerifyTailOutput(tail, []string{"hello", "world", "more", "data", "endofworld"}) go tailTest.VerifyTailOutput(tail, []string{"hello", "world", "more", "data", "endofworld"})
// deletion must trigger reopen // deletion must trigger reopen
<-time.After(delay) <-time.After(delay)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
<-time.After(delay) <-time.After(delay)
t.CreateFile("test.txt", "more\ndata\n") tailTest.CreateFile("test.txt", "more\ndata\n")
// rename must trigger reopen // rename must trigger reopen
<-time.After(delay) <-time.After(delay)
t.RenameFile("test.txt", "test.txt.rotated") tailTest.RenameFile("test.txt", "test.txt.rotated")
<-time.After(delay) <-time.After(delay)
t.CreateFile("test.txt", "endofworld\n") tailTest.CreateFile("test.txt", "endofworld\n")
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(delay) <-time.After(delay)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
<-time.After(delay) <-time.After(delay)
// Do not bother with stopping as it could kill the tomb during // Do not bother with stopping as it could kill the tomb during
// the reading of data written above. Timings can vary based on // the reading of data written above. Timings can vary based on
// test environment. // test environment.
t.Cleanup(tail) tailTest.Cleanup(tail, false)
} }
// The use of polling file watcher could affect file rotation // The use of polling file watcher could affect file rotation
// (detected via renames), so test these explicitly. // (detected via renames), so test these explicitly.
func TestReOpenInotify(_t *testing.T) { func TestReOpenInotify(t *testing.T) {
_TestReOpen(_t, false) reOpen(t, false)
} }
func TestReOpenPolling(_t *testing.T) { func TestReOpenPolling(t *testing.T) {
_TestReOpen(_t, true) reOpen(t, true)
} }
func _TestReSeek(_t *testing.T, poll bool) { func reSeek(t *testing.T, poll bool) {
var name string var name string
if poll { if poll {
name = "reseek-polling" name = "reseek-polling"
} else { } else {
name = "reseek-inotify" name = "reseek-inotify"
} }
t := NewTailTest(name, _t) tailTest := NewTailTest(name, t)
t.CreateFile("test.txt", "a really long string goes here\nhello\nworld\n") tailTest.CreateFile("test.txt", "a really long string goes here\nhello\nworld\n")
tail := t.StartTail( tail := tailTest.StartTail(
"test.txt", "test.txt",
Config{Follow: true, ReOpen: false, Poll: poll}) Config{Follow: true, ReOpen: false, Poll: poll})
go t.VerifyTailOutput(tail, []string{ go tailTest.VerifyTailOutput(tail, []string{
"a really long string goes here", "hello", "world", "h311o", "w0r1d", "endofworld"}) "a really long string goes here", "hello", "world", "h311o", "w0r1d", "endofworld"})
// truncate now // truncate now
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.TruncateFile("test.txt", "h311o\nw0r1d\nendofworld\n") tailTest.TruncateFile("test.txt", "h311o\nw0r1d\nendofworld\n")
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
// Do not bother with stopping as it could kill the tomb during // Do not bother with stopping as it could kill the tomb during
// the reading of data written above. Timings can vary based on // the reading of data written above. Timings can vary based on
// test environment. // test environment.
t.Cleanup(tail) tailTest.Cleanup(tail, false)
} }
// The use of polling file watcher could affect file rotation // The use of polling file watcher could affect file rotation
// (detected via renames), so test these explicitly. // (detected via renames), so test these explicitly.
func TestReSeekInotify(_t *testing.T) { func TestReSeekInotify(t *testing.T) {
_TestReSeek(_t, false) reSeek(t, false)
} }
func TestReSeekPolling(_t *testing.T) { func TestReSeekPolling(t *testing.T) {
_TestReSeek(_t, true) reSeek(t, true)
} }
func TestRateLimiting(_t *testing.T) { func TestRateLimiting(t *testing.T) {
t := NewTailTest("rate-limiting", _t) tailTest := NewTailTest("rate-limiting", t)
t.CreateFile("test.txt", "hello\nworld\nagain\nextra\n") tailTest.CreateFile("test.txt", "hello\nworld\nagain\nextra\n")
config := Config{ config := Config{
Follow: true, Follow: true,
RateLimiter: ratelimiter.NewLeakyBucket(2, time.Second)} RateLimiter: ratelimiter.NewLeakyBucket(2, time.Second)}
leakybucketFull := "Too much log activity; waiting a second before resuming tailing" leakybucketFull := "Too much log activity; waiting a second before resuming tailing"
tail := t.StartTail("test.txt", config) tail := tailTest.StartTail("test.txt", config)
// TODO: also verify that tail resumes after the cooloff period. // TODO: also verify that tail resumes after the cooloff period.
go t.VerifyTailOutput(tail, []string{ go tailTest.VerifyTailOutput(tail, []string{
"hello", "world", "again", "hello", "world", "again",
leakybucketFull, leakybucketFull,
"more", "data", "more", "data",
@ -331,28 +331,28 @@ func TestRateLimiting(_t *testing.T) {
// Add more data only after reasonable delay. // Add more data only after reasonable delay.
<-time.After(1200 * time.Millisecond) <-time.After(1200 * time.Millisecond)
t.AppendFile("test.txt", "more\ndata\n") tailTest.AppendFile("test.txt", "more\ndata\n")
// Delete after a reasonable delay, to give tail sufficient time // Delete after a reasonable delay, to give tail sufficient time
// to read all lines. // to read all lines.
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
t.Cleanup(tail) tailTest.Cleanup(tail, true)
} }
func TestTell(_t *testing.T) { func TestTell(t *testing.T) {
t := NewTailTest("tell-position", _t) tailTest := NewTailTest("tell-position", t)
t.CreateFile("test.txt", "hello\nworld\nagain\nmore\n") tailTest.CreateFile("test.txt", "hello\nworld\nagain\nmore\n")
config := Config{ config := Config{
Follow: false, Follow: false,
Location: &SeekInfo{0, os.SEEK_SET}} Location: &SeekInfo{0, os.SEEK_SET}}
tail := t.StartTail("test.txt", config) tail := tailTest.StartTail("test.txt", config)
// read noe line // read noe line
<-tail.Lines <-tail.Lines
offset, err := tail.Tell() offset, err := tail.Tell()
if err != nil { if err != nil {
t.Errorf("Tell return error: %s", err.Error()) tailTest.Errorf("Tell return error: %s", err.Error())
} }
tail.Done() tail.Done()
// tail.close() // tail.close()
@ -360,39 +360,39 @@ func TestTell(_t *testing.T) {
config = Config{ config = Config{
Follow: false, Follow: false,
Location: &SeekInfo{offset, os.SEEK_SET}} Location: &SeekInfo{offset, os.SEEK_SET}}
tail = t.StartTail("test.txt", config) tail = tailTest.StartTail("test.txt", config)
for l := range tail.Lines { for l := range tail.Lines {
// it may readed one line in the chan(tail.Lines), // it may readed one line in the chan(tail.Lines),
// so it may lost one line. // so it may lost one line.
if l.Text != "world" && l.Text != "again" { if l.Text != "world" && l.Text != "again" {
t.Fatalf("mismatch; expected world or again, but got %s", tailTest.Fatalf("mismatch; expected world or again, but got %s",
l.Text) l.Text)
} }
break break
} }
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
tail.Done() tail.Done()
tail.Cleanup() tail.Cleanup()
} }
func TestBlockUntilExists(_t *testing.T) { func TestBlockUntilExists(t *testing.T) {
t := NewTailTest("block-until-file-exists", _t) tailTest := NewTailTest("block-until-file-exists", t)
config := Config{ config := Config{
Follow: true, Follow: true,
} }
tail := t.StartTail("test.txt", config) tail := tailTest.StartTail("test.txt", config)
go func() { go func() {
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
t.CreateFile("test.txt", "hello world\n") tailTest.CreateFile("test.txt", "hello world\n")
}() }()
for l := range tail.Lines { for l := range tail.Lines {
if l.Text != "hello world" { if l.Text != "hello world" {
t.Fatalf("mismatch; expected hello world, but got %s", tailTest.Fatalf("mismatch; expected hello world, but got %s",
l.Text) l.Text)
} }
break break
} }
t.RemoveFile("test.txt") tailTest.RemoveFile("test.txt")
tail.Stop() tail.Stop()
tail.Cleanup() tail.Cleanup()
} }
@ -495,6 +495,8 @@ func (t TailTest) VerifyTailOutput(tail *Tail, lines []string) {
line, tailedLine.Text) line, tailedLine.Text)
} }
} }
// It is important to return if follow is set
// otherwise we could block on <-tail.Lines
if tail.Follow { if tail.Follow {
return return
} }
@ -504,9 +506,11 @@ func (t TailTest) VerifyTailOutput(tail *Tail, lines []string) {
} }
} }
func (t TailTest) Cleanup(tail *Tail) { func (t TailTest) Cleanup(tail *Tail, stop bool) {
<-time.After(100 * time.Millisecond) <-time.After(100 * time.Millisecond)
if stop {
tail.Stop() tail.Stop()
}
tail.Cleanup() tail.Cleanup()
<-t.done <-t.done
} }