Implement shutdown message

Change-Id: Iedb678f399ff0e0c7284148fc2bf1ca7d5f14eb8

diff --git a/srcs/message.h b/srcs/message.h
index 9e209243e7a1c695f194190538230ce30420a375..bb98b48655647a5be43a088b8ce6c95ea9989e4c 100644 (file)
--- a/srcs/message.h
+++ b/srcs/message.h
@@ -41,6 +41,12 @@ private:
     MessageType m_type;
 };
 
+// Contains only message type (1B)
+class ShutdownMessage : public OutgoingMessage {
+public:
+    ShutdownMessage() : OutgoingMessage(MessageType::SHUTDOWN) {}
+};
+
 class IIncomingMessage {
 public:
     virtual ~IIncomingMessage() = default;

diff --git a/tests/message_tests.cpp b/tests/message_tests.cpp
index 1ea4feff01a460777cc9ff3ea74c0bd7e31cd0cd..bf8c17513d98bc03836943d7a3c778b3ed6e4008 100644 (file)
--- a/tests/message_tests.cpp
+++ b/tests/message_tests.cpp
@@ -414,3 +414,12 @@ TEST(Messages, SerializeMakeCredential2)
     ASSERT_NO_THROW(msg.Serialize(buffer));
     AssertEq(buffer, SAMPLE_MAKE_CREDENTIAL_REQUEST, sizeof(SAMPLE_MAKE_CREDENTIAL_REQUEST));
 }
+
+TEST(Messages, ShutdownMessage)
+{
+    Buffer buffer;
+
+    ShutdownMessage shutdown;
+    ASSERT_NO_THROW(shutdown.Serialize(buffer));
+    AssertEq(buffer, {0x00});
+}