9
9
10
10
use Magento \Email \Model \BackendTemplate ;
11
11
use Magento \Email \Model \Template ;
12
+ use Magento \Framework \App \TemplateTypesInterface ;
13
+ use Magento \Framework \Exception \LocalizedException ;
12
14
use Magento \Framework \Mail \Template \TransportBuilder ;
13
15
use Magento \Framework \ObjectManagerInterface ;
14
16
use Magento \TestFramework \Helper \Bootstrap ;
@@ -47,11 +49,26 @@ protected function setUp()
47
49
*
48
50
* @param string|array $email
49
51
* @dataProvider emailDataProvider
50
- * @throws \Magento\Framework\Exception\ LocalizedException
52
+ * @throws LocalizedException
51
53
*/
52
54
public function testAddToEmail ($ email )
53
55
{
54
- $ templateId = $ this ->template ->load ('email_exception_fixture ' , 'template_code ' )->getId ();
56
+ $ template = $ this ->template ->load ('email_exception_fixture ' , 'template_code ' );
57
+ $ templateId = $ template ->getId ();
58
+
59
+ switch ($ template ->getType ()) {
60
+ case TemplateTypesInterface::TYPE_TEXT :
61
+ $ templateType = MimeInterface::TYPE_TEXT ;
62
+ break ;
63
+
64
+ case TemplateTypesInterface::TYPE_HTML :
65
+ $ templateType = MimeInterface::TYPE_HTML ;
66
+ break ;
67
+
68
+ default :
69
+ $ templateType = '' ;
70
+ $ this ->fail ('Unsupported Mime Type ' );
71
+ }
55
72
56
73
$ this ->builder ->setTemplateModel (BackendTemplate::class);
57
74
@@ -62,9 +79,11 @@ public function testAddToEmail($email)
62
79
$ this ->builder ->addTo ($ email );
63
80
64
81
/** @var EmailMessage $emailMessage */
65
- $ emailMessage = $ this ->builder ->getTransport ();
82
+ $ emailMessage = $ this ->builder ->getTransport ()->getMessage ();
83
+
84
+ $ this ->assertContains ($ templateType , $ emailMessage ->getHeaders ()['Content-Type ' ]);
66
85
67
- $ addresses = $ emailMessage ->getMessage ()-> getTo ();
86
+ $ addresses = $ emailMessage ->getTo ();
68
87
69
88
$ emails = [];
70
89
/** @var Address $toAddress */
0 commit comments