From 0d2c98571113c820804aa896409d03c90383f772 Mon Sep 17 00:00:00 2001
From: Zbysek MRAZ <zmraz@redhat.com>
Date: Mon, 27 Jun 2011 16:37:09 +0200
Subject: Nitrate: adding validity check to classes Status and Priority

---
 Nitrate.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Nitrate.py b/Nitrate.py
index bcb36e1..aa79c84 100644
--- a/Nitrate.py
+++ b/Nitrate.py
@@ -300,6 +300,8 @@ class Priority(Nitrate):
         """
         self._id = priority
         if isinstance(priority, int):
+            if priority < 1 or priority > 5:
+                raise NitrateError("Not a valid Priority id: '{0}'".format(priority))
             self._id = priority
         else:
             try:
@@ -462,6 +464,8 @@ class Status(Nitrate):
         IDLE, PASSED, FAILED, RUNNING, PAUSED, BLOCKED, ERROR, WAIVED
         """
         if isinstance(status, int):
+            if status < 1 or status > 8:
+                raise NitrateError("Not a valid Status id: '{0}'".format(status))
             self._id = status
         else:
             try:
-- 
cgit