Increase Test Code Coverage for batches

Increase Test Code Coverage for batchCCD, batchCTX, batchENR, batchMTE
adamdecaf · Nov 26, 2018 · b85fccd · b85fccd
1 parent f9471f7
commit b85fccd
Show file tree

Hide file tree

Showing 4 changed files with 128 additions and 0 deletions.
diff --git a/batchCCD_test.go b/batchCCD_test.go
@@ -303,3 +303,35 @@ func BenchmarkBatchCCDReceivingCompanyField(b *testing.B) {
 		testBatchCCDReceivingCompanyField(b)
 	}
 }
+
+// TestBatchCCDValidTranCodeForServiceClassCode validates a transactionCode based on ServiceClassCode
+func TestBatchCCDValidTranCodeForServiceClassCode(t *testing.T) {
+	mockBatch := mockBatchCCD()
+	mockBatch.GetHeader().ServiceClassCode = 220
+	if err := mockBatch.Create(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "TransactionCode" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
+
+// TestBatchCCDAddenda02 validates BatchCCD cannot have Addenda02
+func TestBatchCCDAddenda02(t *testing.T) {
+	mockBatch := mockBatchCCD()
+	mockBatch.Entries[0].AddendaRecordIndicator = 1
+	mockBatch.GetEntries()[0].Addenda02 = mockAddenda02()
+	mockBatch.Create()
+	if err := mockBatch.Validate(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "Addenda02" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
diff --git a/batchCTX_test.go b/batchCTX_test.go
@@ -585,3 +585,35 @@ func TestBatchCTXAddendum99(t *testing.T) {
 		}
 	}
 }
+
+// TestBatchCTXValidTranCodeForServiceClassCode validates a transactionCode based on ServiceClassCode
+func TestBatchCTXValidTranCodeForServiceClassCode(t *testing.T) {
+	mockBatch := mockBatchCTX()
+	mockBatch.GetHeader().ServiceClassCode = 225
+	if err := mockBatch.Create(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "TransactionCode" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
+
+// TestBatchCTXAddenda02 validates BatchCTX cannot have Addenda02
+func TestBatchCTXAddenda02(t *testing.T) {
+	mockBatch := mockBatchCTX()
+	mockBatch.Entries[0].AddendaRecordIndicator = 1
+	mockBatch.GetEntries()[0].Addenda02 = mockAddenda02()
+	mockBatch.Create()
+	if err := mockBatch.Validate(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "Addenda02" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
diff --git a/batchENR_test.go b/batchENR_test.go
@@ -320,3 +320,35 @@ func TestBatchENR__PaymentInformation(t *testing.T) {
 		t.Errorf("EnrolleeClassificationCode: %d", v)
 	}
 }
+
+// TestBatchENRValidTranCodeForServiceClassCode validates a transactionCode based on ServiceClassCode
+func TestBatchENRValidTranCodeForServiceClassCode(t *testing.T) {
+	mockBatch := mockBatchENR()
+	mockBatch.GetHeader().ServiceClassCode = 225
+	if err := mockBatch.Create(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "TransactionCode" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
+
+// TestBatchENRAddenda02 validates BatchENR cannot have Addenda02
+func TestBatchENRAddenda02(t *testing.T) {
+	mockBatch := mockBatchENR()
+	mockBatch.Entries[0].AddendaRecordIndicator = 1
+	mockBatch.GetEntries()[0].Addenda02 = mockAddenda02()
+	mockBatch.Create()
+	if err := mockBatch.Validate(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "Addenda02" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
diff --git a/batchMTE_test.go b/batchMTE_test.go
@@ -318,3 +318,35 @@ func TestBatchMTEIdentificationNumber(t *testing.T) {
 		}
 	}
 }
+
+// TestBatchMTEValidTranCodeForServiceClassCode validates a transactionCode based on ServiceClassCode
+func TestBatchMTEValidTranCodeForServiceClassCode(t *testing.T) {
+	mockBatch := mockBatchMTE()
+	mockBatch.GetHeader().ServiceClassCode = 220
+	if err := mockBatch.Create(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "TransactionCode" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}
+
+// TestBatchMTEAddenda05 validates BatchMTE cannot have Addenda05
+func TestBatchMTEAddenda05(t *testing.T) {
+	mockBatch := mockBatchMTE()
+	mockBatch.Entries[0].AddendaRecordIndicator = 1
+	mockBatch.GetEntries()[0].AddAddenda05(mockAddenda05())
+	mockBatch.Create()
+	if err := mockBatch.Validate(); err != nil {
+		if e, ok := err.(*BatchError); ok {
+			if e.FieldName != "Addenda05" {
+				t.Errorf("%T: %s", err, err)
+			}
+		} else {
+			t.Errorf("%T: %s", err, err)
+		}
+	}
+}