ArrQueueCleaner/tests/test-utils.ts

import { Config, QueueItem } from '../src/types';

type DeepPartial<T> = {
    [P in keyof T]?: T[P] extends object ? DeepPartial<T[P]> : T[P];
};

export const createMockConfig = (overrides: DeepPartial<Config> = {}): Config => ({
    sonarr: {
        host: 'http://localhost:8989',
        apiKey: 'test-api-key',
        enabled: true,
        ...overrides.sonarr
    },
    rules: {
        removeQualityBlocked: false,
        blockRemovedQualityReleases: false,
        removeArchiveBlocked: false,
        blockRemovedArchiveReleases: false,
        removeNoFilesReleases: false,
        blockRemovedNoFilesReleases: false,
        removeNotAnUpgrade: false,
        removeSeriesIdMismatch: false,
        blockRemovedSeriesIdMismatchReleases: false,
        removeUndeterminedSample: false,
        blockRemovedUndeterminedSampleReleases: false,
        ...overrides.rules
    },
    dryRun: overrides.dryRun || false,
    schedule: overrides.schedule || '*/5 * * * *',
    logLevel: overrides.logLevel || 'info'
});

export const createMockQueueItem = (overrides: Partial<QueueItem> = {}): QueueItem => ({
    id: 123,
    title: 'Test.Show.S01E01',
    status: 'completed',
    trackedDownloadStatus: 'warning',
    trackedDownloadState: 'importPending',
    statusMessages: [],
    downloadId: `download_${Math.random().toString(36).substr(2, 9)}`,
    ...overrides
});

export const createQualityBlockedItem = (): QueueItem =>
    createMockQueueItem({
        statusMessages: [{
            messages: ['upgrade for existing episode']
        }]
    });

export const createArchiveBlockedItem = (): QueueItem =>
    createMockQueueItem({
        statusMessages: [{
            messages: ['Found archive file, might need to be extracted']
        }]
    });

export const createNoFilesBlockedItem = (): QueueItem =>
    createMockQueueItem({
        statusMessages: [{
            messages: ['No files found are eligible for import']
        }]
    });

export const createNotAnUpgradeItem = (): QueueItem =>
    createMockQueueItem({
        statusMessages: [{
            messages: ['Not an upgrade']
        }]
    });

export const createSeriesIdMismatchItem = (overrides: Partial<QueueItem> = {}): QueueItem =>
    createMockQueueItem({
        trackedDownloadState: 'importBlocked',
        statusMessages: [{
            messages: ['Found matching series via grab history, but release was matched to series by ID. Automatic import is not possible. See the FAQ for details.']
        }],
        ...overrides
    });

export const createUndeterminedSampleItem = (overrides: Partial<QueueItem> = {}): QueueItem =>
    createMockQueueItem({
        statusMessages: [{
            messages: ['Unable to determine if file is a sample']
        }],
        ...overrides
    });