more testing

1 files changed, 12 insertions, 1 deletions

diff --git a/reader/reader_test.go b/reader/reader_test.go
index 44c2a56..ba3172e 100644
--- a/reader/reader_test.go
+++ b/reader/reader_test.go
@@ -7,7 +7,18 @@ import (
 func TestReadDML(t *testing.T) {
 	d, _ := ReadDML("/home/lemon/Documents/datamaps/input/datamap.csv")
 	dmlData := *d
+	// Test Key values
 	if dmlData[0].Key != "Project/Programme Name" {
-		t.Errorf("dmlData[0].key = %s; want Project/Programme Name", dmlData[0].Key)
+		t.Errorf("dmlData[0].Key = %s; want Project/Programme Name", dmlData[0].Key)
+	}
+	if dmlData[1].Key != "Department" {
+		t.Errorf("dmlData[1].Key = %s; want Department (without a space)", dmlData[1].Key)
+	}
+	if dmlData[2].Key != "Delivery Body" {
+		t.Errorf("dmlData[2].Key = %s; want Delivery Body (without a space)", dmlData[2].Key)
+	}
+	// Test Sheet values
+	if dmlData[0].Sheet != "Introduction" {
+		t.Errorf("dmlData[0].Sheet = %s; want Introduction", dmlData[0].Key)
 	}
 }