unleash.unleash/src/lib/db/feature-strategy-store.test.ts

import dbInit from '../../test/e2e/helpers/database-init';
import getLogger from '../../test/fixtures/no-logger';
import FeatureStrategiesStore from './feature-strategy-store';
import FeatureToggleStore from './feature-toggle-store';
import StrategyStore from './strategy-store';

let db;

beforeAll(async () => {
    db = await dbInit('feature_strategy_store_serial', getLogger);
    getLogger.setMuteError(true);
});

afterAll(async () => {
    if (db) {
        await db.destroy();
    }
    getLogger.setMuteError(false);
});

test('returns 0 if no custom strategies', async () => {
    // Arrange
    const featureStrategiesStore: FeatureStrategiesStore =
        db.stores.featureStrategiesStore;

    // Act
    const inUseCount =
        await featureStrategiesStore.getCustomStrategiesInUseCount();

    // Assert
    expect(inUseCount).toEqual(0);
});

test('returns 0 if no custom strategies are in use', async () => {
    // Arrange
    const featureToggleStore: FeatureToggleStore = db.stores.featureToggleStore;
    const featureStrategiesStore: FeatureStrategiesStore =
        db.stores.featureStrategiesStore;
    const strategyStore: StrategyStore = db.stores.strategyStore;

    featureToggleStore.create('default', {
        name: 'test-toggle-2',
    });

    strategyStore.createStrategy({
        name: 'strategy-2',
        built_in: 0,
        parameters: [],
        description: '',
        createdAt: '2023-06-09T09:00:12.242Z',
    });

    // Act
    const inUseCount =
        await featureStrategiesStore.getCustomStrategiesInUseCount();

    // Assert
    expect(inUseCount).toEqual(0);
});

test('counts custom strategies in use', async () => {
    // Arrange
    const featureToggleStore: FeatureToggleStore = db.stores.featureToggleStore;
    const featureStrategiesStore: FeatureStrategiesStore =
        db.stores.featureStrategiesStore;
    const strategyStore: StrategyStore = db.stores.strategyStore;

    await featureToggleStore.create('default', {
        name: 'test-toggle',
    });

    await strategyStore.createStrategy({
        name: 'strategy-1',
        built_in: 0,
        parameters: [],
        description: '',
        createdAt: '2023-06-09T09:00:12.242Z',
    });

    await featureStrategiesStore.createStrategyFeatureEnv({
        projectId: 'default',
        featureName: 'test-toggle',
        strategyName: 'strategy-1',
        environment: 'default',
        parameters: {},
        constraints: [],
    });

    // Act
    const inUseCount =
        await featureStrategiesStore.getCustomStrategiesInUseCount();

    // Assert
    expect(inUseCount).toEqual(1);
});