#176 but only the if block

SparkPost · Aug 10, 2018 · a55235b · a55235b
1 parent e4cf55d
commit a55235b
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 8 deletions.
diff --git a/sparkpost/django/message.py b/sparkpost/django/message.py
@@ -90,6 +90,9 @@ def __init__(self, message):
         if hasattr(message, 'campaign'):
             formatted['campaign'] = message.campaign
 
+        if hasattr(message, 'metadata'):
+            formatted['metadata'] = message.metadata
+
         if message.extra_headers:
             formatted['custom_headers'] = message.extra_headers
             if 'X-MSYS-API' in message.extra_headers:

diff --git a/test/django/test_message.py b/test/django/test_message.py
@@ -186,34 +186,67 @@ def test_campaign():
     assert actual == expected
 
 
+def test_metadata():
+    email_message = EmailMessage(
+        to=[
+            {
+                'address': 'to@example.com',
+                'metadata': {
+                    'key': 'value'
+                }
+            }
+        ],
+        from_email='test@from.com'
+    )
+    email_message.template = 'template-id'
+    email_message.metadata = {'key2': 'value2'}
+    actual = SparkPostMessage(email_message)
+
+    expected = dict(
+        recipients=[
+            {
+                'address': 'to@example.com',
+                'metadata': {
+                    'key': 'value'
+                }
+            }
+        ],
+        from_email='test@from.com',
+        template='template-id',
+        metadata={'key2': 'value2'}
+    )
+
+    assert actual == expected
+
+
 def test_substitution_data():
     email_message = EmailMessage(
         to=[
             {
-                "address": "to@example.com",
-                "substitution_data": {
-                    "key": "value"
+                'address': 'to@example.com',
+                'substitution_data': {
+                    'key': 'value'
                 }
             }
         ],
         from_email='test@from.com'
     )
     email_message.template = 'template-id'
-    email_message.substitution_data = {"key2": "value2"}
+    email_message.substitution_data = {'key2': 'value2'}
     actual = SparkPostMessage(email_message)
 
     expected = dict(
         recipients=[
             {
-                "address": "to@example.com",
-                "substitution_data": {
-                    "key": "value"
+                'address': 'to@example.com',
+                'substitution_data': {
+                    'key': 'value'
                 }
             }
         ],
         from_email='test@from.com',
         template='template-id',
-        substitution_data={"key2": "value2"}
+        substitution_data={'key2': 'value2'}
     )
 
     assert actual == expected