moving

packages/agent/tests/connectors/email-connector.test.ts

@@ -0,0 +1,144 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { EmailConnector } from '../../src/connectors/email-connector.js';
+import type { VaultStore } from '@waggle/core';
+
+function createMockVault(cred?: { value: string }): VaultStore {
+  return {
+    getConnectorCredential: vi.fn((id: string) => {
+      if (id === 'email' && cred) return { ...cred, type: 'api_key', isExpired: false };
+      return null;
+    }),
+    get: vi.fn(() => null),
+    set: vi.fn(),
+    delete: vi.fn(),
+    list: vi.fn(() => []),
+    has: vi.fn(() => false),
+    setConnectorCredential: vi.fn(),
+    migrateFromConfig: vi.fn(() => 0),
+  } as unknown as VaultStore;
+}
+
+describe('EmailConnector', () => {
+  let connector: EmailConnector;
+  let originalFetch: typeof globalThis.fetch;
+
+  beforeEach(() => {
+    connector = new EmailConnector();
+    originalFetch = globalThis.fetch;
+  });
+
+  afterEach(() => {
+    globalThis.fetch = originalFetch;
+  });
+
+  it('implements WaggleConnector interface', () => {
+    expect(connector.id).toBe('email');
+    expect(connector.name).toBe('Email (SendGrid)');
+    expect(connector.authType).toBe('api_key');
+    expect(connector.actions).toHaveLength(3);
+  });
+
+  it('connect() retrieves API key from vault', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+    expect(vault.getConnectorCredential).toHaveBeenCalledWith('email');
+  });
+
+  it('healthCheck() validates API key', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    globalThis.fetch = vi.fn().mockResolvedValue({
+      ok: true, json: async () => ({ username: 'waggle' }),
+    }) as unknown as typeof fetch;
+
+    const health = await connector.healthCheck();
+    expect(health.status).toBe('connected');
+  });
+
+  it('execute(send_email) sends email (mocked)', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    globalThis.fetch = vi.fn().mockResolvedValue({
+      status: 202, ok: true, headers: new Map([['X-Message-Id', 'msg-123']]),
+      text: async () => '',
+    }) as unknown as typeof fetch;
+
+    const result = await connector.execute('send_email', {
+      to: 'user@example.com',
+      subject: 'Test',
+      body: 'Hello from Waggle!',
+    });
+    expect(result.success).toBe(true);
+    expect((result.data as Record<string, unknown>).sent).toBe(true);
+  });
+
+  it('execute(send_email) requires to, subject, body params', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    // Missing 'to' — the connector will still call the API but SendGrid would reject
+    // The connector trusts the agent to provide required params per inputSchema
+    globalThis.fetch = vi.fn().mockResolvedValue({
+      status: 400, ok: false, text: async () => 'Missing to',
+    }) as unknown as typeof fetch;
+
+    const result = await connector.execute('send_email', { subject: 'Test', body: 'Hello' });
+    expect(result.success).toBe(false);
+  });
+
+  it('execute(send_template) sends template email', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    globalThis.fetch = vi.fn().mockResolvedValue({
+      status: 202, ok: true, headers: new Map(),
+      text: async () => '',
+    }) as unknown as typeof fetch;
+
+    const result = await connector.execute('send_template', {
+      to: 'user@example.com',
+      template_id: 'd-abc123',
+      variables: { name: 'Test User' },
+    });
+    expect(result.success).toBe(true);
+  });
+
+  it('execute(check_delivery) returns delivery status', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    globalThis.fetch = vi.fn().mockResolvedValue({
+      ok: true, json: async () => ({ status: 'delivered', events: [] }),
+    }) as unknown as typeof fetch;
+
+    const result = await connector.execute('check_delivery', { message_id: 'msg-123' });
+    expect(result.success).toBe(true);
+    expect((result.data as Record<string, unknown>).status).toBe('delivered');
+  });
+
+  it('rate limiter rejects after max daily sends', async () => {
+    const vault = createMockVault({ value: 'SG.test_key' });
+    await connector.connect(vault);
+
+    // Artificially set send count to max (reach into private rate-limit state)
+    const rateState = connector as unknown as { dailySendCount: number; dailyResetDate: string };
+    rateState.dailySendCount = 100;
+    rateState.dailyResetDate = new Date().toISOString().slice(0, 10);
+
+    const result = await connector.execute('send_email', {
+      to: 'user@example.com', subject: 'Test', body: 'Hello',
+    });
+    expect(result.success).toBe(false);
+    expect(result.error).toContain('Daily email limit');
+  });
+
+  it('all send actions are riskLevel high', () => {
+    const sendActions = connector.actions.filter(a => a.name.startsWith('send'));
+    expect(sendActions).toHaveLength(2);
+    for (const action of sendActions) {
+      expect(action.riskLevel).toBe('high');
+    }
+  });
+});