fix last commit (also skip print statement)

This commit is contained in:
Pavol Rusnak 2018-01-04 16:55:27 +01:00
parent 4e01971e4c
commit f587135b8d
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 3 additions and 1 deletions

View File

@ -405,8 +405,10 @@ class DebugLinkMixin(object):
"Expected %s, got %s" % (pprint(expected), pprint(msg)))
for field, value in expected.__dict__.items():
if getattr(expected, field) is None:
continue
print("EXPECTED", getattr(expected, field), getattr(msg, field), field, value)
if getattr(expected, field) is not None and getattr(msg, field) != value:
if getattr(msg, field) != value:
raise CallException(proto.FailureType.UnexpectedMessage,
"Expected %s, got %s" % (pprint(expected), pprint(msg)))