package blobstor

import (
	"os"
	"testing"

	"github.com/nspcc-dev/neofs-node/pkg/core/object"
	"github.com/stretchr/testify/require"
)

func TestCompression(t *testing.T) {
	dir, err := os.MkdirTemp("", "neofs*")
	require.NoError(t, err)
	t.Cleanup(func() { _ = os.RemoveAll(dir) })

	const (
		smallSizeLimit = 512
		objCount       = 4
	)

	newBlobStor := func(t *testing.T, compress bool) *BlobStor {
		bs := New(WithCompressObjects(compress),
			WithRootPath(dir),
			WithSmallSizeLimit(smallSizeLimit))
		require.NoError(t, bs.Open())
		require.NoError(t, bs.Init())
		return bs
	}

	bigObj := make([]*object.Object, objCount)
	smallObj := make([]*object.Object, objCount)
	for i := 0; i < objCount; i++ {
		bigObj[i] = testObject(smallSizeLimit * 2)
		smallObj[i] = testObject(smallSizeLimit / 2)
	}

	testGet := func(t *testing.T, b *BlobStor, i int) {
		res1, err := b.GetSmall(&GetSmallPrm{address: address{smallObj[i].Address()}})
		require.NoError(t, err)
		require.Equal(t, smallObj[i], res1.Object())

		res2, err := b.GetBig(&GetBigPrm{address: address{bigObj[i].Address()}})
		require.NoError(t, err)
		require.Equal(t, bigObj[i], res2.Object())
	}

	testPut := func(t *testing.T, b *BlobStor, i int) {
		prm := new(PutPrm)
		prm.SetObject(smallObj[i])
		_, err = b.Put(prm)
		require.NoError(t, err)

		prm = new(PutPrm)
		prm.SetObject(bigObj[i])
		_, err = b.Put(prm)
		require.NoError(t, err)
	}

	// Put and Get uncompressed object
	blobStor := newBlobStor(t, false)
	testPut(t, blobStor, 0)
	testGet(t, blobStor, 0)
	require.NoError(t, blobStor.Close())

	blobStor = newBlobStor(t, true)
	testGet(t, blobStor, 0) // get uncompressed object with compress enabled
	testPut(t, blobStor, 1)
	testGet(t, blobStor, 1)
	require.NoError(t, blobStor.Close())

	blobStor = newBlobStor(t, false)
	testGet(t, blobStor, 0) // get old uncompressed object
	testGet(t, blobStor, 1) // get compressed object with compression disabled
	testPut(t, blobStor, 2)
	testGet(t, blobStor, 2)
	require.NoError(t, blobStor.Close())
}