def test_pos_arg__single(self):
func = decorators.allow_list(1)(add_one)
func(self.attr, 'main')
self.assertEquals(self.attr['main'], 1)
func = decorators.allow_list(0)(add_one)
func(self.attr, 'sec')
self.assertEquals(self.attr['sec'], 1)
func(self.params, 'main')
self.assertEquals(self.params['main'], 1)
def test_key_arg__single(self):
func = decorators.allow_list(1, 'field')(add_one)
func(self.attr, 'main')
self.assertEquals(self.attr['main'], 1)
func(self.attr, field='main')
self.assertEquals(self.attr['main'], 2)
def test_key_arg__list(self):
func = decorators.allow_list(1, 'field')(add_one)
func(self.attr, ['main', 'sec', 'main'])
self.assertEquals(self.attr['main'], 2)
self.assertEquals(self.attr['sec'], 1)
func(self.attr, field=['main', 'sec', 'main'])
self.assertEquals(self.attr['main'], 4)
self.assertEquals(self.attr['sec'], 2)
def test_pos_arg__list(self):
func = decorators.allow_list(1)(add_one)
func(self.attr, ['main', 'sec', 'main'])
self.assertEquals(self.attr['main'], 2)
self.assertEquals(self.attr['sec'], 1)
