diff --git a/rtv/oauth.py b/rtv/oauth.py index 56f0257..e1502c6 100644 --- a/rtv/oauth.py +++ b/rtv/oauth.py @@ -311,7 +311,8 @@ class OAuthRateLimitHandler(DefaultHandler): else: self.next_request_timestamp = None - @fix_cache(DefaultHandler.with_cache) + @fix_cache + @DefaultHandler.with_cache def request(self, request, proxies, timeout, verify, **_): settings = self.http.merge_environment_settings( diff --git a/tests/conftest.py b/tests/conftest.py index a42a4ef..b07a00c 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -12,7 +12,7 @@ from vcr import VCR from six.moves.urllib.parse import urlparse, parse_qs from six.moves.BaseHTTPServer import HTTPServer -from rtv.oauth import OAuthHelper, OAuthHandler +from rtv.oauth import OAuthHelper, OAuthHandler, OAuthRateLimitHandler from rtv.config import Config from rtv.packages import praw from rtv.terminal import Terminal @@ -180,9 +180,12 @@ def reddit(vcr, request): with vcr.use_cassette(cassette_name): with patch('rtv.packages.praw.Reddit.get_access_information'): + handler = OAuthRateLimitHandler() + reddit = praw.Reddit(user_agent='rtv test suite', decode_html_entities=False, - disable_update_check=True) + disable_update_check=True, + handler=handler) # praw uses a global cache for requests, so we need to clear it # before each unit test. Otherwise we may fail to generate new # cassettes.