From d4944ff373a6a5fded60c45ea00ea0d3f12c3685 Mon Sep 17 00:00:00 2001
From: Buke Po <bukepo@gmail.com>
Date: Thu, 3 Aug 2017 12:49:55 +0800
Subject: [PATCH] [tests] add missing message type (#2053)

---
 tests/scripts/thread-cert/message.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/scripts/thread-cert/message.py b/tests/scripts/thread-cert/message.py
index e9b80cde6..020a02508 100644
--- a/tests/scripts/thread-cert/message.py
+++ b/tests/scripts/thread-cert/message.py
@@ -45,9 +45,10 @@ class MessageType(IntEnum):
     MLE = 0
     COAP = 1
     ICMP = 2
-    ACK = 3 
+    ACK = 3
     BEACON = 4
     DATA = 5
+    COMMAND = 6
 
 
 class Message(object):
@@ -110,6 +111,10 @@ class Message(object):
 
         elif self._mac_header.frame_type == mac802154.MacHeader.FrameType.DATA:
             self._type = MessageType.DATA
+        elif self._mac_header.frame_type == mac802154.MacHeader.FrameType.COMMAND:
+            self._type = MessageType.COMMAND
+        else:
+            raise ValueError('Invalid mac frame type %d' % self._mac_header.frame_type)
 
     @property
     def ipv6_packet(self):