|
|
@@ -27,6 +27,8 @@ import (
|
|
|
"github.com/coreos/etcd/pkg/pbutil"
|
|
|
"github.com/coreos/etcd/raft/raftpb"
|
|
|
"github.com/coreos/etcd/wal/walpb"
|
|
|
+
|
|
|
+ "go.uber.org/zap"
|
|
|
)
|
|
|
|
|
|
func TestNew(t *testing.T) {
|
|
|
@@ -36,7 +38,7 @@ func TestNew(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- w, err := Create(p, []byte("somedata"))
|
|
|
+ w, err := Create(zap.NewExample(), p, []byte("somedata"))
|
|
|
if err != nil {
|
|
|
t.Fatalf("err = %v, want nil", err)
|
|
|
}
|
|
|
@@ -91,7 +93,7 @@ func TestNewForInitedDir(t *testing.T) {
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
os.Create(filepath.Join(p, walName(0, 0)))
|
|
|
- if _, err = Create(p, nil); err == nil || err != os.ErrExist {
|
|
|
+ if _, err = Create(zap.NewExample(), p, nil); err == nil || err != os.ErrExist {
|
|
|
t.Errorf("err = %v, want %v", err, os.ErrExist)
|
|
|
}
|
|
|
}
|
|
|
@@ -109,7 +111,7 @@ func TestOpenAtIndex(t *testing.T) {
|
|
|
}
|
|
|
f.Close()
|
|
|
|
|
|
- w, err := Open(dir, walpb.Snapshot{})
|
|
|
+ w, err := Open(zap.NewExample(), dir, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatalf("err = %v, want nil", err)
|
|
|
}
|
|
|
@@ -128,7 +130,7 @@ func TestOpenAtIndex(t *testing.T) {
|
|
|
}
|
|
|
f.Close()
|
|
|
|
|
|
- w, err = Open(dir, walpb.Snapshot{Index: 5})
|
|
|
+ w, err = Open(zap.NewExample(), dir, walpb.Snapshot{Index: 5})
|
|
|
if err != nil {
|
|
|
t.Fatalf("err = %v, want nil", err)
|
|
|
}
|
|
|
@@ -145,7 +147,7 @@ func TestOpenAtIndex(t *testing.T) {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
defer os.RemoveAll(emptydir)
|
|
|
- if _, err = Open(emptydir, walpb.Snapshot{}); err != ErrFileNotFound {
|
|
|
+ if _, err = Open(zap.NewExample(), emptydir, walpb.Snapshot{}); err != ErrFileNotFound {
|
|
|
t.Errorf("err = %v, want %v", err, ErrFileNotFound)
|
|
|
}
|
|
|
}
|
|
|
@@ -158,7 +160,7 @@ func TestCut(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- w, err := Create(p, nil)
|
|
|
+ w, err := Create(zap.NewExample(), p, nil)
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -220,7 +222,7 @@ func TestSaveWithCut(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- w, err := Create(p, []byte("metadata"))
|
|
|
+ w, err := Create(zap.NewExample(), p, []byte("metadata"))
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -248,7 +250,7 @@ func TestSaveWithCut(t *testing.T) {
|
|
|
|
|
|
w.Close()
|
|
|
|
|
|
- neww, err := Open(p, walpb.Snapshot{})
|
|
|
+ neww, err := Open(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatalf("err = %v, want nil", err)
|
|
|
}
|
|
|
@@ -283,7 +285,7 @@ func TestRecover(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- w, err := Create(p, []byte("metadata"))
|
|
|
+ w, err := Create(zap.NewExample(), p, []byte("metadata"))
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -302,7 +304,7 @@ func TestRecover(t *testing.T) {
|
|
|
}
|
|
|
w.Close()
|
|
|
|
|
|
- if w, err = Open(p, walpb.Snapshot{}); err != nil {
|
|
|
+ if w, err = Open(zap.NewExample(), p, walpb.Snapshot{}); err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
metadata, state, entries, err := w.ReadAll()
|
|
|
@@ -398,7 +400,7 @@ func TestRecoverAfterCut(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- md, err := Create(p, []byte("metadata"))
|
|
|
+ md, err := Create(zap.NewExample(), p, []byte("metadata"))
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -421,7 +423,7 @@ func TestRecoverAfterCut(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
- w, err := Open(p, walpb.Snapshot{Index: uint64(i)})
|
|
|
+ w, err := Open(zap.NewExample(), p, walpb.Snapshot{Index: uint64(i)})
|
|
|
if err != nil {
|
|
|
if i <= 4 {
|
|
|
if err != ErrFileNotFound {
|
|
|
@@ -456,7 +458,7 @@ func TestOpenAtUncommittedIndex(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
- w, err := Create(p, nil)
|
|
|
+ w, err := Create(zap.NewExample(), p, nil)
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -468,7 +470,7 @@ func TestOpenAtUncommittedIndex(t *testing.T) {
|
|
|
}
|
|
|
w.Close()
|
|
|
|
|
|
- w, err = Open(p, walpb.Snapshot{})
|
|
|
+ w, err = Open(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -490,7 +492,7 @@ func TestOpenForRead(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
// create WAL
|
|
|
- w, err := Create(p, nil)
|
|
|
+ w, err := Create(zap.NewExample(), p, nil)
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -510,7 +512,7 @@ func TestOpenForRead(t *testing.T) {
|
|
|
w.ReleaseLockTo(unlockIndex)
|
|
|
|
|
|
// All are available for read
|
|
|
- w2, err := OpenForRead(p, walpb.Snapshot{})
|
|
|
+ w2, err := OpenForRead(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -545,7 +547,7 @@ func TestReleaseLockTo(t *testing.T) {
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
// create WAL
|
|
|
- w, err := Create(p, nil)
|
|
|
+ w, err := Create(zap.NewExample(), p, nil)
|
|
|
defer func() {
|
|
|
if err = w.Close(); err != nil {
|
|
|
t.Fatal(err)
|
|
|
@@ -618,7 +620,7 @@ func TestTailWriteNoSlackSpace(t *testing.T) {
|
|
|
defer os.RemoveAll(p)
|
|
|
|
|
|
// create initial WAL
|
|
|
- w, err := Create(p, []byte("metadata"))
|
|
|
+ w, err := Create(zap.NewExample(), p, []byte("metadata"))
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -640,7 +642,7 @@ func TestTailWriteNoSlackSpace(t *testing.T) {
|
|
|
w.Close()
|
|
|
|
|
|
// open, write more
|
|
|
- w, err = Open(p, walpb.Snapshot{})
|
|
|
+ w, err = Open(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -661,7 +663,7 @@ func TestTailWriteNoSlackSpace(t *testing.T) {
|
|
|
w.Close()
|
|
|
|
|
|
// confirm all writes
|
|
|
- w, err = Open(p, walpb.Snapshot{})
|
|
|
+ w, err = Open(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -692,7 +694,7 @@ func TestRestartCreateWal(t *testing.T) {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
|
|
|
- w, werr := Create(p, []byte("abc"))
|
|
|
+ w, werr := Create(zap.NewExample(), p, []byte("abc"))
|
|
|
if werr != nil {
|
|
|
t.Fatal(werr)
|
|
|
}
|
|
|
@@ -701,7 +703,7 @@ func TestRestartCreateWal(t *testing.T) {
|
|
|
t.Fatalf("got %q exists, expected it to not exist", tmpdir)
|
|
|
}
|
|
|
|
|
|
- if w, err = OpenForRead(p, walpb.Snapshot{}); err != nil {
|
|
|
+ if w, err = OpenForRead(zap.NewExample(), p, walpb.Snapshot{}); err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
defer w.Close()
|
|
|
@@ -722,7 +724,7 @@ func TestOpenOnTornWrite(t *testing.T) {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
defer os.RemoveAll(p)
|
|
|
- w, err := Create(p, nil)
|
|
|
+ w, err := Create(zap.NewExample(), p, nil)
|
|
|
defer func() {
|
|
|
if err = w.Close(); err != nil && err != os.ErrInvalid {
|
|
|
t.Fatal(err)
|
|
|
@@ -764,7 +766,7 @@ func TestOpenOnTornWrite(t *testing.T) {
|
|
|
}
|
|
|
f.Close()
|
|
|
|
|
|
- w, err = Open(p, walpb.Snapshot{})
|
|
|
+ w, err = Open(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|
|
|
@@ -785,7 +787,7 @@ func TestOpenOnTornWrite(t *testing.T) {
|
|
|
w.Close()
|
|
|
|
|
|
// read back the entries, confirm number of entries matches expectation
|
|
|
- w, err = OpenForRead(p, walpb.Snapshot{})
|
|
|
+ w, err = OpenForRead(zap.NewExample(), p, walpb.Snapshot{})
|
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
|
}
|