@@ -476,9 +476,12 @@ def test_get_variation__experiment_not_running(self):
476
476
def test_get_variation__bucketing_id_provided (self ):
477
477
""" Test that get_variation calls bucket with correct bucketing ID if provided. """
478
478
479
- user = optimizely_user_context .OptimizelyUserContext (optimizely_client = None , user_id = "test_user" ,
480
- user_attributes = {"random_key" : "random_value" ,
481
- "$opt_bucketing_id" : "user_bucket_value" , })
479
+ user = optimizely_user_context .OptimizelyUserContext (optimizely_client = None ,
480
+ user_id = "test_user" ,
481
+ user_attributes = {
482
+ "random_key" : "random_value" ,
483
+ "$opt_bucketing_id" : "user_bucket_value" ,
484
+ })
482
485
experiment = self .project_config .get_experiment_from_key ("test_experiment" )
483
486
with mock .patch (
484
487
"optimizely.decision_service.DecisionService.get_forced_variation" ,
@@ -1312,8 +1315,9 @@ def test_get_variation_for_feature__returns_variation_if_user_not_in_experiment_
1312
1315
with mock .patch (
1313
1316
"optimizely.helpers.audience.does_user_meet_audience_conditions" ,
1314
1317
side_effect = [[False , []], [True , []]],
1315
- ) as mock_audience_check , self .mock_decision_logger as mock_decision_service_logging , mock .patch (
1316
- "optimizely.bucketer.Bucketer.bucket" , return_value = [expected_variation , []]):
1318
+ ) as mock_audience_check , \
1319
+ self .mock_decision_logger as mock_decision_service_logging , mock .patch (
1320
+ "optimizely.bucketer.Bucketer.bucket" , return_value = [expected_variation , []]):
1317
1321
decision , _ = self .decision_service .get_variation_for_feature (
1318
1322
self .project_config , feature , user
1319
1323
)
@@ -1444,7 +1448,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_mutex_group
1444
1448
"group_2_exp_1" , "38901"
1445
1449
)
1446
1450
with mock .patch (
1447
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1451
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1448
1452
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1449
1453
variation_received , _ = self .decision_service .get_variation_for_feature (
1450
1454
self .project_config , feature , user
@@ -1477,7 +1481,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_mutex_group
1477
1481
"group_2_exp_2" , "38905"
1478
1482
)
1479
1483
with mock .patch (
1480
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1484
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1481
1485
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1482
1486
variation_received , _ = self .decision_service .get_variation_for_feature (
1483
1487
self .project_config , feature , user
@@ -1508,7 +1512,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_mutex_group
1508
1512
)
1509
1513
1510
1514
with mock .patch (
1511
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 6500 ) as mock_generate_bucket_value , \
1515
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 6500 ) as mock_generate_bucket_value , \
1512
1516
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1513
1517
variation_received , _ = self .decision_service .get_variation_for_feature (
1514
1518
self .project_config , feature , user
@@ -1534,7 +1538,7 @@ def test_get_variation_for_feature__returns_variation_for_rollout_in_mutex_group
1534
1538
feature = self .project_config .get_feature_from_key ("test_feature_in_exclusion_group" )
1535
1539
1536
1540
with mock .patch (
1537
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 8000 ) as mock_generate_bucket_value , \
1541
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 8000 ) as mock_generate_bucket_value , \
1538
1542
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1539
1543
variation_received , _ = self .decision_service .get_variation_for_feature (
1540
1544
self .project_config , feature , user
@@ -1567,7 +1571,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_experiment_
1567
1571
)
1568
1572
1569
1573
with mock .patch (
1570
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1574
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1571
1575
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1572
1576
variation_received , _ = self .decision_service .get_variation_for_feature (
1573
1577
self .project_config , feature , user
@@ -1597,7 +1601,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_experiment_
1597
1601
"test_experiment4" , "222240"
1598
1602
)
1599
1603
with mock .patch (
1600
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1604
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1601
1605
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1602
1606
variation_received , _ = self .decision_service .get_variation_for_feature (
1603
1607
self .project_config , feature , user
@@ -1628,7 +1632,7 @@ def test_get_variation_for_feature__returns_variation_for_feature_in_experiment_
1628
1632
)
1629
1633
1630
1634
with mock .patch (
1631
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 6500 ) as mock_generate_bucket_value , \
1635
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 6500 ) as mock_generate_bucket_value , \
1632
1636
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1633
1637
variation_received , _ = self .decision_service .get_variation_for_feature (
1634
1638
self .project_config , feature , user
@@ -1654,7 +1658,7 @@ def test_get_variation_for_feature__returns_variation_for_rollout_in_experiment_
1654
1658
feature = self .project_config .get_feature_from_key ("test_feature_in_multiple_experiments" )
1655
1659
1656
1660
with mock .patch (
1657
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 8000 ) as mock_generate_bucket_value , \
1661
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 8000 ) as mock_generate_bucket_value , \
1658
1662
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1659
1663
variation_received , _ = self .decision_service .get_variation_for_feature (
1660
1664
self .project_config , feature , user
@@ -1686,7 +1690,7 @@ def test_get_variation_for_feature__returns_variation_for_rollout_in_mutex_group
1686
1690
"211147" , "211149"
1687
1691
)
1688
1692
with mock .patch (
1689
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1693
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 2400 ) as mock_generate_bucket_value , \
1690
1694
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1691
1695
variation_received , _ = self .decision_service .get_variation_for_feature (
1692
1696
self .project_config , feature , user
@@ -1720,7 +1724,7 @@ def test_get_variation_for_feature_returns_rollout_in_experiment_bucket_range_25
1720
1724
)
1721
1725
1722
1726
with mock .patch (
1723
- 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1727
+ 'optimizely.bucketer.Bucketer._generate_bucket_value' , return_value = 4000 ) as mock_generate_bucket_value , \
1724
1728
mock .patch .object (self .project_config , 'logger' ) as mock_config_logging :
1725
1729
variation_received , _ = self .decision_service .get_variation_for_feature (
1726
1730
self .project_config , feature , user
0 commit comments