Merge pull request #137 from andyzickler/issue-136-django-campaign

Update SparkPostMessage to support campaign

Author: richleland

sparkpost/django/message.py

@@ -87,4 +87,7 @@ def __init__(self, message):
         if hasattr(message, 'substitution_data'):
             formatted['substitution_data'] = message.substitution_data
 
+        if hasattr(message, 'campaign'):
+            formatted['campaign'] = message.campaign
+
         super(SparkPostMessage, self).__init__(formatted)

test/django/test_message.py

@@ -175,6 +175,17 @@ def test_template():
     assert actual == expected
 
 
+def test_campaign():
+    email_message = EmailMessage(**base_options)
+    email_message.campaign = 'campaign-id'
+    actual = SparkPostMessage(email_message)
+    expected = dict(
+        campaign='campaign-id'
+    )
+    expected.update(base_expected)
+    assert actual == expected
+
+
 def test_substitution_data():
     email_message = EmailMessage(
         to=[