TRON -> TRON_MAINNET

api/billing/documents/internal/service/documents/service_test.go

@@ -3,8 +3,6 @@ package documents
 import (
 	"bytes"
 	"context"
-	"crypto/sha256"
-	"encoding/hex"
 	"testing"
 	"time"
 
@@ -100,14 +98,11 @@ func TestGetDocument_IdempotentAndHashed(t *testing.T) {
 		ExecutorFullName: "Jane Doe",
 		Amount:           decimal.RequireFromString("100.00"),
 		Currency:         "USD",
-		OrgLegalName:     "Acme Corp",
-		OrgAddress:       "42 Galaxy Way",
 	}
 
 	record := &model.DocumentRecord{
 		PaymentRef: "PAY-123",
 		Snapshot:   snapshot,
-		Available:  []model.DocumentType{model.DocumentTypeAct},
 	}
 
 	documentsStore := &stubDocumentsStore{record: record}
@@ -144,12 +139,15 @@ func TestGetDocument_IdempotentAndHashed(t *testing.T) {
 		t.Fatalf("expected content on first call")
 	}
 
-	hash1 := sha256.Sum256(resp1.Content)
 	stored := record.Hashes[model.DocumentTypeAct]
 	if stored == "" {
 		t.Fatalf("expected stored hash")
 	}
-	if stored != hex.EncodeToString(hash1[:]) {
+	footerHash := extractFooterHash(resp1.Content)
+	if footerHash == "" {
+		t.Fatalf("expected footer hash in PDF")
+	}
+	if stored != footerHash {
 		t.Fatalf("stored hash mismatch: got %s", stored)
 	}
 
@@ -174,3 +172,24 @@ func TestGetDocument_IdempotentAndHashed(t *testing.T) {
 		t.Fatalf("expected document load on second call")
 	}
 }
+
+func extractFooterHash(pdf []byte) string {
+	prefix := []byte("Document integrity hash: ")
+	idx := bytes.Index(pdf, prefix)
+	if idx == -1 {
+		return ""
+	}
+	start := idx + len(prefix)
+	end := start
+	for end < len(pdf) && isHexDigit(pdf[end]) {
+		end++
+	}
+	if end-start != 64 {
+		return ""
+	}
+	return string(pdf[start:end])
+}
+
+func isHexDigit(b byte) bool {
+	return (b >= '0' && b <= '9') || (b >= 'a' && b <= 'f') || (b >= 'A' && b <= 'F')
+}