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')