def test_set_total(self): result = Mock(_nefertari_meta={'total': 5}) processed = wrappers.set_total(None, 2)(result=result) assert processed._nefertari_meta['total'] == 2 result = Mock(_nefertari_meta={'total': 1}) processed = wrappers.set_total(None, 2)(result=result) assert processed._nefertari_meta['total'] == 1
def test_set_total_no_meta(self): result = Mock(spec=[]) processed = wrappers.set_total(None, 2)(result=result) assert not hasattr(processed, '_nefertari_meta')