mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2024-10-01 11:49:51 -04:00
Added a useful unit test primitive for asserting object attributes
This commit is contained in:
parent
de14853237
commit
7aacd6834a
@ -71,6 +71,17 @@ class TestCase(unittest.TestCase):
|
|||||||
logging.getLogger().setLevel(level)
|
logging.getLogger().setLevel(level)
|
||||||
return orig()
|
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):
|
def DEBUG(target):
|
||||||
"""A decorator to set the .loglevel attribute to logging.DEBUG.
|
"""A decorator to set the .loglevel attribute to logging.DEBUG.
|
||||||
|
Loading…
Reference in New Issue
Block a user