diff --git a/pkg/models/balance_marshal_test.go b/pkg/models/balance_marshal_test.go index 8386847..a469965 100644 --- a/pkg/models/balance_marshal_test.go +++ b/pkg/models/balance_marshal_test.go @@ -14,11 +14,14 @@ func TestBalanceMarshalXML(t *testing.T) { } var buf strings.Builder + encoder := xml.NewEncoder(&buf) + err := balance.MarshalXML(encoder, xml.StartElement{Name: xml.Name{Local: "balance"}}) if err != nil { t.Fatalf("MarshalXML failed: %v", err) } + encoder.Flush() // Convert to string for easier testing @@ -46,11 +49,14 @@ func TestBalanceMarshalXML_PositiveValue(t *testing.T) { } var buf strings.Builder + encoder := xml.NewEncoder(&buf) + err := balance.MarshalXML(encoder, xml.StartElement{Name: xml.Name{Local: "balance"}}) if err != nil { t.Fatalf("MarshalXML failed: %v", err) } + encoder.Flush() xmlStr := buf.String() @@ -76,11 +82,14 @@ func TestBalanceMarshalXML_ZeroValue(t *testing.T) { } var buf strings.Builder + encoder := xml.NewEncoder(&buf) + err := balance.MarshalXML(encoder, xml.StartElement{Name: xml.Name{Local: "balance"}}) if err != nil { t.Fatalf("MarshalXML failed: %v", err) } + encoder.Flush() xmlStr := buf.String() @@ -106,11 +115,14 @@ func TestBalanceMarshalXML_ExtremeValues(t *testing.T) { } var buf strings.Builder + encoder := xml.NewEncoder(&buf) + err := balance.MarshalXML(encoder, xml.StartElement{Name: xml.Name{Local: "balance"}}) if err != nil { t.Fatalf("MarshalXML failed: %v", err) } + encoder.Flush() xmlStr := buf.String() diff --git a/pkg/models/bass_utilities_test.go b/pkg/models/bass_utilities_test.go index 23e4e3b..9dbd493 100644 --- a/pkg/models/bass_utilities_test.go +++ b/pkg/models/bass_utilities_test.go @@ -222,11 +222,14 @@ func TestBassMarshalXML(t *testing.T) { } var buf strings.Builder + encoder := xml.NewEncoder(&buf) + err := bass.MarshalXML(encoder, xml.StartElement{Name: xml.Name{Local: "bass"}}) if err != nil { t.Fatalf("MarshalXML failed: %v", err) } + encoder.Flush() // Convert to string for easier testing