added serialize() method to QDnsHeader and associated tests

src/modules/dns/includes/dns/QDnsHeader.h

@@ -17,6 +17,7 @@ namespace QDNS
         QDnsHeader();
 
         static QDnsHeader parse(const QByteArray& data);
+        QByteArray serialize() const;
 
         bool isValid() const;
 

src/modules/dns/src/QDnsHeader.cpp

@@ -230,4 +230,21 @@ namespace QDNS
         m_additionalCount = additionalCount;
     }
 
+    QByteArray QDnsHeader::serialize() const
+    {
+        if (isValid())
+        {
+            QByteArray data;
+            QDataStream ds(&data, QIODevice::WriteOnly);
+            ds << m_id;
+            ds << m_flags;
+            ds << m_questionCount;
+            ds << m_answerCount;
+            ds << m_authorityCount;
+            ds << m_additionalCount;
+            return data;
+        }
+        return QByteArray();
+    }
+
 }

tests/main.cpp

@@ -1,5 +1,6 @@
 #include <gtest/gtest.h>
-#include <QtCore/QByteArray>
+#include <QByteArray>
+#include <QString>
 
 #include "dns/QDnsHeader.h"
 
@@ -57,12 +58,24 @@ TEST_P(DNSHeaderTest, parseAndGetters)
         ASSERT_EQ(header.getAnswerCount(), params.answerCount);
         ASSERT_EQ(header.getAuthorityCount(), params.authorityCount);
         ASSERT_EQ(header.getAdditionalCount(), params.additionalCount);
+
+        ASSERT_EQ(header.serialize().toHex(), QString(params.hexData).remove(" "));
+    }
+    else
+    {
+        ASSERT_EQ(header.serialize().size(), 0);
     }
 }
 
 TEST_P(DNSHeaderTest, settersAndGetters1)
 {
     auto params = GetParam();
+
+    if (!params.isValid)
+    {
+        return;
+    }
+
     auto data = QByteArray::fromHex(params.hexData);
     QDNS::QDnsHeader header;
 
@@ -90,11 +103,19 @@ TEST_P(DNSHeaderTest, settersAndGetters1)
     ASSERT_EQ(header.getAnswerCount(), params.answerCount);
     ASSERT_EQ(header.getAuthorityCount(), params.authorityCount);
     ASSERT_EQ(header.getAdditionalCount(), params.additionalCount);
+
+    ASSERT_EQ(header.serialize().toHex(), QString(params.hexData).remove(" "));
 }
 
 TEST_P(DNSHeaderTest, settersAndGetters2)
 {
     auto params = GetParam();
+
+    if (!params.isValid)
+    {
+        return;
+    }
+
     auto data = QByteArray::fromHex(params.hexData);
     QDNS::QDnsHeader header;
 
@@ -131,6 +152,8 @@ TEST_P(DNSHeaderTest, settersAndGetters2)
     ASSERT_EQ(header.getAnswerCount(), params.answerCount);
     ASSERT_EQ(header.getAuthorityCount(), params.authorityCount);
     ASSERT_EQ(header.getAdditionalCount(), params.additionalCount);
+
+    ASSERT_EQ(header.serialize().toHex(), QString(params.hexData).remove(" "));
 }
 
 INSTANTIATE_TEST_CASE_P(DNSHeaderTestInst,