From 419c41e2cc39810a4da75c22f32e285fea517533 Mon Sep 17 00:00:00 2001
From: Lucio Seki <lseki@lenovo.com>
Date: Thu, 27 Aug 2015 08:51:42 -0300
Subject: [PATCH] Add NMI diag command

---
 confluent_server/confluent/messages.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/confluent_server/confluent/messages.py b/confluent_server/confluent/messages.py
index 0e92c815..767d9f8b 100644
--- a/confluent_server/confluent/messages.py
+++ b/confluent_server/confluent/messages.py
@@ -538,6 +538,7 @@ class InputPowerMessage(ConfluentInputMessage):
         'off',
         'reset',
         'boot',
+        'diag',
         'shutdown',
     ])
 
@@ -650,6 +651,7 @@ class PowerState(ConfluentChoiceMessage):
         'off',
         'reset',
         'boot',
+        'diag',
     ])
     keyname = 'state'