From 13f2bac16b8fc4f3e3f4df9c7b1a0024be236ef1 Mon Sep 17 00:00:00 2001
From: Alex Meade <alex.meade@rackspace.com>
Date: Thu, 8 Mar 2012 15:52:51 +0000
Subject: Add a few missing tests regarding exception codes.

Change-Id: I051f47555669af6a74fba058867a6ac2cff45a65
---
 nova/tests/test_exception.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/nova/tests/test_exception.py b/nova/tests/test_exception.py
index a748b6567..9fe568229 100644
--- a/nova/tests/test_exception.py
+++ b/nova/tests/test_exception.py
@@ -105,8 +105,7 @@ class NovaExceptionTestCase(test.TestCase):
             message = "default message"
 
         exc = FakeNovaException()
-        self.assertEquals(unicode(exc),
-                          'default message')
+        self.assertEquals(unicode(exc), 'default message')
 
     def test_error_msg(self):
         self.assertEquals(unicode(exception.NovaException('test')),
@@ -117,5 +116,18 @@ class NovaExceptionTestCase(test.TestCase):
             message = "default message: %(code)s"
 
         exc = FakeNovaException(code=500)
-        self.assertEquals(unicode(exc),
-                          'default message: 500')
+        self.assertEquals(unicode(exc), 'default message: 500')
+
+    def test_default_error_code(self):
+        class FakeNovaException(exception.NovaException):
+            code = 404
+
+        exc = FakeNovaException()
+        self.assertEquals(exc.kwargs['code'], 404)
+
+    def test_error_code_from_kwarg(self):
+        class FakeNovaException(exception.NovaException):
+            code = 500
+
+        exc = FakeNovaException(code=404)
+        self.assertEquals(exc.kwargs['code'], 404)
-- 
cgit