diff --git a/apps/api/tests/test_legacy.py b/apps/api/tests/test_legacy.py index d37e9d9cde1..9f16b8f6e76 100644 --- a/apps/api/tests/test_legacy.py +++ b/apps/api/tests/test_legacy.py @@ -140,9 +140,15 @@ class APITest(TestCase): 'base/addon_5299_gcal'] def setUp(self): + self._new_features = settings.NEW_FEATURES + settings.NEW_FEATURES = False + if hasattr(Addon, '_feature'): delattr(Addon, '_feature') + def tearDown(self): + settings.NEW_FEATURES = self._new_features + def test_api_caching(self): response = self.client.get('/en-US/firefox/api/1.5/addon/3615') eq_(response.status_code, 200) @@ -412,7 +418,8 @@ def test_is_category_featured(self): ('en-US', 'firefox', 0), ('ja', 'seamonkey', 0)]: # Clean out the special cache for feature. - delattr(Addon, '_feature') + if hasattr(Addon, '_feature'): + delattr(Addon, '_feature') self.assertContains(make_call('addon/5299', version=1.5, lang=lang, app=app), '%s' % result) @@ -438,7 +445,16 @@ class ListTest(TestCase): """Tests the list view with various urls.""" fixtures = ['base/apps', 'base/addon_3615', 'base/featured'] - @patch.object(settings._wrapped, 'NEW_FEATURES', False) + def setUp(self): + # TODO(cvan): Remove this once featured collections are enabled. + self._new_features = settings.NEW_FEATURES + settings.NEW_FEATURES = False + if hasattr(Addon, '_feature'): + delattr(Addon, '_feature') + + def tearDown(self): + settings.NEW_FEATURES = self._new_features + def test_defaults(self): """ This tests the default settings for /list. @@ -447,7 +463,6 @@ def test_defaults(self): response = make_call('list') self.assertContains(response, 'Persona""", 1) - @patch.object(settings._wrapped, 'NEW_FEATURES', False) def test_persona_search_15(self): response = make_call('list/recommended/9/1', version=1.5) self.assertContains(response, """Persona""", 1) - @patch.object(settings._wrapped, 'NEW_FEATURES', False) def test_limits(self): """ Assert /list/recommended/all/1 gets one item only. @@ -500,7 +512,6 @@ def test_version_filter(self): response = make_call('list/new/1/1/all/4.0') self.assertNotContains(response, "