feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/arch/bot-flags/__tests__/storage.test.ts

@@ -0,0 +1,125 @@
+/*
+ * 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();
+  });
+});