From 227a4b3a3d531cca3e8b6bdf29a0ea4480e6a595 Mon Sep 17 00:00:00 2001
From: Wiebe Cazemier <wiebe@halfgaar.net>
Date: Sun, 7 Aug 2022 13:17:45 +0200
Subject: [PATCH] Merged test_very_big_packet and test_packet_bigger_than_one_doubling

---
 FlashMQTests/tst_maintests.cpp | 60 ++++++++++++++++++++----------------------------------------
 flashmqtestclient.cpp          |  1 +
 2 files changed, 21 insertions(+), 40 deletions(-)

diff --git a/FlashMQTests/tst_maintests.cpp b/FlashMQTests/tst_maintests.cpp
index f20cb4d..1f8ec36 100644
--- a/FlashMQTests/tst_maintests.cpp
+++ b/FlashMQTests/tst_maintests.cpp
@@ -84,8 +84,7 @@ private slots:
     void test_retained_removed();
     void test_retained_tree();
 
-    void test_packet_bigger_than_one_doubling();
-    void test_very_big_packet();
+    void test_various_packet_sizes();
 
     void test_acl_tree();
     void test_acl_tree2();
@@ -540,61 +539,42 @@ void MainTests::test_retained_tree()
 
 }
 
-void MainTests::test_packet_bigger_than_one_doubling()
+void MainTests::test_various_packet_sizes()
 {
     std::vector<ProtocolVersion> protocols {ProtocolVersion::Mqtt311, ProtocolVersion::Mqtt5};
+    std::list<std::string> payloads {std::string(8000,3), std::string(10*1024*1024, 5)};
 
     for (const ProtocolVersion senderVersion : protocols)
     {
         for (const ProtocolVersion receiverVersion : protocols)
         {
-            FlashMQTestClient sender;
-            FlashMQTestClient receiver;
+            for (const std::string &payload : payloads)
+            {
+                FlashMQTestClient sender;
+                FlashMQTestClient receiver;
 
-            std::string payload(8000, 3);
-            std::string topic = "hugepacket";
+                std::string topic = "hugepacket";
 
-            sender.start();
-            sender.connectClient(senderVersion);
+                sender.start();
+                sender.connectClient(senderVersion);
 
-            receiver.start();
-            receiver.connectClient(receiverVersion);
-            receiver.subscribe(topic, 0);
+                receiver.start();
+                receiver.connectClient(receiverVersion);
+                receiver.subscribe(topic, 0);
 
-            sender.publish(topic, payload, 0);
-            receiver.waitForMessageCount(1);
+                sender.publish(topic, payload, 0);
+                receiver.waitForMessageCount(1, 2);
 
-            MYCASTCOMPARE(receiver.receivedPublishes.size(), 1);
+                MYCASTCOMPARE(receiver.receivedPublishes.size(), 1);
 
-            MqttPacket &msg = receiver.receivedPublishes.front();
-            QCOMPARE(msg.getPayloadCopy(), payload);
-            QVERIFY(!msg.getRetain());
+                MqttPacket &msg = receiver.receivedPublishes.front();
+                QCOMPARE(msg.getPayloadCopy(), payload);
+                QVERIFY(!msg.getRetain());
+            }
         }
     }
 }
 
-// This tests our write buffer, and that it's emptied during writing already.
-void MainTests::test_very_big_packet()
-{
-    TwoClientTestContext testContext;
-
-    QByteArray payload(10*1024*1024, 3);
-    QString topic = "hugepacket";
-
-    testContext.connectSender();
-    testContext.connectReceiver();
-    testContext.subscribeReceiver(topic);
-
-    testContext.publish(topic, payload);
-    testContext.waitReceiverReceived(1);
-
-    QCOMPARE(testContext.receivedMessages.count(), 1);
-
-    QMQTT::Message msg = testContext.receivedMessages.first();
-    QCOMPARE(msg.payload(), payload);
-    QVERIFY(!msg.retain());
-}
-
 void MainTests::test_acl_tree()
 {
     AclTree aclTree;
diff --git a/flashmqtestclient.cpp b/flashmqtestclient.cpp
index 6f5b449..9d6a32f 100644
--- a/flashmqtestclient.cpp
+++ b/flashmqtestclient.cpp
@@ -166,6 +166,7 @@ void FlashMQTestClient::connectClient(ProtocolVersion protocolVersion, bool clea
     this->client->writeMqttPacketAndBlameThisClient(connectPack);
 
     waitForConnack();
+    this->client->setAuthenticated(true);
 }
 
 void FlashMQTestClient::subscribe(const std::string topic, char qos)
--
libgit2 0.21.4