Function assertKeyValue

pkg/buffercache/helper_test.go:32–67

(t *testing.T, cache *BufferCache, key string, want []byte)

}

func assertKeyValue(t *testing.T, cache *BufferCache, key string, want []byte) {
	t.Helper()

	loc, ok := cache.m.m[key]
	if !ok {
		t.Fatalf(`key "%s" not found in location map`, key)
	}

	var (
		buf        []byte
		compressed bool
	)
	switch {
	case loc.isCold():
		bidx, cidx := loc.coldBucketIdx(), loc.coldCellIdx()
		buf, compressed = cache.cold.buckets[bidx].get(cidx)
	case loc.isHot():
		buf = cache.hot.m[key]

		// Ensure there's only one buffer
		prefix := binary.LittleEndian.Uint32(buf[:4])
		buflen := prefix & 0x7fffffff
		compressed = (prefix & 0x80000000) != 0
		buf = buf[4:]
		if int(buflen) != len(buf) {
			t.Fatalf("assertKeyValue only compares single buffer: got buflen != len(buf[4:])")
		}
	}

	if compressed {
		buf = cache.decomp.uncompressSimple(buf)
	}
	if !bytes.Equal(buf, want) {
		t.Errorf(`buffers don't match for key "%s", got = %s, want %s`, key, buf, want)
	}
}

func assertKeyNotInCache(t *testing.T, cache *BufferCache, key string) {
	t.Helper()