class test__limit_get():
def setup(self):
self.reddit = Reddit(user_agent=TEST_AGENT, respect=False)
def test_no_limit(self):
r = self.reddit._limit_get()
s = self.reddit.get()
ok_(isinstance(r, things.Listing))
eq_(len(r), len(s))
ok_(r._limit is None)
def test_limit_normal(self):
r = self.reddit._limit_get(limit=10)
ok_(isinstance(r, things.Listing))
eq_(len(r), 10)
eq_(r._limit, 10)
def test_limit_already_params(self):
r = self.reddit._limit_get(params={}, limit=10)
ok_(isinstance(r, things.Listing))
eq_(len(r), 10)
eq_(r._limit, 10)
def test_limit_already_limit_in_params(self):
r = self.reddit._limit_get(params={'limit': 5}, limit=10)
ok_(isinstance(r, things.Listing))
eq_(len(r), 10)
eq_(r._limit, 10)
class test_get():
def setup(self):
self.reddit = Reddit(user_agent=TEST_AGENT)
def test_basic(self):
r = self.reddit.get()
ok_(isinstance(r, things.Listing))
r = self.reddit.get('r', 'test')
ok_(isinstance(r, things.Listing))
r = self.reddit.get('comments', params={'limit': 3})
ok_(isinstance(r, things.Listing))
eq_(len(r), 3)
def test_badresponse(self):
try:
self.reddit.get(genstr(), genstr())
except Exception as e:
ok_(isinstance(e, BadResponse))
ok_(isinstance(e.response, requests.Response))
ok_(e.response.status_code != 200)
