Function TestTransactions

plugin/plugin_test/kv_storage_test.go:189–247

Test transactions including rollback and nested transactions

(t *testing.T)

// Test transactions including rollback and nested transactions
func TestTransactions(t *testing.T) {
	setupTestEnvironment()
	kv := testPlugin.operator
	ctx := context.Background()

	t.Run("SuccessfulTransaction", func(t *testing.T) {
		err := kv.Tx(ctx, func(ctx context.Context, txKv *plugin.KVOperator) error {
			if err := txKv.Set(ctx, plugin.KVParams{Group: "tx_group", Key: "tx_key1", Value: "tx_value1"}); err != nil {
				return err
			}
			if err := txKv.Set(ctx, plugin.KVParams{Group: "tx_group", Key: "tx_key2", Value: "tx_value2"}); err != nil {
				return err
			}
			return nil
		})
		if err != nil {
			t.Fatalf("Successful transaction failed: %v", err)
		}

		mustGet(t, kv, ctx, "tx_group", "tx_key1", "tx_value1")
		mustGet(t, kv, ctx, "tx_group", "tx_key2", "tx_value2")
	})

	t.Run("TransactionRollback", func(t *testing.T) {
		err := kv.Tx(ctx, func(ctx context.Context, txKv *plugin.KVOperator) error {
			if err := txKv.Set(ctx, plugin.KVParams{Group: "tx_group", Key: "tx_key3", Value: "tx_value3"}); err != nil {
				return err
			}
			return fmt.Errorf("mock error")
		})
		if err == nil {
			t.Error("Expected transaction to fail but it succeeded")
		}

		assertNotFound(t, kv, ctx, "tx_group", "tx_key3")
	})

	t.Run("NestedTransactions", func(t *testing.T) {
		err := kv.Tx(ctx, func(ctx context.Context, txKv *plugin.KVOperator) error {
			if err := txKv.Set(ctx, plugin.KVParams{Group: "nested", Key: "key1", Value: "value1"}); err != nil {
				return err
			}

			return txKv.Tx(ctx, func(ctx context.Context, nestedKv *plugin.KVOperator) error {
				if err := nestedKv.Set(ctx, plugin.KVParams{Group: "nested", Key: "key2", Value: "value2"}); err != nil {
					return err
				}
				return fmt.Errorf("mock nested error")
			})
		})
		if err == nil {
			t.Error("Expected nested transaction to fail but it succeeded")
		}

		// Verify outer transaction also rolled back
		assertNotFound(t, kv, ctx, "nested", "key1")
		assertNotFound(t, kv, ctx, "nested", "key2")
	})