billing docs service

api/billing/documents/renderer/renderer_test.go

@@ -0,0 +1,90 @@
+package renderer
+
+import (
+	"bytes"
+	"encoding/hex"
+	"strings"
+	"testing"
+	"unicode/utf16"
+)
+
+func TestRenderer_RenderContainsText(t *testing.T) {
+	blocks := []Block{
+		{Tag: TagTitle, Lines: []string{"ACT"}},
+		{Tag: TagText, Lines: []string{"Executor: Jane Doe", "Amount: 100 USD"}},
+	}
+
+	r := Renderer{
+		Issuer: Issuer{
+			LegalName:    "Sendico Ltd",
+			LegalAddress: "12 Market Street, London, UK",
+		},
+		OwnerPassword: "",
+	}
+
+	pdfBytes, err := r.Render(blocks, "deadbeef")
+	if err != nil {
+		t.Fatalf("Render: %v", err)
+	}
+	if len(pdfBytes) == 0 {
+		t.Fatalf("expected PDF bytes")
+	}
+
+	checks := []string{"Sendico Ltd", "Jane Doe", "100 USD", "Document integrity hash"}
+	for _, token := range checks {
+		if !containsPDFText(pdfBytes, token) {
+			t.Fatalf("expected PDF to contain %q", token)
+		}
+	}
+}
+
+func containsPDFText(pdfBytes []byte, text string) bool {
+	if bytes.Contains(pdfBytes, []byte(text)) {
+		return true
+	}
+	hexText := hex.EncodeToString([]byte(text))
+	if bytes.Contains(pdfBytes, []byte(strings.ToUpper(hexText))) {
+		return true
+	}
+	if bytes.Contains(pdfBytes, []byte(strings.ToLower(hexText))) {
+		return true
+	}
+
+	utf16Bytes := encodeUTF16BE(text, false)
+	if bytes.Contains(pdfBytes, utf16Bytes) {
+		return true
+	}
+	utf16Hex := hex.EncodeToString(utf16Bytes)
+	if bytes.Contains(pdfBytes, []byte(strings.ToUpper(utf16Hex))) {
+		return true
+	}
+	if bytes.Contains(pdfBytes, []byte(strings.ToLower(utf16Hex))) {
+		return true
+	}
+
+	utf16BytesBOM := encodeUTF16BE(text, true)
+	if bytes.Contains(pdfBytes, utf16BytesBOM) {
+		return true
+	}
+	utf16HexBOM := hex.EncodeToString(utf16BytesBOM)
+	if bytes.Contains(pdfBytes, []byte(strings.ToUpper(utf16HexBOM))) {
+		return true
+	}
+	return bytes.Contains(pdfBytes, []byte(strings.ToLower(utf16HexBOM)))
+}
+
+func encodeUTF16BE(text string, withBOM bool) []byte {
+	encoded := utf16.Encode([]rune(text))
+	length := len(encoded) * 2
+	if withBOM {
+		length += 2
+	}
+	out := make([]byte, 0, length)
+	if withBOM {
+		out = append(out, 0xFE, 0xFF)
+	}
+	for _, v := range encoded {
+		out = append(out, byte(v>>8), byte(v))
+	}
+	return out
+}