Fix a data race in TestFileWatcher (#24).
Use a channel to report events to the test, rather than a variable without synchronization.
This commit is contained in:
parent
27a0ec41cc
commit
e374b0a927
1 changed files with 24 additions and 2 deletions
|
@ -18,28 +18,50 @@ package watch
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func prepareTimeout() chan bool {
|
||||||
|
timeoutChan := make(chan bool, 1)
|
||||||
|
go func() {
|
||||||
|
time.Sleep(1 * time.Second)
|
||||||
|
timeoutChan <- true
|
||||||
|
}()
|
||||||
|
return timeoutChan
|
||||||
|
}
|
||||||
|
|
||||||
func TestFileWatcher(t *testing.T) {
|
func TestFileWatcher(t *testing.T) {
|
||||||
file, err := ioutil.TempFile("", "fw")
|
file, err := ioutil.TempFile("", "fw")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
defer file.Close()
|
defer file.Close()
|
||||||
|
defer os.Remove(file.Name())
|
||||||
count := 0
|
count := 0
|
||||||
|
events := make(chan bool, 10)
|
||||||
fw, err := NewFileWatcher(file.Name(), func() {
|
fw, err := NewFileWatcher(file.Name(), func() {
|
||||||
count++
|
count++
|
||||||
if count != 1 {
|
if count != 1 {
|
||||||
t.Fatalf("expected 1 but returned %v", count)
|
t.Fatalf("expected 1 but returned %v", count)
|
||||||
}
|
}
|
||||||
|
events <- true
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
defer fw.Close()
|
defer fw.Close()
|
||||||
if count != 0 {
|
timeoutChan := prepareTimeout()
|
||||||
t.Fatalf("expected 0 but returned %v", count)
|
select {
|
||||||
|
case <-events:
|
||||||
|
t.Fatalf("expected no events before writing a file")
|
||||||
|
case <-timeoutChan:
|
||||||
}
|
}
|
||||||
ioutil.WriteFile(file.Name(), []byte{}, 0644)
|
ioutil.WriteFile(file.Name(), []byte{}, 0644)
|
||||||
|
select {
|
||||||
|
case <-events:
|
||||||
|
case <-timeoutChan:
|
||||||
|
t.Fatalf("expected an event shortly after writing a file")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue