package cassandra

import (
	"context"
	"testing"
	"time"

	"github.com/gocql/gocql"
	"github.com/stretchr/testify/assert"
)

func TestBatchTx_AllSuccess(t *testing.T) {
	t.Parallel()

	ctx := context.Background()
	ks := generateRandomKeySpace(t)
	now := time.Now()
	id1 := gocql.TimeUUID()
	id2 := gocql.TimeUUID()

	tx := cassandraDBTest.NewBatch(ctx, ks)
	err := tx.Insert(&MonkeyEntity{ID: id1, Name: "Alice", UpdateAt: now, CreateAt: now})
	assert.NoError(t, err)
	err = tx.Insert(&MonkeyEntity{ID: id2, Name: "Bob", UpdateAt: now, CreateAt: now})
	assert.NoError(t, err)
	err = tx.Update(&MonkeyEntity{ID: id1, Name: "Alice", UpdateAt: now.Add(5 * time.Minute)})
	assert.NoError(t, err)
	err = tx.Delete(&MonkeyEntity{ID: id2, Name: "Bob"})
	assert.NoError(t, err)

	err = tx.Commit()
	assert.NoError(t, err)

	// Alice 應該還在，且被更新
	var alice MonkeyEntity
	alice.ID, alice.Name = id1, "Alice"
	err = cassandraDBTest.Get(ctx, &alice, ks)
	assert.NoError(t, err)
	assert.WithinDuration(t, now.Add(5*time.Minute), alice.UpdateAt, time.Second)

	// Bob 應該被刪除
	err = cassandraDBTest.Get(ctx, &MonkeyEntity{ID: id2, Name: "Bob"}, ks)
	assert.Error(t, err)
}