mirror of
https://github.com/mendableai/firecrawl.git
synced 2024-11-16 11:42:24 +08:00
Added crawl limit unit test
I think this test is over relying on mocks but I have no idea on how to fix this without changing the code arch structure
This commit is contained in:
parent
45f2765601
commit
009df6c930
|
@ -188,5 +188,38 @@ describe('WebCrawler', () => {
|
|||
// Check that the backward link is included if allowBackwardCrawling is true
|
||||
expect(results.some(r => r.url === 'https://mendable.ai')).toBe(true);
|
||||
});
|
||||
|
||||
it('should respect the limit parameter by not returning more links than specified', async () => {
|
||||
const initialUrl = 'http://example.com';
|
||||
const limit = 2; // Set a limit for the number of links
|
||||
|
||||
crawler = new WebCrawler({
|
||||
initialUrl: initialUrl,
|
||||
includes: [],
|
||||
excludes: [],
|
||||
limit: limit, // Apply the limit
|
||||
maxCrawledDepth: 10
|
||||
});
|
||||
|
||||
// Mock sitemap fetching function to return more links than the limit
|
||||
crawler['tryFetchSitemapLinks'] = jest.fn().mockResolvedValue([
|
||||
initialUrl,
|
||||
initialUrl + '/page1',
|
||||
initialUrl + '/page2',
|
||||
initialUrl + '/page3'
|
||||
]);
|
||||
|
||||
const filteredLinks = crawler['filterLinks'](
|
||||
[initialUrl, initialUrl + '/page1', initialUrl + '/page2', initialUrl + '/page3'],
|
||||
limit,
|
||||
10
|
||||
);
|
||||
|
||||
expect(filteredLinks.length).toBe(limit); // Check if the number of results respects the limit
|
||||
expect(filteredLinks).toEqual([
|
||||
initialUrl,
|
||||
initialUrl + '/page1'
|
||||
]);
|
||||
});
|
||||
});
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user