|
1 | 1 | /* eslint-env jest */ |
2 | 2 |
|
3 | 3 | const originalNodeEnv = process.env.NODE_ENV; |
| 4 | +const originalSolveHistoryUserIds = process.env.FEATURE_SOLVE_HISTORY_USER_IDS; |
| 5 | +const originalFriendsUserIds = process.env.FEATURE_FRIENDS_USER_IDS; |
4 | 6 |
|
5 | | -const loadFeatures = (nodeEnv) => { |
| 7 | +const loadFeatures = ({ nodeEnv, solveHistoryUserIds, friendsUserIds }) => { |
6 | 8 | jest.resetModules(); |
7 | 9 | process.env.NODE_ENV = nodeEnv; |
| 10 | + if (solveHistoryUserIds === undefined) { |
| 11 | + delete process.env.FEATURE_SOLVE_HISTORY_USER_IDS; |
| 12 | + } else { |
| 13 | + process.env.FEATURE_SOLVE_HISTORY_USER_IDS = solveHistoryUserIds; |
| 14 | + } |
| 15 | + if (friendsUserIds === undefined) { |
| 16 | + delete process.env.FEATURE_FRIENDS_USER_IDS; |
| 17 | + } else { |
| 18 | + process.env.FEATURE_FRIENDS_USER_IDS = friendsUserIds; |
| 19 | + } |
8 | 20 | // eslint-disable-next-line global-require |
9 | 21 | return require('./features'); |
10 | 22 | }; |
11 | 23 |
|
12 | 24 | afterEach(() => { |
13 | 25 | process.env.NODE_ENV = originalNodeEnv; |
| 26 | + if (originalSolveHistoryUserIds === undefined) { |
| 27 | + delete process.env.FEATURE_SOLVE_HISTORY_USER_IDS; |
| 28 | + } else { |
| 29 | + process.env.FEATURE_SOLVE_HISTORY_USER_IDS = originalSolveHistoryUserIds; |
| 30 | + } |
| 31 | + if (originalFriendsUserIds === undefined) { |
| 32 | + delete process.env.FEATURE_FRIENDS_USER_IDS; |
| 33 | + } else { |
| 34 | + process.env.FEATURE_FRIENDS_USER_IDS = originalFriendsUserIds; |
| 35 | + } |
14 | 36 | }); |
15 | 37 |
|
16 | 38 | describe('feature flags', () => { |
17 | 39 | it('keeps friends out of production', () => { |
18 | | - expect(loadFeatures('production').isFeatureEnabled('friends')).toBe(false); |
| 40 | + expect(loadFeatures({ nodeEnv: 'production' }).isFeatureEnabled('friends')).toBe(false); |
19 | 41 | }); |
20 | 42 |
|
21 | 43 | it('keeps friends available to local development', () => { |
22 | | - expect(loadFeatures('development').isFeatureEnabled('friends')).toBe(true); |
| 44 | + expect(loadFeatures({ nodeEnv: 'development' }).isFeatureEnabled('friends')).toBe(true); |
23 | 45 | }); |
24 | 46 |
|
25 | 47 | it('keeps solve history development-only', () => { |
26 | | - expect(loadFeatures('development').isFeatureEnabled('solveHistory')).toBe(true); |
27 | | - expect(loadFeatures('dev').isFeatureEnabled('solveHistory')).toBe(true); |
28 | | - expect(loadFeatures('production').isFeatureEnabled('solveHistory')).toBe(false); |
29 | | - expect(loadFeatures('prod').isFeatureEnabled('solveHistory')).toBe(false); |
| 48 | + expect(loadFeatures({ nodeEnv: 'development' }).isFeatureEnabled('solveHistory')).toBe(true); |
| 49 | + expect(loadFeatures({ nodeEnv: 'dev' }).isFeatureEnabled('solveHistory')).toBe(true); |
| 50 | + expect(loadFeatures({ nodeEnv: 'production' }).isFeatureEnabled('solveHistory')).toBe(false); |
| 51 | + expect(loadFeatures({ nodeEnv: 'prod' }).isFeatureEnabled('solveHistory')).toBe(false); |
| 52 | + }); |
| 53 | + |
| 54 | + it('can enable a disabled feature for selected WCA users', () => { |
| 55 | + const features = loadFeatures({ |
| 56 | + nodeEnv: 'prod', |
| 57 | + solveHistoryUserIds: '8184, 42, invalid, 0', |
| 58 | + friendsUserIds: '8184', |
| 59 | + }); |
| 60 | + |
| 61 | + expect(features.isFeatureEnabled('solveHistory')).toBe(false); |
| 62 | + expect(features.isFeatureEnabledForUser('solveHistory', 8184)).toBe(true); |
| 63 | + expect(features.isFeatureEnabledForUser('solveHistory', '42')).toBe(true); |
| 64 | + expect(features.isFeatureEnabledForUser('solveHistory', 8185)).toBe(false); |
| 65 | + expect(features.isFeatureEnabledForUser('friends', 8184)).toBe(true); |
| 66 | + expect(features.isFeatureEnabledForUser('unknown', 8184)).toBe(false); |
30 | 67 | }); |
31 | 68 | }); |
0 commit comments