(WIP) add tests

Author: BewareMyPower

lib/MessageImpl.h

@@ -40,7 +40,6 @@ class BatchMessageContainer;
 
 class MessageImpl {
    public:
-    // TODO: remove this default constructor
     explicit MessageImpl() : encryptionContext_(std::nullopt) {}
     MessageImpl(const MessageId& messageId, const proto::BrokerEntryMetadata& brokerEntryMetadata,
                 const proto::MessageMetadata& metadata, const SharedBuffer& payload,

tests/EncryptionTests.cc

@@ -0,0 +1,89 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+#include <gtest/gtest.h>
+#include <pulsar/Client.h>
+#include <pulsar/ConsumerCryptoFailureAction.h>
+
+static std::string lookupUrl = "pulsar://localhost:6650";
+
+using namespace pulsar;
+
+TEST(EncryptionTests, testEncryptionContext) {
+    Client client{lookupUrl};
+    std::string topic = "test-encryption-context-" + std::to_string(time(nullptr));
+
+    ProducerConfiguration producerConf;
+    producerConf.setCompressionType(CompressionLZ4);
+    producerConf.addEncryptionKey("client-rsa.pem");
+    producerConf.setCryptoKeyReader(std::make_shared<DefaultCryptoKeyReader>(
+        TEST_CONF_DIR "/public-key.client-rsa.pem", TEST_CONF_DIR "/private-key.client-rsa.pem"));
+    // TODO: enable batching
+    producerConf.setBatchingEnabled(false);
+
+    Producer producer;
+    ASSERT_EQ(ResultOk, client.createProducer(topic, producerConf, producer));
+
+    for (int i = 0; i < 5; i++) {
+        producer.sendAsync(MessageBuilder().setContent("msg-" + std::to_string(i)).build(), nullptr);
+    }
+    producer.flush();
+    producer.send(MessageBuilder().setContent("last-msg").build());
+
+    ConsumerConfiguration consumerConf;
+    consumerConf.setSubscriptionInitialPosition(InitialPositionEarliest);
+    consumerConf.setCryptoFailureAction(ConsumerCryptoFailureAction::CONSUME);
+    Consumer consumer;
+    ASSERT_EQ(ResultOk, client.subscribe(topic, "sub", consumerConf, consumer));
+
+    std::vector<Message> messages;
+    for (int i = 0; i < 6; i++) {
+        Message msg;
+        ASSERT_EQ(ResultOk, consumer.receive(msg, 3000));
+        messages.emplace_back(msg);
+    }
+
+    // TODO: improve it
+    for (int i = 0; i < 6; i++) {
+        auto context = messages[i].getEncryptionContext();
+        if (context.has_value()) {
+            std::cout << i << " keys: ";
+            for (auto&& key : context.value()->keys()) {
+                std::cout << key.first << " => " << key.second.keyValue;
+                if (!key.second.metadata.empty()) {
+                    std::cout << " (metadata: ";
+                    for (auto&& entry : key.second.metadata) {
+                        std::cout << entry.first << "=" << entry.second << " ";
+                    }
+                    std::cout << ")";
+                }
+                std::cout << " ";
+            }
+            std::cout << ", algorithm: " << context.value()->algorithm()
+                      << ", param: " << context.value()->param()
+                      << ", compressionType: " << context.value()->compressionType()
+                      << ", uncompressedMessageSize: " << context.value()->uncompressedMessageSize()
+                      << ", batchSize: " << context.value()->batchSize()
+                      << ", isDecryptionFailed: " << context.value()->isDecryptionFailed() << "\n";
+        } else {
+            std::cerr << i << " no encryption context\n";
+        }
+    }
+
+    client.close();
+}