gogcli/internal/cmd/gmail_thread_helpers_test.go

package cmd

import (
	"context"
	"encoding/base64"
	"os"
	"path/filepath"
	"testing"

	"google.golang.org/api/gmail/v1"
)

func TestStripHTMLTags_More(t *testing.T) {
	input := "<div>Hello <b>World</b><script>bad()</script><style>.x{}</style></div>"
	out := stripHTMLTags(input)
	if out != "Hello World" {
		t.Fatalf("unexpected stripped output: %q", out)
	}
}

func TestFormatBytes(t *testing.T) {
	if got := formatBytes(500); got != "500 B" {
		t.Fatalf("unexpected bytes format: %q", got)
	}
	if got := formatBytes(2048); got != "2.0 KB" {
		t.Fatalf("unexpected KB format: %q", got)
	}
	if got := formatBytes(5 * 1024 * 1024); got != "5.0 MB" {
		t.Fatalf("unexpected MB format: %q", got)
	}
	if got := formatBytes(3 * 1024 * 1024 * 1024); got != "3.0 GB" {
		t.Fatalf("unexpected GB format: %q", got)
	}
}

func TestCollectAttachments_More(t *testing.T) {
	part := &gmail.MessagePart{
		Parts: []*gmail.MessagePart{
			{
				Filename: "file.txt",
				MimeType: "text/plain",
				Body: &gmail.MessagePartBody{
					AttachmentId: "a1",
					Size:         12,
				},
			},
			{
				Parts: []*gmail.MessagePart{
					{
						MimeType: "image/png",
						Body: &gmail.MessagePartBody{
							AttachmentId: "a2",
							Size:         34,
						},
					},
				},
			},
		},
	}
	attachments := collectAttachments(part)
	if len(attachments) != 2 {
		t.Fatalf("expected 2 attachments, got %d", len(attachments))
	}
	if attachments[0].Filename != "file.txt" || attachments[1].AttachmentID != "a2" {
		t.Fatalf("unexpected attachments: %#v", attachments)
	}
}

func TestBestBodySelection(t *testing.T) {
	plain := base64.RawURLEncoding.EncodeToString([]byte("plain"))
	html := base64.RawURLEncoding.EncodeToString([]byte("<b>html</b>"))
	part := &gmail.MessagePart{
		Parts: []*gmail.MessagePart{
			{
				MimeType: "text/plain",
				Body:     &gmail.MessagePartBody{Data: plain},
			},
			{
				MimeType: "text/html",
				Body:     &gmail.MessagePartBody{Data: html},
			},
		},
	}
	if got := bestBodyText(part); got != "plain" {
		t.Fatalf("unexpected best body text: %q", got)
	}
	body, isHTML := bestBodyForDisplay(part)
	if body != "plain" || isHTML {
		t.Fatalf("unexpected body display: %q html=%v", body, isHTML)
	}
}

func TestFindPartBodyHTML(t *testing.T) {
	html := base64.RawURLEncoding.EncodeToString([]byte("<p>hi</p>"))
	part := &gmail.MessagePart{
		MimeType: "multipart/alternative",
		Parts: []*gmail.MessagePart{
			{
				MimeType: "text/html; charset=UTF-8",
				Body:     &gmail.MessagePartBody{Data: html},
			},
		},
	}
	got := findPartBody(part, "text/html")
	if got != "<p>hi</p>" {
		t.Fatalf("unexpected html body: %q", got)
	}
}

func TestMimeTypeMatches(t *testing.T) {
	if !mimeTypeMatches("Text/Plain; charset=UTF-8", "text/plain") {
		t.Fatalf("expected mime match")
	}
	if mimeTypeMatches("application/json", "text/plain") {
		t.Fatalf("unexpected mime match")
	}
	if normalizeMimeType("text/plain; charset=utf-8") != "text/plain" {
		t.Fatalf("unexpected normalized mime type")
	}
	if normalizeMimeType("") != "" {
		t.Fatalf("expected empty normalized mime type")
	}
}

func TestDecodeBase64URL_Padded(t *testing.T) {
	encoded := base64.URLEncoding.EncodeToString([]byte("hello"))
	decoded, err := decodeBase64URL(encoded)
	if err != nil {
		t.Fatalf("decodeBase64URL: %v", err)
	}
	if decoded != "hello" {
		t.Fatalf("unexpected decode: %q", decoded)
	}
}

func TestDownloadAttachment_Cached(t *testing.T) {
	dir := t.TempDir()
	messageID := "msg1"
	attachmentID := "att123456"
	filename := "file.txt"
	shortID := attachmentID[:8]
	outPath := filepath.Join(dir, messageID+"_"+shortID+"_"+filename)

	if err := os.WriteFile(outPath, []byte("abc"), 0o600); err != nil {
		t.Fatalf("write file: %v", err)
	}

	info := attachmentInfo{
		Filename:     filename,
		AttachmentID: attachmentID,
		Size:         3,
	}
	gotPath, cached, err := downloadAttachment(context.Background(), nil, messageID, info, dir)
	if err != nil {
		t.Fatalf("downloadAttachment: %v", err)
	}
	if !cached || gotPath != outPath {
		t.Fatalf("expected cached path %q, got %q cached=%v", outPath, gotPath, cached)
	}
}