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': '******'})