package wal import ( "bytes" crand "crypto/rand" "io" "git.crumpington.com/public/jldb/lib/atomicheader" "git.crumpington.com/public/jldb/lib/errs" "path/filepath" "testing" "time" ) func newSegmentForTesting(t *testing.T) *segment { tmpDir := t.TempDir() seg, err := createSegment(filepath.Join(tmpDir, "x"), 1, 100, 200) if err != nil { t.Fatal(err) } return seg } func TestNewSegmentDirNotFound(t *testing.T) { t.Parallel() tmpDir := t.TempDir() p := filepath.Join(tmpDir, "notFound", "1245") if _, err := createSegment(p, 1, 1234, 5678); !errs.IO.Is(err) { t.Fatal(err) } } func TestOpenSegmentNotFound(t *testing.T) { t.Parallel() tmpDir := t.TempDir() p := filepath.Join(tmpDir, "notFound") if _, err := openSegment(p, 1); !errs.IO.Is(err) { t.Fatal(err) } } func TestOpenSegmentTruncatedFile(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) path := seg.f.Name() if err := seg.f.Truncate(4); err != nil { t.Fatal(err) } seg.Close() if _, err := openSegment(path, 1); !errs.IO.Is(err) { t.Fatal(err) } } func TestOpenSegmentCorruptHeader(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) path := seg.f.Name() buf := make([]byte, atomicheader.ReservedBytes) crand.Read(buf) if _, err := seg.f.Seek(0, io.SeekStart); err != nil { t.Fatal(err) } if _, err := seg.f.Write(buf); err != nil { t.Fatal(err) } seg.Close() if _, err := openSegment(path, 1); !errs.Corrupt.Is(err) { t.Fatal(err) } } func TestOpenSegmentCorruptHeader2(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) path := seg.f.Name() buf := make([]byte, 1024) // 2 pages. crand.Read(buf) if _, err := seg.f.Seek(1024, io.SeekStart); err != nil { t.Fatal(err) } if _, err := seg.f.Write(buf); err != nil { t.Fatal(err) } seg.Close() if _, err := openSegment(path, 1); !errs.Corrupt.Is(err) { t.Fatal(err) } } func TestSegmentArchiveTwice(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) for i := 0; i < 2; i++ { if err := seg.Archive(); err != nil { t.Fatal(err) } } } func TestSegmentAppendArchived(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) appendRandomRecords(t, seg, 8) if err := seg.Archive(); err != nil { t.Fatal(err) } _, _, err := seg.Append(4, bytes.NewBuffer([]byte{1, 2, 3, 4})) if !errs.Archived.Is(err) { t.Fatal(err) } } func TestSegmentAppendRecordInvalidSeqNum(t *testing.T) { t.Parallel() seg := newSegmentForTesting(t) appendRandomRecords(t, seg, 8) // 109 is next. _, _, err := seg.appendRecord(Record{ SeqNum: 110, TimestampMS: time.Now().UnixMilli(), DataSize: 100, }) if !errs.Unexpected.Is(err) { t.Fatal(err) } }