coderzc commented on code in PR #17289: URL: https://github.com/apache/pulsar/pull/17289#discussion_r965006126
########## pulsar-client-cpp/tests/ProducerTest.cc: ########## @@ -210,3 +213,84 @@ TEST(ProducerTest, testBacklogQuotasExceeded) { client.close(); } + +TEST(ProducerTest, testMaxMessageSize) { + Client client(serviceUrl); + + const std::string topic = "ProducerTest-MaxMessageSize-" + std::to_string(time(nullptr)); + + Consumer consumer; + ASSERT_EQ(ResultOk, client.subscribe(topic, "sub", consumer)); + + Producer producer; + ASSERT_EQ(ResultOk, client.createProducer(topic, producer)); + + std::string msg = std::string(maxMessageSize / 2, 'a'); + ASSERT_EQ(ResultOk, producer.send(MessageBuilder().setContent(msg).build())); + Message message; + ASSERT_EQ(ResultOk, consumer.receive(message)); + ASSERT_EQ(msg, message.getDataAsString()); + + std::string orderKey = std::string(maxMessageSize, 'a'); + ASSERT_EQ(ResultMessageTooBig, producer.send(MessageBuilder().setOrderingKey(orderKey).build())); + + ASSERT_EQ(ResultMessageTooBig, + producer.send(MessageBuilder().setContent(std::string(maxMessageSize, 'b')).build())); + + client.close(); +} + +TEST(ProducerTest, testNoBatchMaxMessageSize) { + Client client(serviceUrl); + + const std::string topic = "ProducerTest-NoBatchMaxMessageSize-" + std::to_string(time(nullptr)); + + Consumer consumer; + ASSERT_EQ(ResultOk, client.subscribe(topic, "sub", consumer)); + + Producer producer; + ProducerConfiguration conf; + conf.setBatchingEnabled(false); + ASSERT_EQ(ResultOk, client.createProducer(topic, conf, producer)); + + std::string msg = std::string(maxMessageSize / 2, 'a'); + ASSERT_EQ(ResultOk, producer.send(MessageBuilder().setContent(msg).build())); + Message message; + ASSERT_EQ(ResultOk, consumer.receive(message)); + ASSERT_EQ(msg, message.getDataAsString()); + + std::string orderKey = std::string(maxMessageSize, 'a'); + ASSERT_EQ(ResultMessageTooBig, producer.send(MessageBuilder().setOrderingKey(orderKey).build())); + + ASSERT_EQ(ResultMessageTooBig, + producer.send(MessageBuilder().setContent(std::string(maxMessageSize, 'b')).build())); + + client.close(); +} + +TEST(ProducerTest, testChunkingMaxMessageSize) { Review Comment: This is a test to enable Chunking, some cases different -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@pulsar.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org