From de6e6ab13cdf3abc86b94eae2647f5a12ff2a98d Mon Sep 17 00:00:00 2001
From: John Wellbelove <john.wellbelove@etlcpp.com>
Date: Sat, 24 Feb 2024 10:46:24 +0000
Subject: [PATCH] Added tests for message base_type

---
 include/etl/message.h |  4 ++--
 test/test_message.cpp | 15 +++++++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/include/etl/message.h b/include/etl/message.h
index efd87210b..6d3bb4c4c 100644
--- a/include/etl/message.h
+++ b/include/etl/message.h
@@ -209,7 +209,7 @@ namespace etl
   };
 
   //***************************************************************************
-  /// Is T derived from etl::message<>
+  /// Is T an etl::message<> or derived from etl::message<>
   //***************************************************************************
   template <typename T>
   struct is_message_type : public etl::bool_constant<etl::is_base_of<etl::message_tag, typename etl::remove_cvref<T>::type>::value>
@@ -225,7 +225,7 @@ namespace etl
   };
 
   //***************************************************************************
-  /// Is T a user defined base of etl::message<T>
+  /// Is T a user defined base of etl::message<T> and not an etl::imessage
   //***************************************************************************
   template <typename T>
   struct is_user_message_base : public etl::bool_constant<etl::is_message_base<T>::value && !is_imessage<T>::value>
diff --git a/test/test_message.cpp b/test/test_message.cpp
index 8ce9afa86..14d40418f 100644
--- a/test/test_message.cpp
+++ b/test/test_message.cpp
@@ -114,4 +114,19 @@ SUITE(test_message)
     CHECK_FALSE(etl::is_user_message_base<int>::value);
 #endif
   }
+
+  //*************************************************************************
+  TEST(test_message_base_type)
+  {
+    struct Message1 : public etl::message<1> {};
+    
+    struct MessageBase : public etl::imessage {};
+    struct Message2 : public etl::message<2, MessageBase> {};
+
+    CHECK_TRUE((std::is_same_v<etl::imessage, Message1::base_type>));
+    CHECK_TRUE((std::is_same_v<MessageBase,   Message2::base_type>));
+
+    CHECK_FALSE((std::is_same_v<MessageBase,   Message1::base_type>));
+    CHECK_FALSE((std::is_same_v<etl::imessage, Message2::base_type>));
+  }
 };