Add support for notification headers

Author: laurilehmijoki

src/main/java/de/bitzeche/video/transcoding/zencoder/job/ZencoderNotification.java

@@ -27,10 +27,13 @@
 
 import de.bitzeche.video.transcoding.zencoder.util.XmlUtility;
 
+import java.util.Map;
+
 public class ZencoderNotification {
 
 	private String notificationString;
 	private String format;
+	private Map<String, String> headers;
 
 	public ZencoderNotification(String notificationDestination) {
 		this.notificationString = notificationDestination;
@@ -45,6 +48,16 @@ public Element createXML(Document document) {
 			root.appendChild(nfNode);
 		}
 
+        if (this.headers != null) {
+            Node headersNode = document.createElement("headers");
+            root.appendChild(headersNode);
+            for (Map.Entry<String, String> headerEntry : headers.entrySet()) {
+                Node header = document.createElement(headerEntry.getKey());
+                header.setTextContent(headerEntry.getValue());
+                headersNode.appendChild(header);
+            }
+        }
+
 		Node urlNode = document.createElement("url");
 		urlNode.setTextContent(this.notificationString);
 		root.appendChild(urlNode);
@@ -60,6 +73,14 @@ public void setFormat(String format) {
 		this.format = format;
 	}
 
+    public void setHeaders(Map<String, String> headers) {
+        this.headers = headers;
+    }
+
+    public Map<String, String> getHeaders() {
+        return headers;
+    }
+
 	public String toString() {
 		Element elem;
 		Document document;
@@ -84,5 +105,4 @@ public String toString() {
 		}
 		return this.getClass().getSimpleName();
 	}
-
 }

src/test/java/de/bitzeche/video/transcoding/zencoder/test/ZencoderNotificationTest.java

@@ -23,6 +23,8 @@
 
 import de.bitzeche.video.transcoding.zencoder.job.ZencoderNotification;
 
+import java.util.HashMap;
+
 public class ZencoderNotificationTest {
 
 	@Test
@@ -42,4 +44,16 @@ public void testWithOptions() throws ParserConfigurationException {
 		// System.out.println(doc);
 		Assert.assertEquals(doc, expected);
 	}
+
+    @Test
+    public void testWithHeaders() throws ParserConfigurationException {
+        ZencoderNotification notif = new ZencoderNotification("[email protected]");
+        notif.setHeaders(new HashMap<String, String>(){{
+            put("cloudfront_url", "http://asdjaosidjas.cloudfront.net");
+        }});
+        String doc = StringUtil.stripSpacesAndLineBreaksFrom(notif.toString());
+        String expected = "<?xmlversion=\"1.0\"encoding=\"UTF-8\"?><notification><headers><cloudfront_url>http://asdjaosidjas.cloudfront.net</cloudfront_url></headers><url>[email protected]</url></notification>";
+        // System.out.println(doc);
+        Assert.assertEquals(doc, expected);
+    }
 }