add document code

src/main/java/org/mustangproject/ZUGFeRD/IZUGFeRDExportableTransaction.java

@@ -27,10 +27,21 @@ package org.mustangproject.ZUGFeRD;
  * */
 
 
+import org.mustangproject.ZUGFeRD.model.DocumentCodeTypeConstants;
+
 import java.util.Date;
 
 public interface IZUGFeRDExportableTransaction {
 
+	/**
+	 *
+	 *
+	 * @return Code of Document
+	 */
+	default String getDocumentCode() {
+		return DocumentCodeTypeConstants.INVOICE;
+	}
+
 	/**
 	 * Number, typically invoice number of the invoice
 	 *

src/main/java/org/mustangproject/ZUGFeRD/ZUGFeRDImporter.java

@@ -209,6 +209,13 @@ public class ZUGFeRDImporter {
 		return result;
 	}
 
+	/**
+	 * @return the document code
+	 */
+	public String getDocumentCode() {
+		return extractString("//HeaderExchangedDocument/TypeCode");
+	}
+
 	/**
 	 * @return the sender's bank's BLZ code
 	 */

src/main/java/org/mustangproject/ZUGFeRD/ZUGFeRDTransactionModelConverter.java

@@ -94,7 +94,7 @@ class ZUGFeRDTransactionModelConverter {
 		document.setIssueDateTime(issueDateTime);
 
 		DocumentCodeType documentCodeType = xmlFactory.createDocumentCodeType();
-		documentCodeType.setValue(DocumentCodeTypeConstants.INVOICE);
+		documentCodeType.setValue(trans.getDocumentCode());
 		document.setTypeCode(documentCodeType);
 
 		TextType name = xmlFactory.createTextType();

src/test/java/org/mustangproject/ZUGFeRD/MustangReaderWriterEdgeTest.java

@@ -348,6 +348,7 @@ public class MustangReaderWriterEdgeTest extends TestCase implements IZUGFeRDExp
 		assertEquals(zi.getHolder(), getOwnOrganisationName());
 		assertEquals(zi.getDueDate(), "20170530");
 		assertEquals(zi.getForeignReference(), getNumber());
+		assertEquals(zi.getDocumentCode(), "380");
 
 	}
 

src/test/java/org/mustangproject/ZUGFeRD/MustangReaderWriterTest.java

@@ -350,32 +350,13 @@ public class MustangReaderWriterTest extends TestCase implements IZUGFeRDExporta
 		ZUGFeRDImporter zi = new ZUGFeRDImporter(inputStream);
 
 		// Reading ZUGFeRD
-		String amount = null;
-		String blz = null;
-		String bic = null;
-		String iban = null;
-		String kto = null;
-		String holder = null;
-		String ref = null;
-
-		if (zi.canParse()) {
-			amount = zi.getAmount();
-			blz = zi.getBLZ();
-			bic = zi.getBIC();
-			iban = zi.getIBAN();
-			kto = zi.getKTO();
-			holder = zi.getHolder();
-			ref = zi.getForeignReference();
-		}
-		// this resembles the data written in MustangReaderWriterCustomXMLTest
-		assertEquals(amount, "571.04");
-		assertEquals(blz, "41441604");
-		assertEquals(bic, "COBADEFFXXX");
-		assertEquals(iban, "DE88 2008 0000 0970 3757 00");
-		assertEquals(kto, "44421800");
-		assertEquals(holder, "Bei Spiel GmbH");
-		assertEquals(ref, "RE-20170509/505");
-
+		assertEquals(zi.getAmount(), "571.04");
+		assertEquals(zi.getBLZ(), getOwnBLZ());
+		assertEquals(zi.getBIC(), getOwnBIC());
+		assertEquals(zi.getIBAN(), getOwnIBAN());
+		assertEquals(zi.getKTO(), getOwnKto());
+		assertEquals(zi.getHolder(), getOwnOrganisationName());
+		assertEquals(zi.getForeignReference(), "RE-20170509/505");
 	}
 
 	public void testForeignImport() throws IOException {