126 lines
4.6 KiB
TypeScript
126 lines
4.6 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
describe('FeatureFlagStorage', () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
vi.resetModules();
|
|
});
|
|
|
|
it('should initialize correctly', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
expect(featureFlagStorage.inited).toBe(false);
|
|
});
|
|
|
|
it('should return inited status correctly', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
expect(featureFlagStorage.inited).toBe(false);
|
|
featureFlagStorage.setFlags({ feature1: true });
|
|
expect(featureFlagStorage.inited).toBe(true);
|
|
});
|
|
|
|
it('should set and get feature flags correctly', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const flags = { feature1: true, feature2: false };
|
|
featureFlagStorage.setFlags(flags);
|
|
const fgValues = featureFlagStorage.getFlags();
|
|
|
|
expect(fgValues.feature1).toEqual(true);
|
|
expect(fgValues.feature2).toEqual(false);
|
|
// fallback to false if key not exits
|
|
expect(fgValues.feature3).toEqual(false);
|
|
});
|
|
|
|
it('should emit change event on setting new flags', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const flags = { feature1: true, feature2: false };
|
|
const listener = vi.fn();
|
|
featureFlagStorage.on('change', listener);
|
|
|
|
featureFlagStorage.setFlags(flags);
|
|
expect(listener).toHaveBeenCalledWith(flags);
|
|
});
|
|
|
|
it('should not emit change event when setting same flags', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const flags = { feature1: true, feature2: false };
|
|
featureFlagStorage.setFlags(flags);
|
|
|
|
const listener = vi.fn();
|
|
featureFlagStorage.on('change', listener);
|
|
|
|
featureFlagStorage.setFlags(flags);
|
|
expect(listener).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should throw error when getting flags before initialization', async () => {
|
|
vi.stubEnv('NODE_ENV', 'development');
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
expect(() => featureFlagStorage.getFlags()).toThrow(
|
|
'Trying access feature flag values before the storage been init.',
|
|
);
|
|
});
|
|
|
|
it('should clear flags correctly', async () => {
|
|
vi.stubEnv('NODE_ENV', 'development');
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
featureFlagStorage.setFlags({ feature1: true });
|
|
featureFlagStorage.clear();
|
|
expect(() => featureFlagStorage.getFlags()).toThrow();
|
|
});
|
|
|
|
it('should return all keys', async () => {
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
featureFlagStorage.setFlags({ feature1: true, feature2: false });
|
|
const flags = featureFlagStorage.getFlags();
|
|
expect(flags.keys).toEqual(['feature1', 'feature2']);
|
|
});
|
|
|
|
it('should return none keys', async () => {
|
|
vi.stubEnv('NODE_ENV', 'production');
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const flags = featureFlagStorage.getFlags();
|
|
expect(flags.keys).toEqual([]);
|
|
});
|
|
|
|
it('should return isInited', async () => {
|
|
vi.stubEnv('NODE_ENV', 'production');
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const flags = featureFlagStorage.getFlags();
|
|
expect(flags.isInited).toEqual(false);
|
|
featureFlagStorage.setFlags({ feature1: true, feature2: false });
|
|
expect(flags.isInited).toEqual(true);
|
|
});
|
|
|
|
it('should unshift the function into interceptors by calling `use`', async () => {
|
|
vi.stubEnv('NODE_ENV', 'production');
|
|
const { featureFlagStorage } = await import('../src/utils/storage');
|
|
const fnGetter = vi.fn();
|
|
featureFlagStorage.use(fnGetter);
|
|
const flags = featureFlagStorage.getFlags();
|
|
const testKey = 'bot.arch.bot.fg.test.1';
|
|
// trigger get of the proxy
|
|
flags[testKey];
|
|
expect(fnGetter).toHaveBeenCalledWith(testKey);
|
|
|
|
// trigger set of the proxy
|
|
const fnSetter = vi.fn(() => {
|
|
flags['bot.arch.bot.fg.test.1'] = true;
|
|
});
|
|
expect(fnSetter).toThrowError();
|
|
});
|
|
});
|