diff --git a/lms/djangoapps/experiments/tests/test_flags.py b/lms/djangoapps/experiments/tests/test_flags.py index 1d3c34eb04..3f1328df87 100644 --- a/lms/djangoapps/experiments/tests/test_flags.py +++ b/lms/djangoapps/experiments/tests/test_flags.py @@ -102,11 +102,10 @@ class ExperimentWaffleFlagTests(SharedModuleStoreTestCase): (False, 1), ) @ddt.unpack - def test_bucket_override(self, active, expected_bucket): + def test_forcing_bucket(self, active, expected_bucket): bucket_flag = CourseWaffleFlag('experiments', 'test.0') with bucket_flag.override(active=active): - self.assertEqual(self.flag.get_bucket(), expected_bucket) - self.assertEqual(self.flag.is_experiment_on(), active) + self.assertEqual(self.get_bucket(), expected_bucket) def test_tracking(self): # Run twice, with same request @@ -152,6 +151,7 @@ class ExperimentWaffleFlagTests(SharedModuleStoreTestCase): (False, 0, 0), ) @ddt.unpack + # Test the override method def test_override_method(self, active, bucket_override, expected_bucket): with self.flag.override(active=active, bucket=bucket_override): self.assertEqual(self.flag.get_bucket(), expected_bucket)