|
9 | 9 | from splitio.tasks.segment_sync import SegmentSynchronizationTask |
10 | 10 | from splitio.tasks.impressions_sync import ImpressionsSyncTask, ImpressionsCountSyncTask |
11 | 11 | from splitio.tasks.events_sync import EventsSyncTask |
12 | | -from splitio.sync.split import SplitSynchronizer, LocalSplitSynchronizer |
| 12 | +from splitio.sync.split import SplitSynchronizer, LocalSplitSynchronizer, LocalhostMode |
13 | 13 | from splitio.sync.segment import SegmentSynchronizer, LocalSegmentSynchronizer |
14 | 14 | from splitio.sync.impression import ImpressionSynchronizer, ImpressionsCountSynchronizer |
15 | 15 | from splitio.sync.event import EventSynchronizer |
@@ -349,7 +349,7 @@ def test_synchronize_splits(self, mocker): |
349 | 349 | split_sync = LocalSplitSynchronizer(mocker.Mock(), mocker.Mock(), mocker.Mock()) |
350 | 350 | segment_sync = LocalSegmentSynchronizer(mocker.Mock(), mocker.Mock(), mocker.Mock()) |
351 | 351 | synchronizers = SplitSynchronizers(split_sync, segment_sync, None, None, None) |
352 | | - local_synchronizer = LocalhostSynchronizer(synchronizers, mocker.Mock()) |
| 352 | + local_synchronizer = LocalhostSynchronizer(synchronizers, mocker.Mock(), mocker.Mock()) |
353 | 353 |
|
354 | 354 | def synchronize_splits(*args, **kwargs): |
355 | 355 | return ["segmentA", "segmentB"] |
@@ -390,7 +390,7 @@ def segment_task_stop(*args, **kwargs): |
390 | 390 | self.segment_task_stop_called = True |
391 | 391 | segment_task.stop = segment_task_stop |
392 | 392 |
|
393 | | - local_synchronizer = LocalhostSynchronizer(synchronizers, tasks) |
| 393 | + local_synchronizer = LocalhostSynchronizer(synchronizers, tasks, LocalhostMode.JSON) |
394 | 394 | local_synchronizer.start_periodic_fetching() |
395 | 395 | assert(self.split_task_start_called) |
396 | 396 | assert(self.segment_task_start_called) |
|
0 commit comments