Skip to content

Commit cb2f284

Browse files
committed
[AI-FSSDK] [FSSDK-12721] Skip ODP identify event for single-identifier users
1 parent 013ddf0 commit cb2f284

7 files changed

Lines changed: 76 additions & 17 deletions

File tree

optimizely/odp/odp_event_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -267,9 +267,9 @@ def dispatch(self, event: OdpEvent) -> None:
267267
except Full:
268268
self.logger.warning(Errors.ODP_EVENT_FAILED.format("Queue is full"))
269269

270-
def identify_user(self, user_id: str) -> None:
270+
def identify_user(self, identifiers: dict[str, str]) -> None:
271271
self.send_event(OdpManagerConfig.EVENT_TYPE, 'identified',
272-
{OdpManagerConfig.KEY_FOR_USER_ID: user_id}, {})
272+
identifiers, {})
273273

274274
def update_config(self) -> None:
275275
"""Adds update config signal to event_queue."""

optimizely/odp/odp_manager.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,15 +73,20 @@ def fetch_qualified_segments(self, user_id: str, options: list[str]) -> Optional
7373

7474
return self.segment_manager.fetch_qualified_segments(user_key, user_value, options)
7575

76-
def identify_user(self, user_id: str) -> None:
76+
def identify_user(self, identifiers: dict[str, str]) -> None:
7777
if not self.enabled or not self.event_manager:
7878
self.logger.debug('ODP identify event is not dispatched (ODP disabled).')
7979
return
8080
if self.odp_config.odp_state() == OdpConfigState.NOT_INTEGRATED:
8181
self.logger.debug('ODP identify event is not dispatched (ODP not integrated).')
8282
return
8383

84-
self.event_manager.identify_user(user_id)
84+
valid_identifiers = {k: v for k, v in identifiers.items() if v}
85+
if len(valid_identifiers) < 2:
86+
self.logger.debug('ODP identify event is not dispatched (only one identifier provided).')
87+
return
88+
89+
self.event_manager.identify_user(valid_identifiers)
8590

8691
def send_event(self, type: str, action: str, identifiers: dict[str, str], data: dict[str, Any]) -> None:
8792
"""

optimizely/optimizely.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1541,7 +1541,7 @@ def _update_odp_config_on_datafile_update(self) -> None:
15411541
config.all_segments
15421542
)
15431543

1544-
def _identify_user(self, user_id: str) -> None:
1544+
def _identify_user(self, identifiers: dict[str, str]) -> None:
15451545
if not self.is_valid:
15461546
self.logger.error(enums.Errors.INVALID_OPTIMIZELY.format('identify_user'))
15471547
return
@@ -1551,7 +1551,7 @@ def _identify_user(self, user_id: str) -> None:
15511551
self.logger.error(enums.Errors.INVALID_PROJECT_CONFIG.format('identify_user'))
15521552
return
15531553

1554-
self.odp_manager.identify_user(user_id)
1554+
self.odp_manager.identify_user(identifiers)
15551555

15561556
def _fetch_qualified_segments(self, user_id: str, options: Optional[list[str]] = None) -> Optional[list[str]]:
15571557
if not self.is_valid:

optimizely/optimizely_user_context.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from typing import TYPE_CHECKING, Any, Callable, Optional, NewType, Dict
2020

2121
from optimizely.decision import optimizely_decision
22+
from optimizely.helpers.enums import OdpManagerConfig
2223

2324
if TYPE_CHECKING:
2425
# prevent circular dependency by skipping import at runtime
@@ -73,7 +74,8 @@ def __init__(
7374
] = {}
7475

7576
if self.client and identify:
76-
self.client._identify_user(user_id)
77+
identifiers = {OdpManagerConfig.KEY_FOR_USER_ID: user_id}
78+
self.client._identify_user(identifiers)
7779

7880
class OptimizelyDecisionContext:
7981
""" Using class with attributes here instead of namedtuple because

tests/test_odp_manager.py

Lines changed: 59 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_configurations_disable_odp(self):
4848
mock_logger.reset_mock()
4949

5050
# these call should be dropped gracefully with None
51-
manager.identify_user('user1')
51+
manager.identify_user({'fs_user_id': 'user1', 'email': 'user1@example.com'})
5252

5353
manager.send_event('t1', 'a1', {}, {})
5454
mock_logger.error.assert_called_once_with('ODP is not enabled.')
@@ -120,16 +120,68 @@ def test_fetch_qualified_segments__seg_cache_and_seg_mgr_are_none(self):
120120
mock_logger.error.assert_not_called()
121121
mock_fetch_qualif_segments.assert_called_once_with('fs_user_id', 'user1', [])
122122

123+
def test_identify_user_single_identifier_skipped(self):
124+
"""Single identifier should NOT dispatch an identify event."""
125+
mock_logger = mock.MagicMock()
126+
event_manager = OdpEventManager(mock_logger, OdpEventApiManager())
127+
128+
manager = OdpManager(False, LRUCache(10, 20), event_manager=event_manager, logger=mock_logger)
129+
manager.update_odp_config('key1', 'host1', [])
130+
131+
with mock.patch.object(event_manager, 'identify_user') as mock_identify_user:
132+
manager.identify_user({'fs_user_id': 'user1'})
133+
134+
mock_identify_user.assert_not_called()
135+
mock_logger.debug.assert_any_call('ODP identify event is not dispatched (only one identifier provided).')
136+
137+
def test_identify_user_empty_values_not_counted(self):
138+
"""Identifiers with empty or null values should not count toward the minimum."""
139+
mock_logger = mock.MagicMock()
140+
event_manager = OdpEventManager(mock_logger, OdpEventApiManager())
141+
142+
manager = OdpManager(False, LRUCache(10, 20), event_manager=event_manager, logger=mock_logger)
143+
manager.update_odp_config('key1', 'host1', [])
144+
145+
with mock.patch.object(event_manager, 'identify_user') as mock_identify_user:
146+
manager.identify_user({'fs_user_id': 'user1', 'email': '', 'vuid': ''})
147+
148+
mock_identify_user.assert_not_called()
149+
mock_logger.debug.assert_any_call('ODP identify event is not dispatched (only one identifier provided).')
150+
151+
def test_identify_user_multiple_identifiers_sent(self):
152+
"""Multiple valid identifiers should dispatch an identify event."""
153+
mock_logger = mock.MagicMock()
154+
event_manager = OdpEventManager(mock_logger, OdpEventApiManager())
155+
156+
manager = OdpManager(False, LRUCache(10, 20), event_manager=event_manager, logger=mock_logger)
157+
manager.update_odp_config('key1', 'host1', [])
158+
159+
with mock.patch.object(event_manager, 'dispatch') as mock_dispatch_event:
160+
manager.identify_user({'fs_user_id': 'user1', 'email': 'user1@example.com'})
161+
162+
mock_dispatch_event.assert_called_once_with({
163+
'type': 'fullstack',
164+
'action': 'identified',
165+
'identifiers': {'fs_user_id': 'user1', 'email': 'user1@example.com'},
166+
'data': {
167+
'idempotence_id': mock.ANY,
168+
'data_source_type': 'sdk',
169+
'data_source': 'python-sdk',
170+
'data_source_version': version.__version__
171+
}})
172+
mock_logger.error.assert_not_called()
173+
123174
def test_identify_user_datafile_not_ready(self):
175+
"""When datafile is not ready but ODP config allows, identifiers with 2+ valid entries should forward."""
124176
mock_logger = mock.MagicMock()
125177
event_manager = OdpEventManager(OdpConfig(), mock_logger)
126178

127179
manager = OdpManager(False, OptimizelySegmentsCache, event_manager=event_manager, logger=mock_logger)
128180

129181
with mock.patch.object(event_manager, 'identify_user') as mock_identify_user:
130-
manager.identify_user('user1')
182+
manager.identify_user({'fs_user_id': 'user1', 'email': 'user1@example.com'})
131183

132-
mock_identify_user.assert_called_once_with('user1')
184+
mock_identify_user.assert_called_once_with({'fs_user_id': 'user1', 'email': 'user1@example.com'})
133185
mock_logger.error.assert_not_called()
134186

135187
def test_identify_user_odp_integrated(self):
@@ -140,12 +192,12 @@ def test_identify_user_odp_integrated(self):
140192
manager.update_odp_config('key1', 'host1', [])
141193

142194
with mock.patch.object(event_manager, 'dispatch') as mock_dispatch_event:
143-
manager.identify_user('user1')
195+
manager.identify_user({'fs_user_id': 'user1', 'vuid': 'vuid123'})
144196

145197
mock_dispatch_event.assert_called_once_with({
146198
'type': 'fullstack',
147199
'action': 'identified',
148-
'identifiers': {'fs_user_id': 'user1'},
200+
'identifiers': {'fs_user_id': 'user1', 'vuid': 'vuid123'},
149201
'data': {
150202
'idempotence_id': mock.ANY,
151203
'data_source_type': 'sdk',
@@ -162,7 +214,7 @@ def test_identify_user_odp_not_integrated(self):
162214
manager.update_odp_config(None, None, [])
163215

164216
with mock.patch.object(event_manager, 'dispatch') as mock_dispatch_event:
165-
manager.identify_user('user1')
217+
manager.identify_user({'fs_user_id': 'user1', 'email': 'user1@example.com'})
166218

167219
mock_dispatch_event.assert_not_called()
168220
mock_logger.error.assert_not_called()
@@ -176,7 +228,7 @@ def test_identify_user_odp_disabled(self):
176228
manager.enabled = False
177229

178230
with mock.patch.object(event_manager, 'identify_user') as mock_identify_user:
179-
manager.identify_user('user1')
231+
manager.identify_user({'fs_user_id': 'user1', 'email': 'user1@example.com'})
180232

181233
mock_identify_user.assert_not_called()
182234
mock_logger.error.assert_not_called()

tests/test_optimizely.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5343,7 +5343,7 @@ def test_send_identify_event__when_called_with_odp_enabled(self):
53435343
with mock.patch.object(client, '_identify_user') as identify:
53445344
client.create_user_context('user-id')
53455345

5346-
identify.assert_called_once_with('user-id')
5346+
identify.assert_called_once_with({'fs_user_id': 'user-id'})
53475347
mock_logger.error.assert_not_called()
53485348
client.close()
53495349

tests/test_user_context.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2094,7 +2094,7 @@ def test_send_identify_event_when_user_context_created(self):
20942094
with mock.patch.object(client, '_identify_user') as identify:
20952095
OptimizelyUserContext(client, mock_logger, 'user-id')
20962096

2097-
identify.assert_called_once_with('user-id')
2097+
identify.assert_called_once_with({'fs_user_id': 'user-id'})
20982098
mock_logger.error.assert_not_called()
20992099
client.close()
21002100

@@ -2104,7 +2104,7 @@ def test_identify_is_skipped_with_decisions(self):
21042104
with mock.patch.object(client, '_identify_user') as identify:
21052105
user_context = OptimizelyUserContext(client, mock_logger, 'user-id')
21062106

2107-
identify.assert_called_once_with('user-id')
2107+
identify.assert_called_once_with({'fs_user_id': 'user-id'})
21082108
mock_logger.error.assert_not_called()
21092109

21102110
with mock.patch.object(client, '_identify_user') as identify:

0 commit comments

Comments
 (0)