Browse Source

Added a useful unit test primitive for asserting object attributes

tags/v0.3.3
Paul "LeoNerd" Evans 9 years ago
parent
commit
7aacd6834a
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      tests/unittest.py

+ 11
- 0
tests/unittest.py View File

@@ -71,6 +71,17 @@ class TestCase(unittest.TestCase):
logging.getLogger().setLevel(level)
return orig()

def assertObjectHasAttributes(self, attrs, obj):
"""Asserts that the given object has each of the attributes given, and
that the value of each matches according to assertEquals."""
for (key, value) in attrs.items():
if not hasattr(obj, key):
raise AssertionError("Expected obj to have a '.%s'" % key)
try:
self.assertEquals(attrs[key], getattr(obj, key))
except AssertionError as e:
raise (type(e))(e.message + " for '.%s'" % key)


def DEBUG(target):
"""A decorator to set the .loglevel attribute to logging.DEBUG.


Loading…
Cancel
Save