def test_consume_paginate_no_record(self, ses):
'''
test consume_paginate with GET that returns no record
'''
ses.request.return_value = MagicMock()
ses.request.return_value.raise_for_status.return_value = None
ses.request.return_value.json.return_value.get.return_value = 0
uri = 'http://localhost'
res = [entry for entry in cleanup_clicrdv.consume_paginate(ses, uri)]
self.assertEquals(res, [])
def test_consume_paginate_exactly_two_pages(self, ses):
'''
test consume_paginate with GET that returns exactly two pages
'''
ses.request.return_value = MagicMock()
ses.request.return_value.raise_for_status.return_value = None
ses.request.return_value.json.return_value = {
'totalRecords': 50,
'records': ['1'] * 50
}
uri = 'http://localhost'
res = [entry for entry in cleanup_clicrdv.consume_paginate(ses, uri)]
self.assertEquals(len(res), 50)
def test_consume_paginate_one_record(self, ses):
'''
test consume_paginate with GET that returns one record
'''
ses.request.return_value = MagicMock()
ses.request.return_value.raise_for_status.return_value = None
ses.request.return_value.json.return_value = {
'totalRecords': 1,
'records': ['OneVal']
}
uri = 'http://localhost'
res = [entry for entry in cleanup_clicrdv.consume_paginate(ses, uri)]
self.assertEquals(res, ['OneVal'])
def test_consume_paginate_url_one_param(self, ses):
'''
test consume_paginate with url that has no parm
'''
ses.request.return_value = MagicMock()
ses.request.return_value.raise_for_status.return_value = None
ses.request.return_value.json.return_value = {
'totalRecords': 26,
'records': ['1'] * 26
}
uri = 'http://localhost'
res = [entry for entry in cleanup_clicrdv.consume_paginate(ses, uri)]
self.assertEquals(len(res), 26)
ses.request.assert_called_with('GET', 'http://localhost?startIndex=26')