import { MemoryStorage } from '@crawlee/memory-storage';
import type { RequestQueueClient } from '@crawlee/types';

describe('RequestQueue handledRequestCount should update', () => {
    const storage = new MemoryStorage({
        persistStorage: false,
    });

    let requestQueue: RequestQueueClient;

    beforeAll(async () => {
        const { id } = await storage.requestQueues().getOrCreate('handledRequestCount');
        requestQueue = storage.requestQueue(id);
    });

    test('after updating the request, it should increment the handledRequestCount', async () => {
        const { requestId } = await requestQueue.addRequest({ url: 'http://example.com/1', uniqueKey: '1' });

        await requestQueue.updateRequest({
            url: 'http://example.com/1',
            uniqueKey: '1',
            id: requestId,
            handledAt: new Date().toISOString(),
        });

        const updatedStatistics = await requestQueue.get();
        expect(updatedStatistics?.handledRequestCount).toEqual(1);
    });

    test('adding an already handled request should increment the handledRequestCount', async () => {
        await requestQueue.addRequest({
            url: 'http://example.com/2',
            uniqueKey: '2',
            handledAt: new Date().toISOString(),
        });

        const updatedStatistics = await requestQueue.get();
        expect(updatedStatistics?.handledRequestCount).toEqual(2);
    });

    test('deleting a request should decrement the handledRequestCount', async () => {
        const { requestId } = await requestQueue.addRequest({
            url: 'http://example.com/3',
            uniqueKey: '3',
            handledAt: new Date().toISOString(),
        });

        await requestQueue.deleteRequest(requestId);

        const updatedStatistics = await requestQueue.get();
        expect(updatedStatistics?.handledRequestCount).toEqual(2);
    });

    test('updating an already handled request should not increment the handledRequestCount again', async () => {
        const { requestId } = await requestQueue.addRequest({
            url: 'http://example.com/4',
            uniqueKey: '4',
            handledAt: new Date().toISOString(),
        });

        const { handledRequestCount } = (await requestQueue.get())!;

        await requestQueue.updateRequest({
            url: 'http://example.com/4',
            uniqueKey: '4',
            id: requestId,
            handledAt: new Date().toISOString(),
        });

        const updatedStatistics = await requestQueue.get();
        expect(updatedStatistics?.handledRequestCount).toEqual(handledRequestCount);
    });
});