Fix flakey test for log file rotate.
Signed-off-by: Brian Goff <cpuguy83@gmail.com>
(cherry picked from commit 5ea5c02c887392be1560e559a3c4d53445cf6505)
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
diff --git a/daemon/logger/loggerutils/logfile_test.go b/daemon/logger/loggerutils/logfile_test.go
index d7ff34e..c63d67a 100644
--- a/daemon/logger/loggerutils/logfile_test.go
+++ b/daemon/logger/loggerutils/logfile_test.go
@@ -6,6 +6,7 @@
"io"
"io/ioutil"
"os"
+ "path/filepath"
"strings"
"testing"
"time"
@@ -14,6 +15,7 @@
"github.com/docker/docker/pkg/pubsub"
"github.com/docker/docker/pkg/tailfile"
"gotest.tools/assert"
+ "gotest.tools/poll"
)
func TestTailFiles(t *testing.T) {
@@ -225,21 +227,15 @@
defer l.Close()
assert.NilError(t, l.WriteLogEntry(&logger.Message{Line: []byte("hello world!")}))
-
- dStringer := dirStringer{dir}
-
_, err = os.Stat(f.Name() + ".1")
- assert.Assert(t, os.IsNotExist(err), dStringer)
+ assert.Assert(t, os.IsNotExist(err), dirStringer{dir})
assert.NilError(t, l.WriteLogEntry(&logger.Message{Line: []byte("hello world!")}))
- _, err = os.Stat(f.Name() + ".1")
- assert.NilError(t, err, dStringer)
+ poll.WaitOn(t, checkFileExists(f.Name()+".1"), poll.WithTimeout(30*time.Second))
assert.NilError(t, l.WriteLogEntry(&logger.Message{Line: []byte("hello world!")}))
- _, err = os.Stat(f.Name() + ".1")
- assert.NilError(t, err, dStringer)
- _, err = os.Stat(f.Name() + ".2.gz")
- assert.NilError(t, err, dStringer)
+ poll.WaitOn(t, checkFileExists(f.Name()+".1"), poll.WithTimeout(30*time.Second))
+ poll.WaitOn(t, checkFileExists(f.Name()+".2.gz"), poll.WithTimeout(30*time.Second))
// Now let's simulate a failed rotation where the file was able to be closed but something else happened elsewhere
// down the line.
@@ -265,3 +261,18 @@
}
return s.String()
}
+
+func checkFileExists(name string) poll.Check {
+ return func(t poll.LogT) poll.Result {
+ _, err := os.Stat(name)
+ switch {
+ case err == nil:
+ return poll.Success()
+ case os.IsNotExist(err):
+ return poll.Continue("waiting for %s to exist", name)
+ default:
+ t.Logf("%s", dirStringer{filepath.Dir(name)})
+ return poll.Error(err)
+ }
+ }
+}
