unleash.unleash/src/test/e2e/services/environment-service.test.ts

import EnvironmentService from '../../../lib/services/environment-service';
import { createTestConfig } from '../../config/test-config';
import dbInit from '../helpers/database-init';
import NotFoundError from '../../../lib/error/notfound-error';
import { IUnleashStores } from '../../../lib/types/stores';
import NameExistsError from '../../../lib/error/name-exists-error';

let stores: IUnleashStores;
let db;
let service: EnvironmentService;

beforeAll(async () => {
    const config = createTestConfig();
    db = await dbInit('environment_service_serial', config.getLogger);
    stores = db.stores;
    service = new EnvironmentService(stores, config);
});
afterAll(async () => {
    await db.destroy();
});

test('Can create and get environment', async () => {
    const created = await service.create({
        name: 'testenv',
        displayName: 'Environment for testing',
    });

    const retrieved = await service.get('testenv');
    expect(retrieved).toEqual(created);
});

test('Can delete environment', async () => {
    await service.create({
        name: 'testenv',
        displayName: 'Environment for testing',
    });
    await service.delete('testenv');
    return expect(async () => service.get('testenv')).rejects.toThrow(
        NotFoundError,
    );
});

test('Can get all', async () => {
    await service.create({
        name: 'testenv',
        displayName: 'Environment for testing',
    });

    const environments = await service.getAll();
    expect(environments).toHaveLength(2); // the one we created plus ':global:'
});

test('Can update display name', async () => {
    await service.create({
        name: 'testenv',
        displayName: 'Environment for testing',
    });

    await service.update('testenv', { displayName: 'Different name' });
    const updated = await service.get('testenv');
    expect(updated.displayName).toEqual('Different name');
});

test('Can connect environment to project', async () => {
    await service.create({ name: 'test-connection', displayName: '' });
    await stores.featureToggleStore.createFeature('default', {
        name: 'test-connection',
        type: 'release',
        description: '',
        stale: false,
        variants: [],
    });
    await service.connectProjectToEnvironment('test-connection', 'default');
    const overview = await stores.projectStore.getProjectOverview(
        'default',
        false,
    );
    overview.forEach((f) => {
        expect(f.environments).toEqual([
            {
                name: 'test-connection',
                displayName: '',
                enabled: false,
            },
        ]);
    });
});

test('Can remove environment from project', async () => {
    await service.create({ name: 'removal-test', displayName: '' });
    await stores.featureToggleStore.createFeature('default', {
        name: 'removal-test',
    });
    await service.removeEnvironmentFromProject('test-connection', 'default');
    await service.connectProjectToEnvironment('removal-test', 'default');
    let overview = await stores.projectStore.getProjectOverview(
        'default',
        false,
    );
    expect(overview.length).toBeGreaterThan(0);
    overview.forEach((f) => {
        expect(f.environments).toEqual([
            {
                name: 'removal-test',
                displayName: '',
                enabled: false,
            },
        ]);
    });
    await service.removeEnvironmentFromProject('removal-test', 'default');
    overview = await stores.projectStore.getProjectOverview('default', false);
    expect(overview.length).toBeGreaterThan(0);
    overview.forEach((o) => {
        expect(o.environments).toEqual([]);
    });
});

test('Adding same environment twice should throw a NameExistsError', async () => {
    await service.create({ name: 'uniqueness-test', displayName: '' });
    await service.removeEnvironmentFromProject('test-connection', 'default');
    await service.removeEnvironmentFromProject('removal-test', 'default');

    await service.connectProjectToEnvironment('uniqueness-test', 'default');
    return expect(async () =>
        service.connectProjectToEnvironment('uniqueness-test', 'default'),
    ).rejects.toThrow(
        new NameExistsError(
            'default already has the environment uniqueness-test enabled',
        ),
    );
});

test('Removing environment not connected to project should be a noop', async () =>
    expect(async () =>
        service.removeEnvironmentFromProject(
            'some-non-existing-environment',
            'default',
        ),
    ).resolves);