diff options
Diffstat (limited to 'tests/integration/mail/adaptors/test_soledad_adaptor.py')
| -rw-r--r-- | tests/integration/mail/adaptors/test_soledad_adaptor.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/integration/mail/adaptors/test_soledad_adaptor.py b/tests/integration/mail/adaptors/test_soledad_adaptor.py index 72b0be8f..14e0e407 100644 --- a/tests/integration/mail/adaptors/test_soledad_adaptor.py +++ b/tests/integration/mail/adaptors/test_soledad_adaptor.py @@ -396,9 +396,9 @@ class SoledadMailAdaptorTestCase(SoledadTestMixin): msg = adaptor.get_msg_from_string(MessageClass, raw) def check_create_result(created): - # that's one mdoc, one hdoc, one fdoc, one cdoc - self.assertEqual(len(created), 4) - for doc in created: + # that's the wrapper + self.assertEqual(created.__class__.__name__, 'MessageWrapper') + for doc in [created.mdoc, created.fdoc, created.hdoc]: self.assertTrue( doc.__class__.__name__, "SoledadDocument") |