use bytes join

viamrobotics · Nov 21, 2024 · a09687c · a09687c
1 parent 4440034
commit a09687c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/app/billing_client_test.go b/app/billing_client_test.go
@@ -1,6 +1,7 @@
 package app
 
 import (
+	"bytes"
 	"context"
 	"io"
 	"testing"
@@ -224,10 +225,7 @@ func TestBillingClient(t *testing.T) {
 	})
 
 	t.Run("GetInvoicePDF", func(t *testing.T) {
-		var expectedData []byte
-		expectedData = append(expectedData, chunk1...)
-		expectedData = append(expectedData, chunk2...)
-		expectedData = append(expectedData, chunk3...)
+		expectedData := bytes.Join(chunks, nil)
 		var count int
 		mockStream := &inject.BillingServiceGetInvoicePdfClient{
 			RecvFunc: func() (*pb.GetInvoicePdfResponse, error) {