email-mcp/internal/cli/integration_test.go

package cli

import (
	"bytes"
	"context"
	"testing"

	"email-mcp/internal/imapclient"
	"email-mcp/internal/mcpserver"
	"email-mcp/internal/secretstore"
	"email-mcp/internal/secretstore/kwallet"
)

type entrypointPromptStub struct {
	credential secretstore.Credential
}

func (p *entrypointPromptStub) PromptSetup(context.Context) (secretstore.Credential, error) {
	return p.credential, nil
}

type entrypointStoreStub struct {
	saveErr error
	loadErr error
}

func (s *entrypointStoreStub) Save(context.Context, string, secretstore.Credential) error {
	return s.saveErr
}

func (s *entrypointStoreStub) Load(context.Context, string) (secretstore.Credential, error) {
	return secretstore.Credential{}, s.loadErr
}

type entrypointMailServiceStub struct{}

func (entrypointMailServiceStub) ListMailboxes(context.Context, secretstore.Credential) ([]imapclient.Mailbox, error) {
	return nil, nil
}

func (entrypointMailServiceStub) ListMessages(context.Context, secretstore.Credential, string, int) ([]imapclient.MessageSummary, error) {
	return nil, nil
}

func (entrypointMailServiceStub) GetMessage(context.Context, secretstore.Credential, string, uint32) (imapclient.Message, error) {
	return imapclient.Message{}, nil
}

func TestExecuteSetupWritesWalletGuidanceAndReturnsExitCodeOne(t *testing.T) {
	app := NewAppWithDependencies(
		&entrypointPromptStub{
			credential: secretstore.Credential{
				Host:     "imap.example.com",
				Username: "alice",
				Password: "secret",
			},
		},
		&entrypointStoreStub{saveErr: kwallet.ErrKWalletUnavailable},
		nil,
		nil,
	)

	stderr := &bytes.Buffer{}
	if code := Execute(app, []string{"setup"}, stderr); code != 1 {
		t.Fatalf("expected exit code 1, got %d", code)
	}
	if got := stderr.String(); got != "kwallet is not available; make sure KDE Wallet is installed and your session D-Bus is running\n" {
		t.Fatalf("unexpected stderr: %q", got)
	}
}

func TestExecuteMCPWritesMissingCredentialGuidanceAndReturnsExitCodeOne(t *testing.T) {
	store := &entrypointStoreStub{loadErr: kwallet.ErrCredentialNotFound}
	mail := entrypointMailServiceStub{}
	runner := mcpserver.NewRunner(mcpserver.New(store, mail), nil, &bytes.Buffer{}, &bytes.Buffer{})
	app := NewAppWithDependencies(nil, store, runner, nil)

	stderr := &bytes.Buffer{}
	if code := Execute(app, []string{"mcp"}, stderr); code != 1 {
		t.Fatalf("expected exit code 1, got %d", code)
	}
	if got := stderr.String(); got != "credentials not configured; run `email-mcp setup`\n" {
		t.Fatalf("unexpected stderr: %q", got)
	}
}