def test__post_filters_same():
node1 = query.PostFilterNode("predicate1")
node2 = query.PostFilterNode("predicate2")
and_node = query.ConjunctionNode(node1, node2)
post_filters_node = and_node._post_filters()
assert post_filters_node is and_node
def test__to_filter_multiple():
node1 = query.PostFilterNode("predicate1")
node2 = query.PostFilterNode("predicate2")
and_node = query.ConjunctionNode(node1, node2)
with pytest.raises(NotImplementedError):
and_node._to_filter(post=True)
def test__post_filters_multiple():
node1 = query.FilterNode("a", "=", 7)
node2 = query.PostFilterNode("predicate2")
node3 = query.PostFilterNode("predicate3")
and_node = query.ConjunctionNode(node1, node2, node3)
post_filters_node = and_node._post_filters()
assert post_filters_node == query.ConjunctionNode(node2, node3)
def test___eq__():
predicate1 = unittest.mock.sentinel.predicate1
post_filter_node1 = query.PostFilterNode(predicate1)
predicate2 = unittest.mock.sentinel.predicate2
post_filter_node2 = query.PostFilterNode(predicate2)
post_filter_node3 = unittest.mock.sentinel.post_filter_node
assert post_filter_node1 == post_filter_node1
assert not post_filter_node1 == post_filter_node2
assert not post_filter_node1 == post_filter_node3
def test__post_filters_single():
node1 = query.FilterNode("a", "=", 7)
node2 = query.PostFilterNode("predicate2")
and_node = query.ConjunctionNode(node1, node2)
post_filters_node = and_node._post_filters()
assert post_filters_node is node2
def test_pickling():
predicate = "must-be-pickle-able"
post_filter_node = query.PostFilterNode(predicate)
pickled = pickle.dumps(post_filter_node)
unpickled = pickle.loads(pickled)
assert post_filter_node == unpickled
def test__to_filter_single():
node1 = unittest.mock.Mock(spec=query.FilterNode)
node2 = query.PostFilterNode("predicate")
node3 = unittest.mock.Mock(spec=query.FilterNode)
node3._to_filter.return_value = False
and_node = query.ConjunctionNode(node1, node2, node3)
as_filter = and_node._to_filter()
assert as_filter is node1._to_filter.return_value
node1._to_filter.assert_called_once_with(post=False)
def test_constructor():
with pytest.raises(NotImplementedError):
query.PostFilterNode()
def test__to_filter():
predicate = unittest.mock.sentinel.predicate
post_filter_node = query.PostFilterNode(predicate)
assert post_filter_node._to_filter() is None
def test___repr__():
predicate = "predicate-not-repr"
post_filter_node = query.PostFilterNode(predicate)
assert repr(post_filter_node) == "PostFilterNode(predicate-not-repr)"
def test_constructor():
predicate = unittest.mock.sentinel.predicate
post_filter_node = query.PostFilterNode(predicate)
assert post_filter_node.predicate is predicate
