mirror of
https://github.com/TriliumNext/Notes.git
synced 2025-07-29 02:52:27 +08:00
142 lines
4.7 KiB
TypeScript
142 lines
4.7 KiB
TypeScript
import importSingle from '../../../src/services/import/single.js';
|
|
import importUtils from '../../../src/services/import/utils.js';
|
|
import BNote from '../../../src/becca/entities/bnote.js';
|
|
import TaskContext from '../../../src/services/task_context.js';
|
|
|
|
describe('HTML Import', () => {
|
|
let parentNote: BNote;
|
|
let taskContext: TaskContext;
|
|
|
|
beforeEach(() => {
|
|
// Create a mock parent note
|
|
parentNote = new BNote({
|
|
noteId: 'testParent',
|
|
title: 'Test Parent',
|
|
type: 'text',
|
|
mime: 'text/html'
|
|
});
|
|
|
|
// Create a mock task context
|
|
taskContext = new TaskContext('test', 'test');
|
|
// Set textImportedAsText to true to ensure HTML imports are processed
|
|
taskContext.data = { textImportedAsText: true };
|
|
});
|
|
|
|
describe('extractHtmlTitle', () => {
|
|
it('should extract title from HTML content', () => {
|
|
const html = `
|
|
<html>
|
|
<head>
|
|
<title>Test Title</title>
|
|
</head>
|
|
<body>
|
|
<p>Content</p>
|
|
</body>
|
|
</html>
|
|
`;
|
|
|
|
const title = importUtils.extractHtmlTitle(html);
|
|
expect(title).toBe('Test Title');
|
|
});
|
|
|
|
it('should return null if no title tag is present', () => {
|
|
const html = `
|
|
<html>
|
|
<head>
|
|
</head>
|
|
<body>
|
|
<p>Content</p>
|
|
</body>
|
|
</html>
|
|
`;
|
|
|
|
const title = importUtils.extractHtmlTitle(html);
|
|
expect(title).toBeNull();
|
|
});
|
|
});
|
|
|
|
describe('importSingleFile with HTML', () => {
|
|
it('should import HTML file with title from title tag', () => {
|
|
const file = {
|
|
originalname: 'test.html',
|
|
mimetype: 'text/html',
|
|
buffer: Buffer.from(`
|
|
<html>
|
|
<head>
|
|
<title>HTML Title</title>
|
|
</head>
|
|
<body>
|
|
<p>Test content</p>
|
|
</body>
|
|
</html>
|
|
`)
|
|
};
|
|
|
|
const note = importSingle.importSingleFile(taskContext, file, parentNote);
|
|
expect(note.title).toBe('HTML Title');
|
|
expect(note.mime).toBe('text/html');
|
|
});
|
|
|
|
it('should import HTML file with title from h1 when no title tag', () => {
|
|
const file = {
|
|
originalname: 'test.html',
|
|
mimetype: 'text/html',
|
|
buffer: Buffer.from(`
|
|
<html>
|
|
<body>
|
|
<h1>Heading Title</h1>
|
|
<p>Test content</p>
|
|
</body>
|
|
</html>
|
|
`)
|
|
};
|
|
|
|
const note = importSingle.importSingleFile(taskContext, file, parentNote);
|
|
expect(note.title).toBe('Heading Title');
|
|
expect(note.mime).toBe('text/html');
|
|
});
|
|
|
|
it('should import HTML file with filename as title when no title or h1', () => {
|
|
const file = {
|
|
originalname: 'test-document.html',
|
|
mimetype: 'text/html',
|
|
buffer: Buffer.from(`
|
|
<html>
|
|
<body>
|
|
<p>Test content without title</p>
|
|
</body>
|
|
</html>
|
|
`)
|
|
};
|
|
|
|
const note = importSingle.importSingleFile(taskContext, file, parentNote);
|
|
expect(note.title).toBe('test-document');
|
|
expect(note.mime).toBe('text/html');
|
|
});
|
|
|
|
it('should sanitize HTML content during import', () => {
|
|
const file = {
|
|
originalname: 'test.html',
|
|
mimetype: 'text/html',
|
|
buffer: Buffer.from(`
|
|
<html>
|
|
<head>
|
|
<title>Test Title</title>
|
|
<script>alert('xss');</script>
|
|
</head>
|
|
<body>
|
|
<p>Safe content</p>
|
|
<script>alert('xss');</script>
|
|
</body>
|
|
</html>
|
|
`)
|
|
};
|
|
|
|
const note = importSingle.importSingleFile(taskContext, file, parentNote);
|
|
expect(note.title).toBe('Test Title');
|
|
expect(note.content).not.toContain('<script>');
|
|
expect(note.content).toContain('<p>Safe content</p>');
|
|
});
|
|
});
|
|
});
|