def test_traverser_prune_called_with_filter(self):
id_filter = Filter(blacklist='id')
obj = Traverser({'id': 1, 'username': '******'})
obj.prune()
self.assertTrue(obj() == {'id': 1, 'username': '******'})
obj.prune(filter=id_filter)
self.assertTrue(obj() == {'username': '******'})
def test_prune_call_chain(self):
obj = Traverser({})
self.assertTrue(obj.prune() is obj)
def test_traverser_prune_with_whitelist_filter(self):
id_filter = Filter(whitelist='id')
obj = Traverser({'id': 1, 'username': '******'}, filter=id_filter)
obj.prune()
self.assertTrue(obj() == {'id': 1})
def test_traverser_prune_no_filter(self):
obj = Traverser({'id': 1, 'username': '******'})
obj.prune()
self.assertTrue(obj() == {'id': 1, 'username': '******'})
