def test_synchronize_segment(self, mocker):
"""Test particular segment update."""
split_storage = mocker.Mock(spec=SplitStorage)
storage = mocker.Mock(spec=SegmentStorage)
def change_number_mock(segment_name):
if change_number_mock._count_a == 0:
change_number_mock._count_a = 1
return -1
return 123
change_number_mock._count_a = 0
storage.get_change_number.side_effect = change_number_mock
def fetch_segment_mock(segment_name, change_number):
if fetch_segment_mock._count_a == 0:
fetch_segment_mock._count_a = 1
return {'name': 'segmentA', 'added': ['key1', 'key2', 'key3'], 'removed': [],
'since': -1, 'till': 123}
return {'added': [], 'removed': [], 'since': 123, 'till': 123}
fetch_segment_mock._count_a = 0
api = mocker.Mock()
api.fetch_segment.side_effect = fetch_segment_mock
segments_synchronizer = SegmentSynchronizer(api, split_storage, storage)
segments_synchronizer.synchronize_segment('segmentA')
api_calls = [call for call in api.fetch_segment.mock_calls]
assert mocker.call('segmentA', -1) in api_calls
assert mocker.call('segmentA', 123) in api_calls
def test_synchronize_segment_cdn(self, mocker):
"""Test particular segment update cdn bypass."""
mocker.patch(
'splitio.sync.segment._ON_DEMAND_FETCH_BACKOFF_MAX_RETRIES', new=3)
from splitio.sync.segment import SegmentSynchronizer
split_storage = mocker.Mock(spec=SplitStorage)
storage = mocker.Mock(spec=SegmentStorage)
def change_number_mock(segment_name):
change_number_mock._count_a += 1
if change_number_mock._count_a == 1:
return -1
elif change_number_mock._count_a >= 2 and change_number_mock._count_a <= 3:
return 123
elif change_number_mock._count_a <= 7:
return 1234
return 12345 # Return proper cn for CDN Bypass
change_number_mock._count_a = 0
storage.get_change_number.side_effect = change_number_mock
def fetch_segment_mock(segment_name, change_number, fetch_options):
fetch_segment_mock._count_a += 1
if fetch_segment_mock._count_a == 1:
return {
'name': 'segmentA',
'added': ['key1', 'key2', 'key3'],
'removed': [],
'since': -1,
'till': 123
}
elif fetch_segment_mock._count_a == 2:
return {'added': [], 'removed': [], 'since': 123, 'till': 123}
elif fetch_segment_mock._count_a == 3:
return {'added': [], 'removed': [], 'since': 123, 'till': 1234}
elif fetch_segment_mock._count_a >= 4 and fetch_segment_mock._count_a <= 6:
return {
'added': [],
'removed': [],
'since': 1234,
'till': 1234
}
elif fetch_segment_mock._count_a == 7:
return {
'added': [],
'removed': [],
'since': 1234,
'till': 12345
}
return {'added': [], 'removed': [], 'since': 12345, 'till': 12345}
fetch_segment_mock._count_a = 0
api = mocker.Mock()
api.fetch_segment.side_effect = fetch_segment_mock
segments_synchronizer = SegmentSynchronizer(api, split_storage,
storage)
segments_synchronizer.synchronize_segment('segmentA')
assert mocker.call('segmentA', -1,
FetchOptions(True)) in api.fetch_segment.mock_calls
assert mocker.call('segmentA', 123,
FetchOptions(True)) in api.fetch_segment.mock_calls
segments_synchronizer._backoff = Backoff(1, 0.1)
segments_synchronizer.synchronize_segment('segmentA', 12345)
assert mocker.call('segmentA', 12345,
FetchOptions(True,
1234)) in api.fetch_segment.mock_calls
assert len(
api.fetch_segment.mock_calls
) == 8 # 2 ok + BACKOFF(2 since==till + 2 re-attempts) + CDN(2 since==till)
