Add a few missing tests regarding exception codes.

Change-Id: I051f47555669af6a74fba058867a6ac2cff45a65

1 files 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)