poller_test.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. package filenotify
  2. import (
  3. "fmt"
  4. "io/ioutil"
  5. "os"
  6. "runtime"
  7. "testing"
  8. "time"
  9. "github.com/fsnotify/fsnotify"
  10. )
  11. func TestPollerAddRemove(t *testing.T) {
  12. w := NewPollingWatcher()
  13. if err := w.Add("no-such-file"); err == nil {
  14. t.Fatal("should have gotten error when adding a non-existent file")
  15. }
  16. if err := w.Remove("no-such-file"); err == nil {
  17. t.Fatal("should have gotten error when removing non-existent watch")
  18. }
  19. f, err := ioutil.TempFile("", "asdf")
  20. if err != nil {
  21. t.Fatal(err)
  22. }
  23. defer os.RemoveAll(f.Name())
  24. if err := w.Add(f.Name()); err != nil {
  25. t.Fatal(err)
  26. }
  27. if err := w.Remove(f.Name()); err != nil {
  28. t.Fatal(err)
  29. }
  30. }
  31. func TestPollerEvent(t *testing.T) {
  32. if runtime.GOOS == "windows" {
  33. t.Skip("No chmod on Windows")
  34. }
  35. w := NewPollingWatcher()
  36. f, err := ioutil.TempFile("", "test-poller")
  37. if err != nil {
  38. t.Fatal("error creating temp file")
  39. }
  40. defer os.RemoveAll(f.Name())
  41. f.Close()
  42. if err := w.Add(f.Name()); err != nil {
  43. t.Fatal(err)
  44. }
  45. select {
  46. case <-w.Events():
  47. t.Fatal("got event before anything happened")
  48. case <-w.Errors():
  49. t.Fatal("got error before anything happened")
  50. default:
  51. }
  52. if err := ioutil.WriteFile(f.Name(), []byte("hello"), 0644); err != nil {
  53. t.Fatal(err)
  54. }
  55. if err := assertEvent(w, fsnotify.Write); err != nil {
  56. t.Fatal(err)
  57. }
  58. if err := os.Chmod(f.Name(), 600); err != nil {
  59. t.Fatal(err)
  60. }
  61. if err := assertEvent(w, fsnotify.Chmod); err != nil {
  62. t.Fatal(err)
  63. }
  64. if err := os.Remove(f.Name()); err != nil {
  65. t.Fatal(err)
  66. }
  67. if err := assertEvent(w, fsnotify.Remove); err != nil {
  68. t.Fatal(err)
  69. }
  70. }
  71. func TestPollerClose(t *testing.T) {
  72. w := NewPollingWatcher()
  73. if err := w.Close(); err != nil {
  74. t.Fatal(err)
  75. }
  76. // test double-close
  77. if err := w.Close(); err != nil {
  78. t.Fatal(err)
  79. }
  80. f, err := ioutil.TempFile("", "asdf")
  81. if err != nil {
  82. t.Fatal(err)
  83. }
  84. defer os.RemoveAll(f.Name())
  85. if err := w.Add(f.Name()); err == nil {
  86. t.Fatal("should have gotten error adding watch for closed watcher")
  87. }
  88. }
  89. func assertEvent(w FileWatcher, eType fsnotify.Op) error {
  90. var err error
  91. select {
  92. case e := <-w.Events():
  93. if e.Op != eType {
  94. err = fmt.Errorf("got wrong event type, expected %q: %v", eType, e.Op)
  95. }
  96. case e := <-w.Errors():
  97. err = fmt.Errorf("got unexpected error waiting for events %v: %v", eType, e)
  98. case <-time.After(watchWaitTime * 3):
  99. err = fmt.Errorf("timeout waiting for event %v", eType)
  100. }
  101. return err
  102. }