Function TestNewMemoryWriter

internal/caches/writer_memory_test.go:16–55

(t *testing.T)

)

func TestNewMemoryWriter(t *testing.T) {
	var storage = caches.NewMemoryStorage(&serverconfigs.HTTPCachePolicy{
		Id:          0,
		IsOn:        false,
		Name:        "",
		Description: "",
		Capacity: &shared.SizeCapacity{
			Count: 8,
			Unit:  shared.SizeCapacityUnitGB,
		},
	}, nil)
	err := storage.Init()
	if err != nil {
		t.Fatal(err)
	}

	const size = 1 << 20
	const chunkSize = 16 << 10
	var data = bytes.Repeat([]byte{'A'}, chunkSize)

	var before = time.Now()

	var writer = caches.NewMemoryWriter(storage, "a", time.Now().Unix()+3600, 200, false, size, 1<<30, func(valueItem *caches.MemoryItem) {
		t.Log(len(valueItem.BodyValue), "bytes")
	})

	for i := 0; i < size/chunkSize; i++ {
		_, err = writer.Write(data)
		if err != nil {
			t.Fatal(err)
		}
	}

	err = writer.Close()
	if err != nil {
		t.Fatal(err)
	}

	t.Log("cost:", time.Since(before).Seconds()*1000, "ms")
}

func BenchmarkMemoryWriter_Capacity(b *testing.B) {
	b.ReportAllocs()