@@ -186,34 +186,67 @@ def test_campaign():
186186 assert actual == expected
187187
188188
189+ def test_metadata ():
190+ email_message = EmailMessage (
191+ to = [
192+ {
193+ 'address' : 'to@example.com' ,
194+ 'metadata' : {
195+ 'key' : 'value'
196+ }
197+ }
198+ ],
199+ from_email = 'test@from.com'
200+ )
201+ email_message .template = 'template-id'
202+ email_message .metadata = {'key2' : 'value2' }
203+ actual = SparkPostMessage (email_message )
204+
205+ expected = dict (
206+ recipients = [
207+ {
208+ 'address' : 'to@example.com' ,
209+ 'metadata' : {
210+ 'key' : 'value'
211+ }
212+ }
213+ ],
214+ from_email = 'test@from.com' ,
215+ template = 'template-id' ,
216+ metadata = {'key2' : 'value2' }
217+ )
218+
219+ assert actual == expected
220+
221+
189222def test_substitution_data ():
190223 email_message = EmailMessage (
191224 to = [
192225 {
193- " address" : " to@example.com" ,
194- " substitution_data" : {
195- " key" : " value"
226+ ' address' : ' to@example.com' ,
227+ ' substitution_data' : {
228+ ' key' : ' value'
196229 }
197230 }
198231 ],
199232 from_email = 'test@from.com'
200233 )
201234 email_message .template = 'template-id'
202- email_message .substitution_data = {" key2" : " value2" }
235+ email_message .substitution_data = {' key2' : ' value2' }
203236 actual = SparkPostMessage (email_message )
204237
205238 expected = dict (
206239 recipients = [
207240 {
208- " address" : " to@example.com" ,
209- " substitution_data" : {
210- " key" : " value"
241+ ' address' : ' to@example.com' ,
242+ ' substitution_data' : {
243+ ' key' : ' value'
211244 }
212245 }
213246 ],
214247 from_email = 'test@from.com' ,
215248 template = 'template-id' ,
216- substitution_data = {" key2" : " value2" }
249+ substitution_data = {' key2' : ' value2' }
217250 )
218251
219252 assert actual == expected
0 commit comments