jackwener--opencli/clis/notebooklm/source-guide.test.js

import { beforeEach, describe, expect, it, vi } from 'vitest';
const { mockListNotebooklmSourcesViaRpc, mockListNotebooklmSourcesFromPage, mockGetNotebooklmSourceGuideViaRpc, mockGetNotebooklmPageState, mockRequireNotebooklmSession, } = vi.hoisted(() => ({
    mockListNotebooklmSourcesViaRpc: vi.fn(),
    mockListNotebooklmSourcesFromPage: vi.fn(),
    mockGetNotebooklmSourceGuideViaRpc: vi.fn(),
    mockGetNotebooklmPageState: vi.fn(),
    mockRequireNotebooklmSession: vi.fn(),
}));
vi.mock('./utils.js', async () => {
    const actual = await vi.importActual('./utils.js');
    return {
        ...actual,
        listNotebooklmSourcesViaRpc: mockListNotebooklmSourcesViaRpc,
        listNotebooklmSourcesFromPage: mockListNotebooklmSourcesFromPage,
        getNotebooklmSourceGuideViaRpc: mockGetNotebooklmSourceGuideViaRpc,
        getNotebooklmPageState: mockGetNotebooklmPageState,
        requireNotebooklmSession: mockRequireNotebooklmSession,
    };
});
import { getRegistry } from '@jackwener/opencli/registry';
import './source-guide.js';
describe('notebooklm source-guide', () => {
    const command = getRegistry().get('notebooklm/source-guide');
    beforeEach(() => {
        mockListNotebooklmSourcesViaRpc.mockReset();
        mockListNotebooklmSourcesFromPage.mockReset();
        mockGetNotebooklmSourceGuideViaRpc.mockReset();
        mockGetNotebooklmPageState.mockReset();
        mockRequireNotebooklmSession.mockReset();
        mockRequireNotebooklmSession.mockResolvedValue(undefined);
        mockGetNotebooklmPageState.mockResolvedValue({
            url: 'https://notebooklm.google.com/notebook/nb-demo',
            title: 'Browser Automation',
            hostname: 'notebooklm.google.com',
            kind: 'notebook',
            notebookId: 'nb-demo',
            loginRequired: false,
            notebookCount: 1,
        });
    });
    it('returns source guide for a source matched from rpc source rows', async () => {
        mockListNotebooklmSourcesViaRpc.mockResolvedValue([
            {
                id: 'src-yt',
                notebook_id: 'nb-demo',
                title: 'Video Source',
                url: 'https://notebooklm.google.com/notebook/nb-demo',
                source: 'rpc',
                type: 'youtube',
                type_code: 9,
            },
        ]);
        mockGetNotebooklmSourceGuideViaRpc.mockResolvedValue({
            source_id: 'src-yt',
            notebook_id: 'nb-demo',
            title: 'Video Source',
            type: 'youtube',
            summary: 'Guide summary.',
            keywords: ['AI', 'agents'],
            source: 'rpc',
        });
        const result = await command.func({}, { source: 'src-yt' });
        expect(result).toEqual([
            {
                source_id: 'src-yt',
                notebook_id: 'nb-demo',
                title: 'Video Source',
                type: 'youtube',
                summary: 'Guide summary.',
                keywords: ['AI', 'agents'],
                source: 'rpc',
            },
        ]);
    });
    it('matches by title from dom rows when rpc source list is unavailable', async () => {
        mockListNotebooklmSourcesViaRpc.mockResolvedValue([]);
        mockListNotebooklmSourcesFromPage.mockResolvedValue([
            {
                id: 'src-1',
                notebook_id: 'nb-demo',
                title: 'Example Source',
                url: 'https://notebooklm.google.com/notebook/nb-demo',
                source: 'current-page',
            },
        ]);
        mockGetNotebooklmSourceGuideViaRpc.mockResolvedValue({
            source_id: 'src-1',
            notebook_id: 'nb-demo',
            title: 'Example Source',
            type: null,
            summary: 'Guide summary.',
            keywords: ['topic'],
            source: 'rpc',
        });
        const result = await command.func({}, { source: 'example source' });
        expect(result).toEqual([
            expect.objectContaining({
                source_id: 'src-1',
                title: 'Example Source',
                summary: 'Guide summary.',
            }),
        ]);
    });
});