flagsmith-nodejs 6.1.0 → 6.2.0

package/tests/sdk/flagsmith.test.ts

@@ -9,7 +9,7 @@ import {
     badFetch
 } from './utils.js';
 import { DefaultFlag, Flags } from '../../sdk/models.js';
-import { delay } from '../../sdk/utils.js';
+import { delay, getUserAgent } from '../../sdk/utils.js';
 import { EnvironmentModel } from '../../flagsmith-engine/environments/models.js';
 import { BaseOfflineHandler } from '../../sdk/offline_handlers.js';
 import { Agent } from 'undici';
@@ -40,6 +40,160 @@ test('test_update_environment_sets_environment', async () => {
     expect(await flg.getEnvironment()).toStrictEqual(model);
 });
 
+test('test_update_environment_handles_paginated_document', async () => {
+    type EnvDocumentMockResponse = {
+        responseHeader: string | null;
+        page: any;
+    };
+
+    const createMockFetch = (pages: EnvDocumentMockResponse[]) => {
+        let callCount = 0;
+        return vi.fn((url: string, options?: RequestInit) => {
+            if (url.includes('/environment-document')) {
+                const document = envDocumentMockResponse[callCount];
+                if (document) {
+                    callCount++;
+
+                    const responseHeaders: Record<string, string> = {};
+
+                    if (document.responseHeader) {
+                        responseHeaders['Link'] = `<${document.responseHeader}>; rel="next"`;
+                    }
+
+                    return Promise.resolve(
+                        new Response(JSON.stringify(document.page), {
+                            status: 200,
+                            headers: responseHeaders
+                        })
+                    );
+                }
+            }
+            return Promise.resolve(new Response('unknown url ' + url, { status: 404 }));
+        });
+    };
+
+    const envDocumentMockResponse: EnvDocumentMockResponse[] = [
+        {
+            responseHeader: '/api/v1/environment-document?page=2',
+            page: {
+                id: 1,
+                api_key: 'test-key',
+                project: {
+                    id: 1,
+                    name: 'test',
+                    organisation: {
+                        id: 1,
+                        name: 'Test Org',
+                        feature_analytics: false,
+                        persist_trait_data: true,
+                        stop_serving_flags: false
+                    },
+                    hide_disabled_flags: false,
+                    segments: []
+                },
+                feature_states: [
+                    {
+                        feature_state_value: 'first_page_feature_state',
+                        multivariate_feature_state_values: [],
+                        django_id: 81027,
+                        feature: {
+                            id: 15058,
+                            type: 'STANDARD',
+                            name: 'string_feature'
+                        },
+                        enabled: false
+                    },
+                    {
+                        feature_state_value: 'second_page_feature_state',
+                        multivariate_feature_state_values: [],
+                        django_id: 81027,
+                        feature: {
+                            id: 15058,
+                            type: 'STANDARD',
+                            name: 'string_feature'
+                        },
+                        enabled: false
+                    },
+                    {
+                        feature_state_value: 'third_page_feature_state',
+                        multivariate_feature_state_values: [],
+                        django_id: 81027,
+                        feature: {
+                            id: 15058,
+                            type: 'STANDARD',
+                            name: 'string_feature'
+                        },
+                        enabled: false
+                    }
+                ],
+                identity_overrides: [{ id: 1, identifier: 'user1' }]
+            }
+        },
+        {
+            responseHeader: '/api/v1/environment-document?page=3',
+            page: {
+                api_key: 'test-key',
+                project: {
+                    id: 1,
+                    name: 'test',
+                    organisation: {
+                        id: 1,
+                        name: 'Test Org',
+                        feature_analytics: false,
+                        persist_trait_data: true,
+                        stop_serving_flags: false
+                    },
+                    hide_disabled_flags: false,
+                    segments: []
+                },
+                feature_states: [],
+                identity_overrides: [{ id: 2, identifier: 'user2' }]
+            }
+        },
+        {
+            responseHeader: null,
+            page: {
+                api_key: 'test-key',
+                project: {
+                    id: 1,
+                    name: 'test',
+                    organisation: {
+                        id: 1,
+                        name: 'Test Org',
+                        feature_analytics: false,
+                        persist_trait_data: true,
+                        stop_serving_flags: false
+                    },
+                    hide_disabled_flags: false,
+                    segments: []
+                },
+                feature_states: [],
+                identity_overrides: [{ id: 2, identifier: 'user3' }]
+            }
+        }
+    ];
+
+    const flg = new Flagsmith({
+        environmentKey: 'ser.key',
+        enableLocalEvaluation: true,
+        fetch: createMockFetch(envDocumentMockResponse)
+    });
+
+    const environment = await flg.getEnvironment();
+
+    expect(environment.identityOverrides).toHaveLength(3);
+    expect(environment.identityOverrides[0].identifier).toBe('user1');
+    expect(environment.identityOverrides[1].identifier).toBe('user2');
+    expect(environment.identityOverrides[2].identifier).toBe('user3');
+    expect(environment.featureStates).toHaveLength(3);
+    expect(environment.featureStates[0].getValue()).toBe('first_page_feature_state');
+    expect(environment.featureStates[1].getValue()).toBe('second_page_feature_state');
+    expect(environment.featureStates[2].getValue()).toBe('third_page_feature_state');
+    expect(environment.project.name).toBe('test');
+    expect(environment.project.organisation.name).toBe('Test Org');
+    expect(environment.project.organisation.id).toBe(1);
+});
+
 test('test_set_agent_options', async () => {
     const agent = new Agent({});
 
@@ -358,3 +512,10 @@ test('getIdentityFlags succeeds if initial fetch failed then succeeded', async (
     const flags2 = await flg.getIdentityFlags('test-user');
     expect(flags2.isFeatureEnabled('some_feature')).toBe(true);
 });
+
+test('get_user_agent_extracts_version_from_package_json', async () => {
+    const userAgent = getUserAgent();
+    const packageJson = require('../../package.json');
+
+    expect(userAgent).toBe(`flagsmith-nodejs-sdk/${packageJson.version}`);
+});