@adtrackify/at-service-common 4.0.13 → 4.0.15

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (109) hide show
  1. package/dist/cjs/__tests__/identity-cache/helpers.spec.js +37 -38
  2. package/dist/cjs/__tests__/identity-cache/helpers.spec.js.map +1 -1
  3. package/dist/cjs/__tests__/identity-cache/identity-cache-tier-routing.spec.js +92 -349
  4. package/dist/cjs/__tests__/identity-cache/identity-cache-tier-routing.spec.js.map +1 -1
  5. package/dist/cjs/__tests__/identity-cache/identity-client.spec.js +360 -0
  6. package/dist/cjs/__tests__/identity-cache/identity-client.spec.js.map +1 -0
  7. package/dist/cjs/__tests__/identity-cache/identity-gateway-cache-hit.spec.js +138 -0
  8. package/dist/cjs/__tests__/identity-cache/identity-gateway-cache-hit.spec.js.map +1 -0
  9. package/dist/cjs/__tests__/identity-cache/identity-gateway-cache-miss.spec.js +153 -0
  10. package/dist/cjs/__tests__/identity-cache/identity-gateway-cache-miss.spec.js.map +1 -0
  11. package/dist/cjs/__tests__/identity-cache/identity-gateway-constructor.spec.js +72 -0
  12. package/dist/cjs/__tests__/identity-cache/identity-gateway-constructor.spec.js.map +1 -0
  13. package/dist/cjs/__tests__/identity-cache/identity-gateway-edge-cases.spec.js +89 -0
  14. package/dist/cjs/__tests__/identity-cache/identity-gateway-edge-cases.spec.js.map +1 -0
  15. package/dist/cjs/__tests__/identity-cache/identity-gateway-enqueue-async.spec.js +136 -0
  16. package/dist/cjs/__tests__/identity-cache/identity-gateway-enqueue-async.spec.js.map +1 -0
  17. package/dist/cjs/__tests__/identity-cache/identity-gateway-stale-cache.spec.js +224 -0
  18. package/dist/cjs/__tests__/identity-cache/identity-gateway-stale-cache.spec.js.map +1 -0
  19. package/dist/cjs/__tests__/identity-cache/identity-gateway-test-setup.d.ts +28 -0
  20. package/dist/cjs/__tests__/identity-cache/identity-gateway-test-setup.js +90 -0
  21. package/dist/cjs/__tests__/identity-cache/identity-gateway-test-setup.js.map +1 -0
  22. package/dist/cjs/clients/internal-api/identity-client.d.ts +12 -2
  23. package/dist/cjs/clients/internal-api/identity-client.js +59 -13
  24. package/dist/cjs/clients/internal-api/identity-client.js.map +1 -1
  25. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.d.ts +1 -7
  26. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.js +0 -305
  27. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.js.map +1 -1
  28. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-gateway-service.d.ts +14 -14
  29. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-gateway-service.js +122 -330
  30. package/dist/cjs/services/db/identity-cache-dynamodb-service/identity-gateway-service.js.map +1 -1
  31. package/dist/cjs/services/db/identity-cache-dynamodb-service/index.d.ts +0 -1
  32. package/dist/cjs/services/db/identity-cache-dynamodb-service/index.js +0 -1
  33. package/dist/cjs/services/db/identity-cache-dynamodb-service/index.js.map +1 -1
  34. package/dist/cjs/services/db/identity-cache-dynamodb-service/types.d.ts +8 -158
  35. package/dist/cjs/services/db/identity-cache-dynamodb-service/types.js +1 -15
  36. package/dist/cjs/services/db/identity-cache-dynamodb-service/types.js.map +1 -1
  37. package/dist/esm/__tests__/identity-cache/helpers.spec.js +26 -27
  38. package/dist/esm/__tests__/identity-cache/helpers.spec.js.map +1 -1
  39. package/dist/esm/__tests__/identity-cache/identity-cache-tier-routing.spec.js +92 -349
  40. package/dist/esm/__tests__/identity-cache/identity-cache-tier-routing.spec.js.map +1 -1
  41. package/dist/esm/__tests__/identity-cache/identity-client.spec.js +358 -0
  42. package/dist/esm/__tests__/identity-cache/identity-client.spec.js.map +1 -0
  43. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-hit.spec.d.ts +1 -0
  44. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-hit.spec.js +136 -0
  45. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-hit.spec.js.map +1 -0
  46. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-miss.spec.d.ts +1 -0
  47. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-miss.spec.js +151 -0
  48. package/dist/esm/__tests__/identity-cache/identity-gateway-cache-miss.spec.js.map +1 -0
  49. package/dist/esm/__tests__/identity-cache/identity-gateway-constructor.spec.d.ts +1 -0
  50. package/dist/esm/__tests__/identity-cache/identity-gateway-constructor.spec.js +70 -0
  51. package/dist/esm/__tests__/identity-cache/identity-gateway-constructor.spec.js.map +1 -0
  52. package/dist/esm/__tests__/identity-cache/identity-gateway-edge-cases.spec.d.ts +1 -0
  53. package/dist/esm/__tests__/identity-cache/identity-gateway-edge-cases.spec.js +87 -0
  54. package/dist/esm/__tests__/identity-cache/identity-gateway-edge-cases.spec.js.map +1 -0
  55. package/dist/esm/__tests__/identity-cache/identity-gateway-enqueue-async.spec.d.ts +1 -0
  56. package/dist/esm/__tests__/identity-cache/identity-gateway-enqueue-async.spec.js +134 -0
  57. package/dist/esm/__tests__/identity-cache/identity-gateway-enqueue-async.spec.js.map +1 -0
  58. package/dist/esm/__tests__/identity-cache/identity-gateway-stale-cache.spec.d.ts +1 -0
  59. package/dist/esm/__tests__/identity-cache/identity-gateway-stale-cache.spec.js +222 -0
  60. package/dist/esm/__tests__/identity-cache/identity-gateway-stale-cache.spec.js.map +1 -0
  61. package/dist/esm/__tests__/identity-cache/identity-gateway-test-setup.d.ts +28 -0
  62. package/dist/esm/__tests__/identity-cache/identity-gateway-test-setup.js +83 -0
  63. package/dist/esm/__tests__/identity-cache/identity-gateway-test-setup.js.map +1 -0
  64. package/dist/esm/clients/internal-api/identity-client.d.ts +12 -2
  65. package/dist/esm/clients/internal-api/identity-client.js +60 -14
  66. package/dist/esm/clients/internal-api/identity-client.js.map +1 -1
  67. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.d.ts +1 -7
  68. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.js +1 -303
  69. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.js.map +1 -1
  70. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-gateway-service.d.ts +14 -14
  71. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-gateway-service.js +125 -330
  72. package/dist/esm/services/db/identity-cache-dynamodb-service/identity-gateway-service.js.map +1 -1
  73. package/dist/esm/services/db/identity-cache-dynamodb-service/index.d.ts +0 -1
  74. package/dist/esm/services/db/identity-cache-dynamodb-service/index.js +0 -1
  75. package/dist/esm/services/db/identity-cache-dynamodb-service/index.js.map +1 -1
  76. package/dist/esm/services/db/identity-cache-dynamodb-service/types.d.ts +8 -158
  77. package/dist/esm/services/db/identity-cache-dynamodb-service/types.js +0 -14
  78. package/dist/esm/services/db/identity-cache-dynamodb-service/types.js.map +1 -1
  79. package/package.json +2 -2
  80. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-db-service-v2-comparison.spec.js +0 -646
  81. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-db-service-v2-comparison.spec.js.map +0 -1
  82. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-service-v2-comparison.spec.js +0 -325
  83. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-service-v2-comparison.spec.js.map +0 -1
  84. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-service.spec.js +0 -1286
  85. package/dist/cjs/__tests__/identity-cache/identity-cache-dynamodb-service.spec.js.map +0 -1
  86. package/dist/cjs/__tests__/identity-cache/identity-gateway-service.spec.js +0 -2170
  87. package/dist/cjs/__tests__/identity-cache/identity-gateway-service.spec.js.map +0 -1
  88. package/dist/cjs/services/db/identity-cache-dynamodb-service/helpers.d.ts +0 -26
  89. package/dist/cjs/services/db/identity-cache-dynamodb-service/helpers.js +0 -252
  90. package/dist/cjs/services/db/identity-cache-dynamodb-service/helpers.js.map +0 -1
  91. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-db-service-v2-comparison.spec.js +0 -644
  92. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-db-service-v2-comparison.spec.js.map +0 -1
  93. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-service-v2-comparison.spec.js +0 -323
  94. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-service-v2-comparison.spec.js.map +0 -1
  95. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-service.spec.js +0 -1284
  96. package/dist/esm/__tests__/identity-cache/identity-cache-dynamodb-service.spec.js.map +0 -1
  97. package/dist/esm/__tests__/identity-cache/identity-gateway-service.spec.js +0 -2168
  98. package/dist/esm/__tests__/identity-cache/identity-gateway-service.spec.js.map +0 -1
  99. package/dist/esm/services/db/identity-cache-dynamodb-service/helpers.d.ts +0 -26
  100. package/dist/esm/services/db/identity-cache-dynamodb-service/helpers.js +0 -235
  101. package/dist/esm/services/db/identity-cache-dynamodb-service/helpers.js.map +0 -1
  102. /package/dist/cjs/__tests__/identity-cache/{identity-cache-dynamodb-db-service-v2-comparison.spec.d.ts → identity-client.spec.d.ts} +0 -0
  103. /package/dist/cjs/__tests__/identity-cache/{identity-cache-dynamodb-service-v2-comparison.spec.d.ts → identity-gateway-cache-hit.spec.d.ts} +0 -0
  104. /package/dist/cjs/__tests__/identity-cache/{identity-cache-dynamodb-service.spec.d.ts → identity-gateway-cache-miss.spec.d.ts} +0 -0
  105. /package/dist/cjs/__tests__/identity-cache/{identity-gateway-service.spec.d.ts → identity-gateway-constructor.spec.d.ts} +0 -0
  106. /package/dist/{esm/__tests__/identity-cache/identity-cache-dynamodb-db-service-v2-comparison.spec.d.ts → cjs/__tests__/identity-cache/identity-gateway-edge-cases.spec.d.ts} +0 -0
  107. /package/dist/{esm/__tests__/identity-cache/identity-cache-dynamodb-service-v2-comparison.spec.d.ts → cjs/__tests__/identity-cache/identity-gateway-enqueue-async.spec.d.ts} +0 -0
  108. /package/dist/{esm/__tests__/identity-cache/identity-cache-dynamodb-service.spec.d.ts → cjs/__tests__/identity-cache/identity-gateway-stale-cache.spec.d.ts} +0 -0
  109. /package/dist/esm/__tests__/identity-cache/{identity-gateway-service.spec.d.ts → identity-client.spec.d.ts} +0 -0
@@ -1,1284 +0,0 @@
1
- import { buildEmailPk as buildTrackingTypesEmailPk, buildForcePurgePk as buildTrackingTypesForcePurgePk, buildIdentityMapPk as buildTrackingTypesIdentityMapPk, buildIdentityPk as buildTrackingTypesIdentityPk, buildUserIdPk as buildTrackingTypesUserIdPk, } from '@adtrackify/at-tracking-event-types';
2
- const mockInvokeFunction = jest.fn();
3
- jest.mock('../../clients/index.js', () => ({
4
- DynamoDbClient: {
5
- safeGet: jest.fn(),
6
- safePut: jest.fn(),
7
- safeDelete: jest.fn(),
8
- safeBatchGet: jest.fn(),
9
- safeBatchWrite: jest.fn(),
10
- safeQueryByGSI: jest.fn(),
11
- batchWrite: jest.fn(),
12
- },
13
- LambdaInvokeClient: jest.fn().mockImplementation(() => ({
14
- invokeFunction: mockInvokeFunction,
15
- })),
16
- }));
17
- jest.mock('../../helpers/index.js', () => ({
18
- Logger: {
19
- debug: jest.fn(),
20
- info: jest.fn(),
21
- warn: jest.fn(),
22
- error: jest.fn(),
23
- },
24
- }));
25
- import { buildEmailPk, buildForcePurgePk, buildIdentityMapPk, buildIdentityPk, buildUserIdPk, } from '../../services/db/identity-cache-dynamodb-service/key-builders.js';
26
- import { IdentityGatewayService, } from '../../services/db/identity-cache-dynamodb-service/identity-gateway-service.js';
27
- import { IDENTITY_CACHE_WITH_CACHING_OUTCOME, } from '../../services/db/identity-cache-dynamodb-service/types.js';
28
- import { IdentityCacheDynamoDbDbService } from '../../services/db/identity-cache-dynamodb-service/identity-cache-dynamodb-db-service.js';
29
- import { DynamoDbClient } from '../../clients/index.js';
30
- const mockedDynamoDbClient = DynamoDbClient;
31
- describe('IdentityGatewayService', () => {
32
- beforeEach(() => {
33
- jest.clearAllMocks();
34
- mockedDynamoDbClient.safeGet.mockReset();
35
- mockedDynamoDbClient.safePut.mockReset();
36
- mockedDynamoDbClient.safeDelete.mockReset();
37
- mockedDynamoDbClient.safeBatchGet.mockReset();
38
- mockedDynamoDbClient.safeBatchWrite.mockReset();
39
- mockedDynamoDbClient.safeQueryByGSI.mockReset();
40
- mockedDynamoDbClient.batchWrite.mockReset();
41
- mockInvokeFunction.mockReset();
42
- IdentityGatewayService.lambdaInvokeClient = undefined;
43
- });
44
- describe('identity cache key builders', () => {
45
- it('should build primary identity cache keys', () => {
46
- expect(buildIdentityPk('pixel123', 'identity456')).toBe('identity#pixel123#identity456');
47
- });
48
- it('should build lowercase email pointer keys', () => {
49
- expect(buildEmailPk('pixel123', 'Test@Email.COM')).toBe('email#pixel123#test@email.com');
50
- });
51
- it('should build trimmed user id pointer keys', () => {
52
- expect(buildUserIdPk('pixel123', ' user-1 ')).toBe('user_id#pixel123#user-1');
53
- });
54
- it('should build identity map keys', () => {
55
- expect(buildIdentityMapPk('pixel123', 'identity456')).toBe('identity_map#pixel123#identity456');
56
- });
57
- it('should build force purge keys', () => {
58
- expect(buildForcePurgePk('pixel123', 'identity456')).toBe('force_purge#pixel123#identity456');
59
- });
60
- it('should match the installed tracking-event-types key builder behavior during migration', () => {
61
- expect(buildIdentityPk('pixel123', 'identity456')).toBe(buildTrackingTypesIdentityPk('pixel123', 'identity456'));
62
- expect(buildEmailPk('pixel123', 'Test@Email.COM')).toBe(buildTrackingTypesEmailPk('pixel123', 'Test@Email.COM'));
63
- expect(buildUserIdPk('pixel123', ' user-1 ')).toBe(buildTrackingTypesUserIdPk('pixel123', ' user-1 '));
64
- expect(buildIdentityMapPk('pixel123', 'identity456')).toBe(buildTrackingTypesIdentityMapPk('pixel123', 'identity456'));
65
- expect(buildForcePurgePk('pixel123', 'identity456')).toBe(buildTrackingTypesForcePurgePk('pixel123', 'identity456'));
66
- });
67
- });
68
- describe('getIdentityWithCachingV2', () => {
69
- const pixelId = 'pixel123';
70
- const lambdaArn = 'arn:aws:lambda:us-east-1:123456789:function:identity-private';
71
- describe('cache hit (fresh cache)', () => {
72
- it('should return cached identity without invoking Neptune Lambda', async () => {
73
- const incomingIdentity = {
74
- identityId: 'identity456',
75
- traits: { emails: ['test@email.com'] },
76
- };
77
- const cachedResponse = {
78
- pk: `identity#${pixelId}#identity456`,
79
- pixelId,
80
- identityId: 'identity456',
81
- response: {
82
- identityId: 'identity456',
83
- traits: { emails: ['test@email.com'] },
84
- },
85
- updatedAt: new Date().toISOString(),
86
- };
87
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
88
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
89
- expect(result.identity).toBeDefined();
90
- expect(result.identity?.identityId).toBe('identity456');
91
- expect(mockInvokeFunction).not.toHaveBeenCalled();
92
- });
93
- });
94
- describe('cache miss', () => {
95
- it('should invoke Neptune Lambda and return resolved identity', async () => {
96
- const incomingIdentity = {
97
- identityId: 'identity456',
98
- traits: { emails: ['test@email.com'] },
99
- };
100
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
101
- mockInvokeFunction.mockResolvedValueOnce({
102
- statusCode: 200,
103
- body: JSON.stringify({
104
- identity: {
105
- identityId: 'resolved-identity',
106
- traits: { emails: ['test@email.com', 'resolved@email.com'] },
107
- },
108
- }),
109
- });
110
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
111
- expect(mockInvokeFunction).toHaveBeenCalledWith(lambdaArn, {
112
- pixelId,
113
- context: { identity: incomingIdentity },
114
- });
115
- expect(result.identity?.identityId).toBe('resolved-identity');
116
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
117
- });
118
- it('should return incomingIdentity when Neptune Lambda fails to resolve', async () => {
119
- const incomingIdentity = {
120
- identityId: 'identity456',
121
- traits: { emails: ['test@email.com'] },
122
- };
123
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
124
- mockInvokeFunction.mockResolvedValueOnce({
125
- statusCode: 200,
126
- body: JSON.stringify({ identity: undefined }),
127
- });
128
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
129
- expect(result.identity).toEqual(incomingIdentity);
130
- });
131
- });
132
- describe('cache stale', () => {
133
- it('should invoke Neptune Lambda when cache is stale', async () => {
134
- const incomingIdentity = {
135
- identityId: 'identity456',
136
- traits: { emails: ['old@email.com', 'new@email.com'] },
137
- };
138
- const cachedResponse = {
139
- pk: `identity#${pixelId}#identity456`,
140
- pixelId,
141
- identityId: 'identity456',
142
- response: {
143
- identityId: 'identity456',
144
- traits: { emails: ['old@email.com'] },
145
- },
146
- updatedAt: new Date().toISOString(),
147
- };
148
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
149
- mockInvokeFunction.mockResolvedValueOnce({
150
- statusCode: 200,
151
- body: JSON.stringify({
152
- identity: {
153
- identityId: 'identity456',
154
- traits: { emails: ['old@email.com', 'new@email.com'] },
155
- },
156
- }),
157
- });
158
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
159
- expect(mockInvokeFunction).toHaveBeenCalled();
160
- expect(result.identity?.identityId).toBe('identity456');
161
- expect(result.identity?.traits?.emails).toContain('new@email.com');
162
- });
163
- it('should use merged cache identity when Neptune fails', async () => {
164
- const incomingIdentity = {
165
- identityId: 'identity456',
166
- traits: { emails: ['old@email.com', 'new@email.com'] },
167
- };
168
- const cachedResponse = {
169
- pk: `identity#${pixelId}#identity456`,
170
- pixelId,
171
- identityId: 'identity456',
172
- response: {
173
- identityId: 'identity456',
174
- traits: { emails: ['old@email.com'] },
175
- },
176
- updatedAt: new Date().toISOString(),
177
- };
178
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
179
- mockInvokeFunction.mockResolvedValueOnce({
180
- statusCode: 500,
181
- body: 'Internal Server Error',
182
- });
183
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
184
- expect(result.identity).toBeDefined();
185
- expect(result.identity?.identityId).toBe('identity456');
186
- });
187
- });
188
- describe('Neptune Lambda invocation', () => {
189
- it('should write-back Neptune response to cache on success', async () => {
190
- const incomingIdentity = {
191
- traits: { emails: ['test@email.com'] },
192
- };
193
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
194
- mockInvokeFunction.mockResolvedValueOnce({
195
- statusCode: 200,
196
- body: JSON.stringify({
197
- identity: {
198
- identityId: 'neptune-resolved-id',
199
- traits: { emails: ['test@email.com'] },
200
- },
201
- }),
202
- });
203
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
204
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
205
- expect(result.summary.didWriteCache).toBe(false);
206
- expect(result.identity?.identityId).toBe('neptune-resolved-id');
207
- });
208
- it('should not write to cache when Neptune returns undefined', async () => {
209
- const incomingIdentity = {
210
- traits: { emails: ['test@email.com'] },
211
- };
212
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
213
- mockInvokeFunction.mockResolvedValueOnce({
214
- statusCode: 200,
215
- body: JSON.stringify({ identity: undefined }),
216
- });
217
- await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
218
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
219
- });
220
- });
221
- describe('error handling (fail-open)', () => {
222
- it('should return undefined identity when pixelId is missing', async () => {
223
- const result = await IdentityGatewayService.getIdentityWithCachingV2(null, { identityId: 'test-id' }, lambdaArn);
224
- expect(result.identity).toBeUndefined();
225
- expect(result.summary.outcome).toBe(IDENTITY_CACHE_WITH_CACHING_OUTCOME.INVALID_INPUT);
226
- expect(mockInvokeFunction).not.toHaveBeenCalled();
227
- });
228
- it('should return undefined identity when incomingIdentity is missing', async () => {
229
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, null, lambdaArn);
230
- expect(result.identity).toBeUndefined();
231
- expect(result.summary.outcome).toBe(IDENTITY_CACHE_WITH_CACHING_OUTCOME.INVALID_INPUT);
232
- expect(mockInvokeFunction).not.toHaveBeenCalled();
233
- });
234
- it('should return incomingIdentity when Neptune Lambda throws', async () => {
235
- const incomingIdentity = {
236
- identityId: 'identity456',
237
- traits: { emails: ['test@email.com'] },
238
- };
239
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
240
- mockInvokeFunction.mockRejectedValueOnce(new Error('Lambda invocation failed'));
241
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
242
- expect(result.identity).toEqual(incomingIdentity);
243
- expect(result.summary.outcome).toBe(IDENTITY_CACHE_WITH_CACHING_OUTCOME.NEPTUNE_RESOLUTION_FAILED);
244
- });
245
- it('should not throw when cache write fails after Neptune success', async () => {
246
- const incomingIdentity = {
247
- traits: { emails: ['test@email.com'] },
248
- };
249
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
250
- mockInvokeFunction.mockResolvedValueOnce({
251
- statusCode: 200,
252
- body: JSON.stringify({
253
- identity: {
254
- identityId: 'neptune-resolved-id',
255
- traits: { emails: ['test@email.com'] },
256
- },
257
- }),
258
- });
259
- mockedDynamoDbClient.safeBatchWrite.mockRejectedValueOnce(new Error('DynamoDB write failed'));
260
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
261
- expect(result.identity?.identityId).toBe('neptune-resolved-id');
262
- });
263
- });
264
- });
265
- describe('Key Builders', () => {
266
- describe('buildIdentityPk', () => {
267
- it('should build correct pk for identity lookup', async () => {
268
- const pixelId = 'pixel123';
269
- const identityId = 'identity456';
270
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
271
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, { identityId });
272
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledWith(expect.any(String), 'pk', `identity#${pixelId}#${identityId}`);
273
- });
274
- });
275
- describe('buildEmailPk', () => {
276
- it('should build correct pk for email lookup (lowercase)', async () => {
277
- const pixelId = 'pixel123';
278
- const email = 'Test@Email.COM';
279
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
280
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
281
- traits: { emails: [email] },
282
- });
283
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledWith(expect.any(String), 'pk', `email#${pixelId}#test@email.com`);
284
- });
285
- });
286
- describe('buildUserIdPk (no longer used for lookup)', () => {
287
- it('should NOT lookup by userId (optimization: email-only secondary lookup)', async () => {
288
- const pixelId = 'pixel123';
289
- const userId = ' user789 ';
290
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
291
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
292
- traits: { userIds: [userId] },
293
- });
294
- expect(mockedDynamoDbClient.safeBatchGet).not.toHaveBeenCalled();
295
- });
296
- });
297
- });
298
- describe('getIdentityFromCache', () => {
299
- const pixelId = 'pixel123';
300
- describe('with identityId (primary lookup)', () => {
301
- it('should return cache hit when identity found', async () => {
302
- const identityId = 'identity456';
303
- const cachedResponse = {
304
- pk: `identity#${pixelId}#${identityId}`,
305
- pixelId,
306
- identityId,
307
- response: {
308
- identityId,
309
- traits: { emails: ['test@email.com'] },
310
- },
311
- updatedAt: new Date().toISOString(),
312
- };
313
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
314
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
315
- identityId,
316
- traits: { emails: ['test@email.com'] },
317
- });
318
- expect(result.resolvedIdentity).toBeDefined();
319
- expect(result.resolvedIdentity?.identityId).toBe(identityId);
320
- expect(result.isCacheStale).toBe(false);
321
- });
322
- it('should return cache miss when identity not found', async () => {
323
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
324
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
325
- identityId: 'unknown-id',
326
- });
327
- expect(result.resolvedIdentity).toBeUndefined();
328
- expect(result.isCacheStale).toBe(true);
329
- });
330
- it('should NOT fallback to secondary keys when identityId lookup misses', async () => {
331
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
332
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
333
- identityId: 'unknown-id',
334
- traits: { emails: ['test@email.com'] },
335
- });
336
- expect(mockedDynamoDbClient.safeBatchGet).not.toHaveBeenCalled();
337
- });
338
- it('should detect stale cache when new traits present', async () => {
339
- const identityId = 'identity456';
340
- const cachedResponse = {
341
- pk: `identity#${pixelId}#${identityId}`,
342
- pixelId,
343
- identityId,
344
- response: {
345
- identityId,
346
- traits: { emails: ['old@email.com'] },
347
- },
348
- updatedAt: new Date().toISOString(),
349
- };
350
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
351
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
352
- identityId,
353
- traits: { emails: ['old@email.com', 'new@email.com'] },
354
- });
355
- expect(result.resolvedIdentity).toBeDefined();
356
- expect(result.isCacheStale).toBe(true);
357
- });
358
- });
359
- describe('without identityId (secondary lookup)', () => {
360
- it('should use single get for email lookup (pointer-based)', async () => {
361
- const pointerRecord = {
362
- pk: `email#${pixelId}#test@email.com`,
363
- pixelId,
364
- identityId: 'resolved-id',
365
- gsi1pk: 'resolved-id',
366
- updatedAt: new Date().toISOString(),
367
- };
368
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(pointerRecord);
369
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
370
- traits: { emails: ['test@email.com'], userIds: ['user123'] },
371
- });
372
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledWith(expect.any(String), 'pk', `email#${pixelId}#test@email.com`);
373
- expect(mockedDynamoDbClient.safeBatchGet).not.toHaveBeenCalled();
374
- expect(result.resolvedIdentity).toBeDefined();
375
- expect(result.resolvedIdentity?.traits?.emails).toContain('test@email.com');
376
- expect(result.isCacheStale).toBe(true);
377
- });
378
- it('should return discovered identityId with stale flag for Neptune resolution', async () => {
379
- const pointerRecord = {
380
- pk: `email#${pixelId}#test@email.com`,
381
- pixelId,
382
- identityId: 'discovered-id',
383
- gsi1pk: 'discovered-id',
384
- updatedAt: new Date().toISOString(),
385
- };
386
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(pointerRecord);
387
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
388
- traits: { emails: ['test@email.com'], userIds: ['user123'] },
389
- });
390
- expect(result.resolvedIdentity).toBeDefined();
391
- expect(result.resolvedIdentity?.identityId).toBe('discovered-id');
392
- expect(result.isCacheStale).toBe(true);
393
- });
394
- it('should return cache miss when no secondary keys match', async () => {
395
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
396
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
397
- traits: { emails: ['unknown@email.com'] },
398
- });
399
- expect(result.resolvedIdentity).toBeUndefined();
400
- expect(result.isCacheStale).toBe(true);
401
- });
402
- it('should return cache miss when no emails to lookup', async () => {
403
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
404
- traits: { userIds: ['user123'] },
405
- });
406
- expect(result.resolvedIdentity).toBeUndefined();
407
- expect(result.isCacheStale).toBe(true);
408
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
409
- });
410
- it('should return cache miss when empty traits', async () => {
411
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
412
- traits: {},
413
- });
414
- expect(result.resolvedIdentity).toBeUndefined();
415
- expect(result.isCacheStale).toBe(true);
416
- });
417
- });
418
- describe('error handling (fail-open)', () => {
419
- it('should return cache miss on safeGet error', async () => {
420
- mockedDynamoDbClient.safeGet.mockRejectedValueOnce(new Error('DynamoDB error'));
421
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
422
- identityId: 'test-id',
423
- });
424
- expect(result.resolvedIdentity).toBeUndefined();
425
- expect(result.isCacheStale).toBe(true);
426
- });
427
- it('should return cache miss on safeGet error for email lookup', async () => {
428
- mockedDynamoDbClient.safeGet.mockRejectedValueOnce(new Error('DynamoDB error'));
429
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
430
- traits: { emails: ['test@email.com'] },
431
- });
432
- expect(result.resolvedIdentity).toBeUndefined();
433
- expect(result.isCacheStale).toBe(true);
434
- });
435
- it('should return cache miss for null pixelId', async () => {
436
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(null, {
437
- identityId: 'test-id',
438
- });
439
- expect(result.resolvedIdentity).toBeUndefined();
440
- expect(result.isCacheStale).toBe(true);
441
- });
442
- it('should return cache miss for null incomingIdentity', async () => {
443
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, null);
444
- expect(result.resolvedIdentity).toBeUndefined();
445
- expect(result.isCacheStale).toBe(true);
446
- });
447
- });
448
- });
449
- describe('updateIdentityCache', () => {
450
- const pixelId = 'pixel123';
451
- it('should write identity with email pointer (no full blob on secondary keys)', async () => {
452
- const identity = {
453
- identityId: 'identity456',
454
- traits: {
455
- emails: ['test@email.com', 'other@email.com'],
456
- userIds: ['user1', 'user2'],
457
- },
458
- };
459
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
460
- expect(mockedDynamoDbClient.safeBatchWrite).toHaveBeenCalledTimes(1);
461
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
462
- expect(writtenItems).toHaveLength(3);
463
- const pks = writtenItems.map((item) => item.pk);
464
- expect(pks).toContain(`identity#${pixelId}#identity456`);
465
- expect(pks).toContain(`email#${pixelId}#test@email.com`);
466
- expect(pks).toContain(`email#${pixelId}#other@email.com`);
467
- expect(pks).not.toContain(`user_id#${pixelId}#user1`);
468
- expect(pks).not.toContain(`user_id#${pixelId}#user2`);
469
- const identityItem = writtenItems.find((item) => item.pk.startsWith('identity#'));
470
- expect(identityItem).toBeDefined();
471
- expect(identityItem?.response).toBeDefined();
472
- expect(identityItem?.response.identityId).toBe('identity456');
473
- const emailItem = writtenItems.find((item) => item.pk.startsWith('email#'));
474
- expect(emailItem).toBeDefined();
475
- expect(emailItem?.response).toBeUndefined();
476
- expect(emailItem?.identityId).toBe('identity456');
477
- });
478
- it('should not throw on safeBatchWrite error (fail-open)', async () => {
479
- mockedDynamoDbClient.safeBatchWrite.mockRejectedValueOnce(new Error('DynamoDB error'));
480
- const identity = {
481
- identityId: 'identity456',
482
- traits: { emails: ['test@email.com'] },
483
- };
484
- await expect(IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity)).resolves.toBeUndefined();
485
- });
486
- it('should return early for missing identityId', async () => {
487
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, {});
488
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
489
- });
490
- it('should return early for missing pixelId', async () => {
491
- await IdentityCacheDynamoDbDbService.updateIdentityCache(null, {
492
- identityId: 'test-id',
493
- });
494
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
495
- });
496
- it('should handle identity with no traits', async () => {
497
- const identity = {
498
- identityId: 'identity456',
499
- };
500
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
501
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
502
- expect(writtenItems).toHaveLength(1);
503
- });
504
- it('should include gsi1pk for reverse lookups', async () => {
505
- const identity = {
506
- identityId: 'identity456',
507
- traits: { emails: ['test@email.com'] },
508
- };
509
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
510
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
511
- for (const item of writtenItems) {
512
- expect(item.gsi1pk).toBe('identity456');
513
- }
514
- });
515
- });
516
- describe('getIdentityMap', () => {
517
- it('should return identity map when found', async () => {
518
- const mockMap = {
519
- pk: 'identity_map#pixel123#identity456',
520
- pixelId: 'pixel123',
521
- identityId: 'identity456',
522
- linkedIdentities: ['identity789'],
523
- updatedAt: new Date().toISOString(),
524
- };
525
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(mockMap);
526
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('pixel123', 'identity456');
527
- expect(result).toEqual(mockMap);
528
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledWith(expect.any(String), 'pk', 'identity_map#pixel123#identity456');
529
- });
530
- it('should return undefined when not found', async () => {
531
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
532
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('pixel123', 'identity456');
533
- expect(result).toBeUndefined();
534
- });
535
- it('should return undefined for missing params (fail-open)', async () => {
536
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('', '');
537
- expect(result).toBeUndefined();
538
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
539
- });
540
- it('should return undefined when only pixelId is missing', async () => {
541
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('', 'identity456');
542
- expect(result).toBeUndefined();
543
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
544
- });
545
- it('should return undefined when only identityId is missing', async () => {
546
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('pixel123', '');
547
- expect(result).toBeUndefined();
548
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
549
- });
550
- it('should return undefined on safeGet error (fail-open)', async () => {
551
- mockedDynamoDbClient.safeGet.mockRejectedValueOnce(new Error('DynamoDB error'));
552
- const result = await IdentityCacheDynamoDbDbService.getIdentityMap('pixel123', 'identity456');
553
- expect(result).toBeUndefined();
554
- });
555
- });
556
- describe('getForcePurgeFlag', () => {
557
- it('should return true when flag exists', async () => {
558
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce({
559
- pk: 'force_purge#pixel123#identity456',
560
- });
561
- const result = await IdentityCacheDynamoDbDbService.getForcePurgeFlag('pixel123', 'identity456');
562
- expect(result).toBe(true);
563
- });
564
- it('should return false when flag not found', async () => {
565
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
566
- const result = await IdentityCacheDynamoDbDbService.getForcePurgeFlag('pixel123', 'identity456');
567
- expect(result).toBe(false);
568
- });
569
- it('should return false on error (fail-open)', async () => {
570
- mockedDynamoDbClient.safeGet.mockRejectedValueOnce(new Error('DynamoDB error'));
571
- const result = await IdentityCacheDynamoDbDbService.getForcePurgeFlag('pixel123', 'identity456');
572
- expect(result).toBe(false);
573
- });
574
- it('should return false for missing params', async () => {
575
- const result = await IdentityCacheDynamoDbDbService.getForcePurgeFlag('', '');
576
- expect(result).toBe(false);
577
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
578
- });
579
- });
580
- describe('setForcePurgeFlag', () => {
581
- it('should set flag with TTL', async () => {
582
- await IdentityCacheDynamoDbDbService.setForcePurgeFlag('pixel123', 'identity456');
583
- expect(mockedDynamoDbClient.safePut).toHaveBeenCalledTimes(1);
584
- const putItem = mockedDynamoDbClient.safePut.mock.calls[0][1];
585
- expect(putItem.pk).toBe('force_purge#pixel123#identity456');
586
- expect(putItem.pixelId).toBe('pixel123');
587
- expect(putItem.identityId).toBe('identity456');
588
- expect(putItem.ttl).toBeGreaterThan(0);
589
- });
590
- it('should accept custom TTL', async () => {
591
- await IdentityCacheDynamoDbDbService.setForcePurgeFlag('pixel123', 'identity456', 3600);
592
- const putItem = mockedDynamoDbClient.safePut.mock.calls[0][1];
593
- expect(putItem.ttl).toBeGreaterThan(0);
594
- });
595
- it('should include createdAt timestamp', async () => {
596
- const before = new Date().toISOString();
597
- await IdentityCacheDynamoDbDbService.setForcePurgeFlag('pixel123', 'identity456');
598
- const after = new Date().toISOString();
599
- const putItem = mockedDynamoDbClient.safePut.mock.calls[0][1];
600
- expect(putItem.createdAt).toBeDefined();
601
- expect(putItem.createdAt >= before).toBe(true);
602
- expect(putItem.createdAt <= after).toBe(true);
603
- });
604
- it('should return early when pixelId is missing', async () => {
605
- await IdentityCacheDynamoDbDbService.setForcePurgeFlag('', 'identity456');
606
- expect(mockedDynamoDbClient.safePut).not.toHaveBeenCalled();
607
- });
608
- it('should return early when identityId is missing', async () => {
609
- await IdentityCacheDynamoDbDbService.setForcePurgeFlag('pixel123', '');
610
- expect(mockedDynamoDbClient.safePut).not.toHaveBeenCalled();
611
- });
612
- it('should not throw on error (fail-open)', async () => {
613
- mockedDynamoDbClient.safePut.mockRejectedValueOnce(new Error('DynamoDB error'));
614
- await expect(IdentityCacheDynamoDbDbService.setForcePurgeFlag('pixel123', 'identity456')).resolves.toBeUndefined();
615
- });
616
- });
617
- describe('putIdentityMap', () => {
618
- it('should write identity map', async () => {
619
- await IdentityCacheDynamoDbDbService.putIdentityMap('pixel123', 'identity456', ['linked1', 'linked2']);
620
- expect(mockedDynamoDbClient.safePut).toHaveBeenCalledTimes(1);
621
- const putItem = mockedDynamoDbClient.safePut.mock.calls[0][1];
622
- expect(putItem.pk).toBe('identity_map#pixel123#identity456');
623
- expect(putItem.linkedIdentities).toEqual(['linked1', 'linked2']);
624
- });
625
- it('should return early for missing params', async () => {
626
- await IdentityCacheDynamoDbDbService.putIdentityMap('', '', []);
627
- expect(mockedDynamoDbClient.safePut).not.toHaveBeenCalled();
628
- });
629
- it('should return early when only pixelId is missing', async () => {
630
- await IdentityCacheDynamoDbDbService.putIdentityMap('', 'identity456', ['linked1']);
631
- expect(mockedDynamoDbClient.safePut).not.toHaveBeenCalled();
632
- });
633
- it('should return early when only identityId is missing', async () => {
634
- await IdentityCacheDynamoDbDbService.putIdentityMap('pixel123', '', ['linked1']);
635
- expect(mockedDynamoDbClient.safePut).not.toHaveBeenCalled();
636
- });
637
- it('should include updatedAt timestamp', async () => {
638
- const before = new Date().toISOString();
639
- await IdentityCacheDynamoDbDbService.putIdentityMap('pixel123', 'identity456', ['linked1']);
640
- const after = new Date().toISOString();
641
- const putItem = mockedDynamoDbClient.safePut.mock.calls[0][1];
642
- expect(putItem.updatedAt).toBeDefined();
643
- expect(putItem.updatedAt >= before).toBe(true);
644
- expect(putItem.updatedAt <= after).toBe(true);
645
- });
646
- it('should not throw on safePut error (fail-open)', async () => {
647
- mockedDynamoDbClient.safePut.mockRejectedValueOnce(new Error('DynamoDB error'));
648
- await expect(IdentityCacheDynamoDbDbService.putIdentityMap('pixel123', 'identity456', ['linked1'])).resolves.toBeUndefined();
649
- });
650
- });
651
- describe('deleteIdentityMap', () => {
652
- it('should delete identity map', async () => {
653
- await IdentityCacheDynamoDbDbService.deleteIdentityMap('pixel123', 'identity456');
654
- expect(mockedDynamoDbClient.safeDelete).toHaveBeenCalledWith(expect.any(String), 'pk', 'identity_map#pixel123#identity456');
655
- });
656
- it('should return early for missing params', async () => {
657
- await IdentityCacheDynamoDbDbService.deleteIdentityMap('', '');
658
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
659
- });
660
- it('should return early when only pixelId is missing', async () => {
661
- await IdentityCacheDynamoDbDbService.deleteIdentityMap('', 'identity456');
662
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
663
- });
664
- it('should return early when only identityId is missing', async () => {
665
- await IdentityCacheDynamoDbDbService.deleteIdentityMap('pixel123', '');
666
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
667
- });
668
- it('should not throw on safeDelete error (fail-open)', async () => {
669
- mockedDynamoDbClient.safeDelete.mockRejectedValueOnce(new Error('DynamoDB error'));
670
- await expect(IdentityCacheDynamoDbDbService.deleteIdentityMap('pixel123', 'identity456')).resolves.toBeUndefined();
671
- });
672
- });
673
- describe('deleteForcePurgeFlag', () => {
674
- it('should delete force purge flag', async () => {
675
- await IdentityCacheDynamoDbDbService.deleteForcePurgeFlag('pixel123', 'identity456');
676
- expect(mockedDynamoDbClient.safeDelete).toHaveBeenCalledWith(expect.any(String), 'pk', 'force_purge#pixel123#identity456');
677
- });
678
- it('should return early for missing params', async () => {
679
- await IdentityCacheDynamoDbDbService.deleteForcePurgeFlag('', '');
680
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
681
- });
682
- it('should return early when only pixelId is missing', async () => {
683
- await IdentityCacheDynamoDbDbService.deleteForcePurgeFlag('', 'identity456');
684
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
685
- });
686
- it('should return early when only identityId is missing', async () => {
687
- await IdentityCacheDynamoDbDbService.deleteForcePurgeFlag('pixel123', '');
688
- expect(mockedDynamoDbClient.safeDelete).not.toHaveBeenCalled();
689
- });
690
- it('should not throw on safeDelete error (fail-open)', async () => {
691
- mockedDynamoDbClient.safeDelete.mockRejectedValueOnce(new Error('DynamoDB error'));
692
- await expect(IdentityCacheDynamoDbDbService.deleteForcePurgeFlag('pixel123', 'identity456')).resolves.toBeUndefined();
693
- });
694
- });
695
- describe('deleteIdentityCache', () => {
696
- const pixelId = 'pixel123';
697
- it('should delete all related cache items', async () => {
698
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([
699
- { pk: `identity#${pixelId}#identity456`, pixelId },
700
- { pk: `email#${pixelId}#test@email.com`, pixelId },
701
- ]);
702
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
703
- const incomingIdentity = {
704
- identityId: 'identity456',
705
- traits: { emails: ['test@email.com'] },
706
- };
707
- const resolvedIdentity = {
708
- identityId: 'identity456',
709
- traits: { emails: ['test@email.com', 'other@email.com'] },
710
- };
711
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, incomingIdentity, resolvedIdentity);
712
- expect(mockedDynamoDbClient.safeQueryByGSI).toHaveBeenCalled();
713
- expect(mockedDynamoDbClient.batchWrite).toHaveBeenCalled();
714
- });
715
- it('should return early for missing pixelId', async () => {
716
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(null, {}, {});
717
- expect(mockedDynamoDbClient.safeQueryByGSI).not.toHaveBeenCalled();
718
- });
719
- it('should handle different identityIds in incoming vs resolved', async () => {
720
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([{ pk: `identity#${pixelId}#incoming-id`, pixelId }]);
721
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
722
- const incomingIdentity = {
723
- identityId: 'incoming-id',
724
- traits: { emails: ['test@email.com'] },
725
- };
726
- const resolvedIdentity = {
727
- identityId: 'resolved-id',
728
- traits: { emails: ['test@email.com'] },
729
- };
730
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, incomingIdentity, resolvedIdentity);
731
- expect(mockedDynamoDbClient.safeQueryByGSI).toHaveBeenCalledTimes(2);
732
- });
733
- it('should not throw on batchWrite error (fail-open)', async () => {
734
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([{ pk: `identity#${pixelId}#identity456`, pixelId }]);
735
- mockedDynamoDbClient.batchWrite.mockRejectedValueOnce(new Error('DynamoDB error'));
736
- await expect(IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456' }, { identityId: 'identity456' })).resolves.toBeUndefined();
737
- });
738
- it('should deduplicate keys when same items returned from GSI query', async () => {
739
- const duplicateItem = { pk: `identity#${pixelId}#identity456`, pixelId };
740
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([duplicateItem, duplicateItem]);
741
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
742
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456' }, { identityId: 'identity456' });
743
- expect(mockedDynamoDbClient.batchWrite).toHaveBeenCalled();
744
- const batchWriteCall = mockedDynamoDbClient.batchWrite.mock.calls[0];
745
- expect(batchWriteCall).toBeDefined();
746
- });
747
- });
748
- describe('merge and staleness logic', () => {
749
- const pixelId = 'pixel123';
750
- it('should merge incoming traits with cached traits', async () => {
751
- const cachedResponse = {
752
- pk: `identity#${pixelId}#identity456`,
753
- pixelId,
754
- identityId: 'identity456',
755
- response: {
756
- identityId: 'identity456',
757
- traits: {
758
- emails: ['cached@email.com'],
759
- userIds: ['cached-user'],
760
- },
761
- },
762
- updatedAt: new Date().toISOString(),
763
- };
764
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
765
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
766
- identityId: 'identity456',
767
- traits: {
768
- emails: ['incoming@email.com'],
769
- phones: ['+1234567890'],
770
- },
771
- });
772
- expect(result.resolvedIdentity?.traits?.emails).toContain('incoming@email.com');
773
- expect(result.resolvedIdentity?.traits?.emails).toContain('cached@email.com');
774
- expect(result.resolvedIdentity?.traits?.userIds).toContain('cached-user');
775
- expect(result.resolvedIdentity?.traits?.phones).toContain('+1234567890');
776
- });
777
- it('should mark cache stale when new data present', async () => {
778
- const cachedResponse = {
779
- pk: `identity#${pixelId}#identity456`,
780
- pixelId,
781
- identityId: 'identity456',
782
- response: {
783
- identityId: 'identity456',
784
- traits: { emails: ['cached@email.com'] },
785
- },
786
- updatedAt: new Date().toISOString(),
787
- };
788
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
789
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
790
- identityId: 'identity456',
791
- traits: {
792
- emails: ['cached@email.com', 'new@email.com'],
793
- },
794
- });
795
- expect(result.isCacheStale).toBe(true);
796
- });
797
- it('should not mark cache stale when incoming is subset of cached', async () => {
798
- const cachedResponse = {
799
- pk: `identity#${pixelId}#identity456`,
800
- pixelId,
801
- identityId: 'identity456',
802
- response: {
803
- identityId: 'identity456',
804
- traits: {
805
- emails: ['a@email.com', 'b@email.com'],
806
- userIds: ['user1', 'user2'],
807
- },
808
- },
809
- updatedAt: new Date().toISOString(),
810
- };
811
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
812
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
813
- identityId: 'identity456',
814
- traits: {
815
- emails: ['a@email.com'],
816
- },
817
- });
818
- expect(result.isCacheStale).toBe(false);
819
- });
820
- });
821
- describe('multiple emails pointer writes', () => {
822
- const pixelId = 'pixel123';
823
- it('should write ALL emails as pointer records (not just first)', async () => {
824
- const identity = {
825
- identityId: 'identity456',
826
- traits: {
827
- emails: ['first@email.com', 'second@email.com', 'third@email.com', 'fourth@email.com'],
828
- },
829
- };
830
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
831
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
832
- expect(writtenItems).toHaveLength(5);
833
- const emailPks = writtenItems
834
- .filter((item) => item.pk.startsWith('email#'))
835
- .map((item) => item.pk);
836
- expect(emailPks).toContain(`email#${pixelId}#first@email.com`);
837
- expect(emailPks).toContain(`email#${pixelId}#second@email.com`);
838
- expect(emailPks).toContain(`email#${pixelId}#third@email.com`);
839
- expect(emailPks).toContain(`email#${pixelId}#fourth@email.com`);
840
- });
841
- it('should skip null/undefined emails in array', async () => {
842
- const identity = {
843
- identityId: 'identity456',
844
- traits: {
845
- emails: ['valid@email.com', null, undefined, '', 'another@email.com'],
846
- },
847
- };
848
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
849
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
850
- const emailItems = writtenItems.filter((item) => item.pk.startsWith('email#'));
851
- expect(emailItems).toHaveLength(2);
852
- expect(emailItems.map((e) => e.pk)).toContain(`email#${pixelId}#valid@email.com`);
853
- expect(emailItems.map((e) => e.pk)).toContain(`email#${pixelId}#another@email.com`);
854
- });
855
- it('should handle empty emails array', async () => {
856
- const identity = {
857
- identityId: 'identity456',
858
- traits: { emails: [] },
859
- };
860
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
861
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
862
- expect(writtenItems).toHaveLength(1);
863
- expect(writtenItems[0].pk).toBe(`identity#${pixelId}#identity456`);
864
- });
865
- it('should handle email normalization (lowercase)', async () => {
866
- const identity = {
867
- identityId: 'identity456',
868
- traits: {
869
- emails: ['UPPERCASE@EMAIL.COM', 'MixedCase@Email.Com'],
870
- },
871
- };
872
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
873
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
874
- const emailPks = writtenItems
875
- .filter((item) => item.pk.startsWith('email#'))
876
- .map((item) => item.pk);
877
- expect(emailPks).toContain(`email#${pixelId}#uppercase@email.com`);
878
- expect(emailPks).toContain(`email#${pixelId}#mixedcase@email.com`);
879
- });
880
- });
881
- describe('null/undefined traits edge cases', () => {
882
- const pixelId = 'pixel123';
883
- it('should handle null traits object', async () => {
884
- const identity = {
885
- identityId: 'identity456',
886
- traits: null,
887
- };
888
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
889
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
890
- expect(writtenItems).toHaveLength(1);
891
- });
892
- it('should handle undefined traits object', async () => {
893
- const identity = {
894
- identityId: 'identity456',
895
- traits: undefined,
896
- };
897
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
898
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
899
- expect(writtenItems).toHaveLength(1);
900
- });
901
- it('should return cache miss for identity with null traits in lookup', async () => {
902
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
903
- traits: null,
904
- });
905
- expect(result.resolvedIdentity).toBeUndefined();
906
- expect(result.isCacheStale).toBe(true);
907
- expect(mockedDynamoDbClient.safeGet).not.toHaveBeenCalled();
908
- });
909
- it('should handle identity with undefined emails in traits', async () => {
910
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
911
- traits: { userIds: ['user123'] },
912
- });
913
- expect(result.resolvedIdentity).toBeUndefined();
914
- expect(result.isCacheStale).toBe(true);
915
- });
916
- });
917
- describe('backward compatibility (old full-blob format)', () => {
918
- const pixelId = 'pixel123';
919
- it('should read old format email record with full response blob', async () => {
920
- const oldFormatEmailRecord = {
921
- pk: `email#${pixelId}#test@email.com`,
922
- pixelId,
923
- identityId: 'discovered-id',
924
- gsi1pk: 'discovered-id',
925
- response: {
926
- identityId: 'discovered-id',
927
- traits: { emails: ['test@email.com', 'other@email.com'] },
928
- },
929
- updatedAt: new Date().toISOString(),
930
- };
931
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(oldFormatEmailRecord);
932
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
933
- traits: { emails: ['test@email.com'] },
934
- });
935
- expect(result.resolvedIdentity).toBeDefined();
936
- expect(result.resolvedIdentity?.identityId).toBe('discovered-id');
937
- expect(result.isCacheStale).toBe(true);
938
- });
939
- it('should read new format pointer record (no response blob)', async () => {
940
- const newFormatPointerRecord = {
941
- pk: `email#${pixelId}#test@email.com`,
942
- pixelId,
943
- identityId: 'discovered-id',
944
- gsi1pk: 'discovered-id',
945
- updatedAt: new Date().toISOString(),
946
- };
947
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(newFormatPointerRecord);
948
- const result = await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
949
- traits: { emails: ['test@email.com'] },
950
- });
951
- expect(result.resolvedIdentity).toBeDefined();
952
- expect(result.resolvedIdentity?.identityId).toBe('discovered-id');
953
- expect(result.isCacheStale).toBe(true);
954
- });
955
- it('should trigger Neptune resolution after finding pointer record', async () => {
956
- const lambdaArn = 'arn:aws:lambda:us-east-1:123456789:function:identity-private';
957
- const pointerRecord = {
958
- pk: `email#${pixelId}#test@email.com`,
959
- pixelId,
960
- identityId: 'discovered-id',
961
- gsi1pk: 'discovered-id',
962
- updatedAt: new Date().toISOString(),
963
- };
964
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(pointerRecord);
965
- mockInvokeFunction.mockResolvedValueOnce({
966
- statusCode: 200,
967
- body: JSON.stringify({
968
- identity: {
969
- identityId: 'discovered-id',
970
- traits: { emails: ['test@email.com', 'resolved@email.com'] },
971
- },
972
- }),
973
- });
974
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, { traits: { emails: ['test@email.com'] } }, lambdaArn);
975
- expect(mockInvokeFunction).toHaveBeenCalledWith(lambdaArn, expect.objectContaining({
976
- pixelId,
977
- context: {
978
- identity: expect.objectContaining({
979
- traits: { emails: ['test@email.com'] },
980
- }),
981
- },
982
- }));
983
- expect(result.identity?.identityId).toBe('discovered-id');
984
- });
985
- });
986
- describe('deleteIdentityCache with multiple email pointers', () => {
987
- const pixelId = 'pixel123';
988
- it('should delete all email pointers from both incoming and resolved identities', async () => {
989
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([
990
- { pk: `identity#${pixelId}#identity456`, pixelId },
991
- ]);
992
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
993
- const incomingIdentity = {
994
- identityId: 'identity456',
995
- traits: { emails: ['incoming1@email.com', 'incoming2@email.com'] },
996
- };
997
- const resolvedIdentity = {
998
- identityId: 'identity456',
999
- traits: { emails: ['resolved1@email.com', 'resolved2@email.com', 'resolved3@email.com'] },
1000
- };
1001
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, incomingIdentity, resolvedIdentity);
1002
- expect(mockedDynamoDbClient.batchWrite).toHaveBeenCalled();
1003
- const batchWriteCall = mockedDynamoDbClient.batchWrite.mock.calls[0][0];
1004
- const deleteRequests = batchWriteCall.RequestItems[Object.keys(batchWriteCall.RequestItems)[0]];
1005
- const deletedPks = deleteRequests.map((req) => req.DeleteRequest.Key.pk);
1006
- expect(deletedPks).toContain(`identity#${pixelId}#identity456`);
1007
- expect(deletedPks).toContain(`email#${pixelId}#incoming1@email.com`);
1008
- expect(deletedPks).toContain(`email#${pixelId}#incoming2@email.com`);
1009
- expect(deletedPks).toContain(`email#${pixelId}#resolved1@email.com`);
1010
- expect(deletedPks).toContain(`email#${pixelId}#resolved2@email.com`);
1011
- expect(deletedPks).toContain(`email#${pixelId}#resolved3@email.com`);
1012
- });
1013
- it('should handle empty emails in delete', async () => {
1014
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([
1015
- { pk: `identity#${pixelId}#identity456`, pixelId },
1016
- ]);
1017
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
1018
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456', traits: { emails: [] } }, { identityId: 'identity456' });
1019
- expect(mockedDynamoDbClient.batchWrite).toHaveBeenCalled();
1020
- });
1021
- it('should handle missing traits in delete', async () => {
1022
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([
1023
- { pk: `identity#${pixelId}#identity456`, pixelId },
1024
- ]);
1025
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
1026
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456' }, { identityId: 'identity456' });
1027
- expect(mockedDynamoDbClient.batchWrite).toHaveBeenCalled();
1028
- });
1029
- it('should not delete items from other pixelIds', async () => {
1030
- const otherPixelId = 'other-pixel';
1031
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([
1032
- { pk: `identity#${pixelId}#identity456`, pixelId },
1033
- { pk: `identity#${otherPixelId}#identity456`, pixelId: otherPixelId },
1034
- ]);
1035
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
1036
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456' }, { identityId: 'identity456' });
1037
- const batchWriteCall = mockedDynamoDbClient.batchWrite.mock.calls[0][0];
1038
- const deleteRequests = batchWriteCall.RequestItems[Object.keys(batchWriteCall.RequestItems)[0]];
1039
- const deletedPks = deleteRequests.map((req) => req.DeleteRequest.Key.pk);
1040
- expect(deletedPks).toContain(`identity#${pixelId}#identity456`);
1041
- expect(deletedPks).not.toContain(`identity#${otherPixelId}#identity456`);
1042
- });
1043
- });
1044
- describe('GSI1 query for reverse lookups', () => {
1045
- const pixelId = 'pixel123';
1046
- it('should find all records by identityId via GSI1', async () => {
1047
- const identityId = 'identity456';
1048
- const mockResults = [
1049
- { pk: `identity#${pixelId}#${identityId}`, pixelId, identityId },
1050
- { pk: `email#${pixelId}#email1@test.com`, pixelId, identityId },
1051
- { pk: `email#${pixelId}#email2@test.com`, pixelId, identityId },
1052
- ];
1053
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue(mockResults);
1054
- mockedDynamoDbClient.batchWrite.mockResolvedValue({ $metadata: {} });
1055
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId }, { identityId });
1056
- expect(mockedDynamoDbClient.safeQueryByGSI).toHaveBeenCalledWith(expect.any(String), expect.any(String), 'gsi1pk', identityId);
1057
- });
1058
- it('should handle GSI1 query returning empty results', async () => {
1059
- mockedDynamoDbClient.safeQueryByGSI.mockResolvedValue([]);
1060
- await IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'unknown-id' }, { identityId: 'unknown-id' });
1061
- expect(mockedDynamoDbClient.batchWrite).not.toHaveBeenCalled();
1062
- });
1063
- it('should handle GSI1 query error gracefully (fail-open)', async () => {
1064
- mockedDynamoDbClient.safeQueryByGSI.mockRejectedValueOnce(new Error('GSI query failed'));
1065
- await expect(IdentityCacheDynamoDbDbService.deleteIdentityCache(pixelId, { identityId: 'identity456' }, { identityId: 'identity456' })).resolves.toBeUndefined();
1066
- });
1067
- });
1068
- describe('email pointer lookups', () => {
1069
- const pixelId = 'pixel123';
1070
- it('should only lookup first email (optimization)', async () => {
1071
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1072
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
1073
- traits: { emails: ['first@email.com', 'second@email.com', 'third@email.com'] },
1074
- });
1075
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledTimes(1);
1076
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalledWith(expect.any(String), 'pk', `email#${pixelId}#first@email.com`);
1077
- });
1078
- it('should handle whitespace in email during lookup', async () => {
1079
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1080
- await IdentityCacheDynamoDbDbService.getIdentityFromCache(pixelId, {
1081
- traits: { emails: [' test@email.com '] },
1082
- });
1083
- expect(mockedDynamoDbClient.safeGet).toHaveBeenCalled();
1084
- });
1085
- });
1086
- describe('fail-open behavior comprehensive', () => {
1087
- const pixelId = 'pixel123';
1088
- const lambdaArn = 'arn:aws:lambda:us-east-1:123456789:function:identity-private';
1089
- it('should return incoming identity when Neptune throws (fail-open fallback)', async () => {
1090
- const incomingIdentity = {
1091
- identityId: 'identity456',
1092
- traits: { emails: ['test@email.com'] },
1093
- };
1094
- mockedDynamoDbClient.safeGet.mockRejectedValueOnce(new Error('DynamoDB down'));
1095
- mockInvokeFunction.mockRejectedValueOnce(new Error('Lambda down'));
1096
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1097
- expect(result.identity).toEqual(incomingIdentity);
1098
- });
1099
- it('should continue when cache write fails after Neptune success', async () => {
1100
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1101
- mockInvokeFunction.mockResolvedValueOnce({
1102
- statusCode: 200,
1103
- body: JSON.stringify({
1104
- identity: { identityId: 'resolved-id', traits: { emails: ['test@email.com'] } },
1105
- }),
1106
- });
1107
- mockedDynamoDbClient.safeBatchWrite.mockRejectedValueOnce(new Error('Write failed'));
1108
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, { traits: { emails: ['test@email.com'] } }, lambdaArn);
1109
- expect(result.identity?.identityId).toBe('resolved-id');
1110
- });
1111
- it('should return incoming identity when Neptune returns non-200', async () => {
1112
- const incomingIdentity = {
1113
- identityId: 'incoming-id',
1114
- traits: { emails: ['test@email.com'] },
1115
- };
1116
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1117
- mockInvokeFunction.mockResolvedValueOnce({
1118
- statusCode: 500,
1119
- body: 'Internal Server Error',
1120
- });
1121
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1122
- expect(result.identity).toEqual(incomingIdentity);
1123
- });
1124
- it('should return incoming identity when Neptune returns malformed JSON (fail-open)', async () => {
1125
- const incomingIdentity = {
1126
- identityId: 'test-id',
1127
- traits: { emails: ['test@email.com'] },
1128
- };
1129
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1130
- mockInvokeFunction.mockResolvedValueOnce({
1131
- statusCode: 200,
1132
- body: 'not valid json',
1133
- });
1134
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1135
- expect(result.identity).toEqual(incomingIdentity);
1136
- });
1137
- it('should return undefined identity only when both inputs are invalid', async () => {
1138
- const result = await IdentityGatewayService.getIdentityWithCachingV2(null, null, lambdaArn);
1139
- expect(result.identity).toBeUndefined();
1140
- expect(result.summary.outcome).toBe(IDENTITY_CACHE_WITH_CACHING_OUTCOME.INVALID_INPUT);
1141
- });
1142
- });
1143
- describe('getIdentityWithCachingV2', () => {
1144
- const pixelId = 'pixel123';
1145
- const lambdaArn = 'arn:aws:lambda:us-east-1:123456789:function:identity-private';
1146
- it('should return invalid input summary without resolving', async () => {
1147
- const result = await IdentityGatewayService.getIdentityWithCachingV2('', { identityId: 'identity456' }, lambdaArn);
1148
- expect(result.identity).toBeUndefined();
1149
- expect(result.summary).toEqual(expect.objectContaining({
1150
- outcome: IDENTITY_CACHE_WITH_CACHING_OUTCOME.INVALID_INPUT,
1151
- didReadCache: false,
1152
- didCallNeptune: false,
1153
- didWriteCache: false,
1154
- }));
1155
- expect(mockInvokeFunction).not.toHaveBeenCalled();
1156
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
1157
- });
1158
- it('should return fresh cache hit summary without invoking Neptune', async () => {
1159
- const incomingIdentity = {
1160
- identityId: 'identity456',
1161
- traits: { emails: ['test@email.com'] },
1162
- };
1163
- const cachedResponse = {
1164
- pk: `identity#${pixelId}#identity456`,
1165
- pixelId,
1166
- identityId: 'identity456',
1167
- response: {
1168
- identityId: 'identity456',
1169
- traits: { emails: ['test@email.com'] },
1170
- },
1171
- updatedAt: new Date().toISOString(),
1172
- };
1173
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
1174
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1175
- expect(result.identity?.identityId).toBe('identity456');
1176
- expect(result.summary).toEqual(expect.objectContaining({
1177
- outcome: IDENTITY_CACHE_WITH_CACHING_OUTCOME.FRESH_CACHE_HIT,
1178
- pixelId,
1179
- identityId: 'identity456',
1180
- didReadCache: true,
1181
- didCallNeptune: false,
1182
- didWriteCache: false,
1183
- cachedIdentity: cachedResponse.response,
1184
- incomingIdentity,
1185
- resolvedIdentity: expect.objectContaining({ identityId: 'identity456' }),
1186
- }));
1187
- expect(mockInvokeFunction).not.toHaveBeenCalled();
1188
- });
1189
- it('should return Neptune write-back summary on cache miss', async () => {
1190
- const incomingIdentity = {
1191
- identityId: 'identity456',
1192
- traits: { emails: ['test@email.com'] },
1193
- };
1194
- const neptuneIdentity = {
1195
- identityId: 'resolved-identity',
1196
- traits: { emails: ['test@email.com', 'resolved@email.com'] },
1197
- };
1198
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(null);
1199
- mockInvokeFunction.mockResolvedValueOnce({
1200
- statusCode: 200,
1201
- body: JSON.stringify({ identity: neptuneIdentity }),
1202
- });
1203
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1204
- expect(result.identity).toEqual(neptuneIdentity);
1205
- expect(result.summary).toEqual(expect.objectContaining({
1206
- outcome: IDENTITY_CACHE_WITH_CACHING_OUTCOME.NEPTUNE_RESOLVED,
1207
- pixelId,
1208
- identityId: 'resolved-identity',
1209
- didReadCache: true,
1210
- didCallNeptune: true,
1211
- didWriteCache: false,
1212
- incomingIdentity,
1213
- identityForNeptune: incomingIdentity,
1214
- neptuneIdentity,
1215
- traitChanges: { reason: { resolved: 'no cached identity' } },
1216
- }));
1217
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
1218
- });
1219
- it('should return Neptune skip-write summary when Neptune matches cached identity', async () => {
1220
- const incomingIdentity = {
1221
- identityId: 'identity456',
1222
- traits: { emails: ['old@email.com', 'new@email.com'] },
1223
- };
1224
- const cachedIdentity = {
1225
- identityId: 'identity456',
1226
- traits: { emails: ['old@email.com'] },
1227
- };
1228
- const cachedResponse = {
1229
- pk: `identity#${pixelId}#identity456`,
1230
- pixelId,
1231
- identityId: 'identity456',
1232
- response: cachedIdentity,
1233
- updatedAt: new Date().toISOString(),
1234
- };
1235
- mockedDynamoDbClient.safeGet.mockResolvedValueOnce(cachedResponse);
1236
- mockInvokeFunction.mockResolvedValueOnce({
1237
- statusCode: 200,
1238
- body: JSON.stringify({ identity: cachedIdentity }),
1239
- });
1240
- const result = await IdentityGatewayService.getIdentityWithCachingV2(pixelId, incomingIdentity, lambdaArn);
1241
- expect(result.identity).toEqual(cachedIdentity);
1242
- expect(result.summary).toEqual(expect.objectContaining({
1243
- outcome: IDENTITY_CACHE_WITH_CACHING_OUTCOME.NEPTUNE_RESOLVED,
1244
- pixelId,
1245
- identityId: 'identity456',
1246
- didReadCache: true,
1247
- didCallNeptune: true,
1248
- didWriteCache: false,
1249
- cachedIdentity,
1250
- incomingIdentity,
1251
- identityForNeptune: expect.objectContaining({ identityId: 'identity456' }),
1252
- neptuneIdentity: cachedIdentity,
1253
- }));
1254
- expect(mockedDynamoDbClient.safeBatchWrite).not.toHaveBeenCalled();
1255
- });
1256
- });
1257
- describe('gsi1pk attribute on all records', () => {
1258
- const pixelId = 'pixel123';
1259
- it('should include gsi1pk on identity record', async () => {
1260
- const identity = {
1261
- identityId: 'identity456',
1262
- traits: { emails: ['test@email.com'] },
1263
- };
1264
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
1265
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
1266
- const identityItem = writtenItems.find((item) => item.pk.startsWith('identity#'));
1267
- expect(identityItem).toBeDefined();
1268
- expect(identityItem?.gsi1pk).toBe('identity456');
1269
- });
1270
- it('should include gsi1pk on all email pointer records', async () => {
1271
- const identity = {
1272
- identityId: 'identity456',
1273
- traits: { emails: ['email1@test.com', 'email2@test.com'] },
1274
- };
1275
- await IdentityCacheDynamoDbDbService.updateIdentityCache(pixelId, identity);
1276
- const writtenItems = mockedDynamoDbClient.safeBatchWrite.mock.calls[0][1];
1277
- const emailItems = writtenItems.filter((item) => item.pk.startsWith('email#'));
1278
- for (const emailItem of emailItems) {
1279
- expect(emailItem.gsi1pk).toBe('identity456');
1280
- }
1281
- });
1282
- });
1283
- });
1284
- //# sourceMappingURL=identity-cache-dynamodb-service.spec.js.map