@@ -186,34 +186,67 @@ def test_campaign():
186
186
assert actual == expected
187
187
188
188
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
+
189
222
def test_substitution_data ():
190
223
email_message = EmailMessage (
191
224
to = [
192
225
{
193
- " address" : " to@example.com" ,
194
- " substitution_data" : {
195
- " key" : " value"
226
+ ' address' : ' to@example.com' ,
227
+ ' substitution_data' : {
228
+ ' key' : ' value'
196
229
}
197
230
}
198
231
],
199
232
from_email = 'test@from.com'
200
233
)
201
234
email_message .template = 'template-id'
202
- email_message .substitution_data = {" key2" : " value2" }
235
+ email_message .substitution_data = {' key2' : ' value2' }
203
236
actual = SparkPostMessage (email_message )
204
237
205
238
expected = dict (
206
239
recipients = [
207
240
{
208
- " address" : " to@example.com" ,
209
- " substitution_data" : {
210
- " key" : " value"
241
+ ' address' : ' to@example.com' ,
242
+ ' substitution_data' : {
243
+ ' key' : ' value'
211
244
}
212
245
}
213
246
],
214
247
from_email = 'test@from.com' ,
215
248
template = 'template-id' ,
216
- substitution_data = {" key2" : " value2" }
249
+ substitution_data = {' key2' : ' value2' }
217
250
)
218
251
219
252
assert actual == expected
0 commit comments