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

Update SparkPostMessage to support campaign
SparkPost · Feb 1, 2017 · c874555 · c874555
2 parents 1ec1c40 + 9d56ffb
commit c874555
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/sparkpost/django/message.py b/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)
diff --git a/test/django/test_message.py b/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=[