build bundle control records when creating files

moov-io · Oct 23, 2024 · 71355ab · 71355ab
1 parent eaa50fc
commit 71355ab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/file.go b/file.go
@@ -253,6 +253,10 @@ func (f *File) Create() error {
 
 				fileTotalAmount = fileTotalAmount + rd.ItemAmount
 			}
+
+			if err := b.build(); err != nil {
+				return fmt.Errorf("building control record for bundle %s: %w", b.BundleHeader.BundleSequenceNumber, err)
+			}
 		}
 	}
 

diff --git a/internal/files/files_test.go b/internal/files/files_test.go
@@ -16,15 +16,12 @@ import (
 	"strings"
 	"testing"
 
-	"github.com/stretchr/testify/assert"
-
-	"github.com/stretchr/testify/require"
-
-	"github.com/moov-io/base"
-	"github.com/moov-io/imagecashletter"
-
 	"github.com/gorilla/mux"
+	"github.com/moov-io/base"
 	"github.com/moov-io/base/log"
+	"github.com/moov-io/imagecashletter"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestFileId(t *testing.T) {