add_partitions_to_txn_request_test.go

//go:build !functional

package sarama

import "testing"

var (
	addPartitionsToTxnRequest = []byte{
		0, 3, 't', 'x', 'n',
		0, 0, 0, 0, 0, 0, 31, 64, // ProducerID
		0, 0, // ProducerEpoch
		0, 0, 0, 1, // 1 topic
		0, 5, 't', 'o', 'p', 'i', 'c',
		0, 0, 0, 1, 0, 0, 0, 1,
	}
	addPartitionsToTxnRequestV3 = []byte{
		4, 't', 'x', 'n',
		0, 0, 0, 0, 0, 0, 31, 64, // ProducerID
		0, 0, // ProducerEpoch
		2, // 1 topic
		6, 't', 'o', 'p', 'i', 'c',
		2, // 1 partition
		0, 0, 0, 1,
		0,
		0,
	}
)

func TestAddPartitionsToTxnRequest(t *testing.T) {
	req := &AddPartitionsToTxnRequest{
		TransactionalID: "txn",
		ProducerID:      8000,
		ProducerEpoch:   0,
		TopicPartitions: map[string][]int32{
			"topic": {1},
		},
	}

	testRequest(t, "", req, addPartitionsToTxnRequest)
}

func TestAddPartitionsToTxnRequestV3(t *testing.T) {
	req := &AddPartitionsToTxnRequest{
		Version:         3,
		TransactionalID: "txn",
		ProducerID:      8000,
		ProducerEpoch:   0,
		TopicPartitions: map[string][]int32{
			"topic": {1},
		},
	}

	testRequest(t, "", req, addPartitionsToTxnRequestV3)
}