package s3

import (
	"bytes"
	"io/ioutil"
	"os"
	"testing"

	"github.com/micro/go-micro/v3/store"
	"github.com/stretchr/testify/assert"
)

func TestBlobStore(t *testing.T) {
	region := os.Getenv("S3_BLOB_STORE_REGION")
	if len(region) == 0 {
		t.Skipf("Missing required config S3_BLOB_STORE_REGION")
	}

	endpoint := os.Getenv("S3_BLOB_STORE_ENDPOINT")
	if len(endpoint) == 0 {
		t.Skipf("Missing required config S3_BLOB_STORE_ENDPOINT")
	}

	accessKey := os.Getenv("S3_BLOB_STORE_ACCESS_KEY")
	if len(accessKey) == 0 {
		t.Skipf("Missing required config S3_BLOB_STORE_ACCESS_KEY")
	}

	secretKey := os.Getenv("S3_BLOB_STORE_SECRET_KEY")
	if len(secretKey) == 0 {
		t.Skipf("Missing required config S3_BLOB_STORE_SECRET_KEY")
	}

	blob, err := NewBlobStore(
		Region(region),
		Endpoint(endpoint),
		Credentials(accessKey, secretKey),
	)
	assert.NotNilf(t, blob, "Blob should not be nil")
	assert.Nilf(t, err, "Error should be nil")
	if err != nil {
		return
	}

	t.Run("ReadMissingKey", func(t *testing.T) {
		res, err := blob.Read("")
		assert.Equal(t, store.ErrMissingKey, err, "Error should be missing key")
		assert.Nil(t, res, "Result should be nil")
	})

	t.Run("ReadNotFound", func(t *testing.T) {
		res, err := blob.Read("foo")
		assert.Equal(t, store.ErrNotFound, err, "Error should be not found")
		assert.Nil(t, res, "Result should be nil")
	})

	t.Run("WriteMissingKey", func(t *testing.T) {
		buf := bytes.NewBuffer([]byte("HelloWorld"))
		err := blob.Write("", buf)
		assert.Equal(t, store.ErrMissingKey, err, "Error should be missing key")
	})

	t.Run("WriteValid", func(t *testing.T) {
		buf := bytes.NewBuffer([]byte("world"))
		err := blob.Write("hello", buf)
		assert.Nilf(t, err, "Error should be nil")
	})

	t.Run("ReadValid", func(t *testing.T) {
		val, err := blob.Read("hello")
		assert.Nilf(t, err, "Error should be nil")
		assert.NotNilf(t, val, "Value should not be nil")

		if val != nil {
			bytes, err := ioutil.ReadAll(val)
			assert.Nilf(t, err, "Error should be nil")
			assert.Equal(t, "world", string(bytes), "Value should be world")
		}
	})

	t.Run("ReadIncorrectNamespace", func(t *testing.T) {
		val, err := blob.Read("hello", store.BlobNamespace("bar"))
		assert.Equal(t, store.ErrNotFound, err, "Error should be not found")
		assert.Nil(t, val, "Value should be nil")
	})

	t.Run("ReadCorrectNamespace", func(t *testing.T) {
		val, err := blob.Read("hello", store.BlobNamespace("micro"))
		assert.Nil(t, err, "Error should be nil")
		assert.NotNilf(t, val, "Value should not be nil")

		if val != nil {
			bytes, err := ioutil.ReadAll(val)
			assert.Nilf(t, err, "Error should be nil")
			assert.Equal(t, "world", string(bytes), "Value should be world")
		}
	})

	t.Run("DeleteIncorrectNamespace", func(t *testing.T) {
		err := blob.Delete("hello", store.BlobNamespace("bar"))
		assert.Nil(t, err, "Error should be nil")
	})

	t.Run("DeleteCorrectNamespaceIncorrectKey", func(t *testing.T) {
		err := blob.Delete("world", store.BlobNamespace("micro"))
		assert.Nil(t, err, "Error should be nil")
	})

	t.Run("DeleteCorrectNamespace", func(t *testing.T) {
		err := blob.Delete("hello", store.BlobNamespace("micro"))
		assert.Nil(t, err, "Error should be nil")
	})

	t.Run("ReadDeletedKey", func(t *testing.T) {
		res, err := blob.Read("hello", store.BlobNamespace("micro"))
		assert.Equal(t, store.ErrNotFound, err, "Error should be not found")
		assert.Nil(t, res, "Result should be nil")
	})
}