func benchmarkLogEntryDecoderDecode(b *testing.B, sz int) { var buf bytes.Buffer enc := raft.NewLogEntryEncoder(&buf) dec := raft.NewLogEntryDecoder(&buf) // Encode a single record and record its size. enc.Encode(&raft.LogEntry{Data: make([]byte, sz)}) b.SetBytes(int64(buf.Len())) // Encode all the records on the buffer first. buf.Reset() for i := 0; i < b.N; i++ { if err := enc.Encode(&raft.LogEntry{Data: make([]byte, sz)}); err != nil { b.Fatalf("encode: %s", err) } } b.ReportAllocs() // Decode from the buffer. b.ResetTimer() for i := 0; i < b.N; i++ { var entry raft.LogEntry if err := dec.Decode(&entry); err != nil { b.Fatal(err) } } b.StopTimer() runtime.GC() }
// Ensure that the encoder can handle write errors during encoding of the data. func TestLogEntryEncoder_Encode_ErrShortWrite_Data(t *testing.T) { w := newLimitWriter(25) enc := raft.NewLogEntryEncoder(w) if err := enc.Encode(&raft.LogEntry{Data: []byte{0, 0, 0, 0}}); err != io.ErrShortWrite { t.Fatalf("unexpected error: %s", err) } }
// Ensure that log entries can be encoded to a writer. func TestLogEntryEncoder_Encode(t *testing.T) { var buf bytes.Buffer // Encode the entry to the buffer. enc := raft.NewLogEntryEncoder(&buf) if err := enc.Encode(&raft.LogEntry{Index: 2, Term: 3, Data: []byte{4, 5, 6}}); err != nil { t.Fatalf("unexpected error: %s", err) } // Check that the encoded bytes match what's expected. exp := []byte{0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 3, 4, 5, 6} if v := buf.Bytes(); !bytes.Equal(exp, v) { t.Fatalf("value:\n\nexp: %x\n\ngot: %x\n\n", exp, v) } }
func benchmarkLogEntryEncoderEncode(b *testing.B, sz int) { var buf bytes.Buffer enc := raft.NewLogEntryEncoder(&buf) entry := &raft.LogEntry{Data: make([]byte, sz)} // Record single encoding size. enc.Encode(entry) b.SetBytes(int64(buf.Len())) b.ReportAllocs() b.ResetTimer() for i := 0; i < b.N; i++ { if err := enc.Encode(entry); err != nil { b.Fatal(err) } } b.StopTimer() runtime.GC() }
// Ensure that random entries can be encoded and decoded correctly. func TestLogEntryEncodeDecode(t *testing.T) { f := func(entries []raft.LogEntry) bool { var buf bytes.Buffer enc := raft.NewLogEntryEncoder(&buf) dec := raft.NewLogEntryDecoder(&buf) // Encode entries. for _, e := range entries { if e.Type == 0xFF { buf.WriteByte(0xFF) continue } if err := enc.Encode(&e); err != nil { t.Fatalf("encode: %s", err) } } // Decode entries. for _, e := range entries { var entry raft.LogEntry if err := dec.Decode(&entry); err != nil { t.Fatalf("decode: %s", err) } else if entry.Type == 0xFF { if !reflect.DeepEqual(&entry, &raft.LogEntry{Type: 0xFF}) { t.Fatalf("invalid snapshot entry: %#v", &entry) } } else if !reflect.DeepEqual(e, entry) { t.Fatalf("mismatch:\n\nexp: %#v\n\ngot: %#v\n\n", e, entry) } } return true } if err := quick.Check(f, nil); err != nil { t.Error(err) } }