Merge pull request #5 from foca/custom-headers

Allow configuring the email envelope

lib/malone.rb

@@ -30,6 +30,7 @@ def initialize(config)
 
   def deliver(dict)
     mail = envelope(dict)
+    yield mail if block_given?
 
     smtp = Net::SMTP.new(config.host, config.port)
     smtp.enable_starttls_auto if config.tls

test/malone.rb

@@ -186,6 +186,15 @@ def SMTP.new(host, port)
     assert $smtp[:finish]
   end
 
+  test "adding custom headers" do |m|
+    m.deliver(to: "[email protected]", from: "[email protected]",
+              subject: "Happy new year!", text: "TEXT") do |mail|
+      mail.add_header("X-MC-SendAt", "2016-01-01 00:00:00")
+    end
+
+    assert $smtp[:blob].include?("X-MC-SendAt: 2016-01-01 00:00:00")
+  end
+
   test "calls #finish even when it fails during send_message" do |m|
     class FakeSMTP
       def send_message(*args)