85 lines
2.5 KiB
Go
85 lines
2.5 KiB
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)
|
|
}
|
|
}
|