diff options
| -rw-r--r-- | addons/test_mail_full/tests/common.py | 2 | ||||
| -rw-r--r-- | addons/test_mass_mailing/tests/common.py | 5 |
2 files changed, 5 insertions, 2 deletions
diff --git a/addons/test_mail_full/tests/common.py b/addons/test_mail_full/tests/common.py index 1f75394df99..10b2528b5ce 100644 --- a/addons/test_mail_full/tests/common.py +++ b/addons/test_mail_full/tests/common.py @@ -10,7 +10,7 @@ class TestMailFullCommon(TestMassMailCommon, MassSMSCommon): @classmethod def _create_records_for_batch(cls, model, count): - records = cls.env['mail.test.sms'] + records = cls.env[model] partners = cls.env['res.partner'] country_id = cls.env.ref('base.be').id, for x in range(count): diff --git a/addons/test_mass_mailing/tests/common.py b/addons/test_mass_mailing/tests/common.py index 22c3be91464..84f04bb29b1 100644 --- a/addons/test_mass_mailing/tests/common.py +++ b/addons/test_mass_mailing/tests/common.py @@ -25,9 +25,12 @@ class TestMassMailCommon(MassMailCommon, TestMailCommon): @classmethod def _create_test_blacklist_records(cls, model='mailing.test.blacklist', count=1): """ Helper to create data. Currently simple, to be improved. """ + Model = cls.env[model] + email_field = 'email' if 'email' in Model else 'email_from' + records = cls.env[model].create([{ 'name': 'TestRecord_%02d' % x, - 'email_from': '"TestCustomer %02d" <test.record.%02d@test.example.com>' % (x, x), + email_field: '"TestCustomer %02d" <test.record.%02d@test.example.com>' % (x, x), } for x in range(0, count)]) return records |