def test_double_traverse(): nescafe = Coffee(name='Nescafe plus', price=99).save() tesco = Supplier(name='Asda', delivery_cost=2).save() nescafe.suppliers.connect(tesco) tesco.coffees.connect(Coffee(name='Decafe', price=2).save()) ns = NodeSet(NodeSet(source=nescafe).suppliers.match()).coffees.match() qb = QueryBuilder(ns).build_ast() results = qb._execute() assert len(results) == 1 assert results[0].name == 'Decafe'
def test_simple_has_via_label(): nescafe = Coffee(name='Nescafe', price=99).save() tesco = Supplier(name='Tesco', delivery_cost=2).save() nescafe.suppliers.connect(tesco) ns = NodeSet(Coffee).has(suppliers=True) qb = QueryBuilder(ns).build_ast() results = qb._execute() assert 'SUPPLIES' in qb._ast['where'][0] assert len(results) == 1 assert results[0].name == 'Nescafe' Coffee(name='nespresso', price=99).save() ns = NodeSet(Coffee).has(suppliers=False) qb = QueryBuilder(ns).build_ast() results = qb._execute() assert len(results) > 0 assert 'NOT' in qb._ast['where'][0]
def test_simple_has_via_label(): nescafe = Coffee(name="Nescafe", price=99).save() tesco = Supplier(name="Tesco", delivery_cost=2).save() nescafe.suppliers.connect(tesco) ns = NodeSet(Coffee).has(suppliers=True) qb = QueryBuilder(ns).build_ast() results = qb._execute() assert "COFFEE SUPPLIERS" in qb._ast["where"][0] assert len(results) == 1 assert results[0].name == "Nescafe" Coffee(name="nespresso", price=99).save() ns = NodeSet(Coffee).has(suppliers=False) qb = QueryBuilder(ns).build_ast() results = qb._execute() assert len(results) > 0 assert "NOT" in qb._ast["where"][0]
def test_simple_traverse_with_filter(): nescafe = Coffee(name='Nescafe2', price=99).save() tesco = Supplier(name='Sainsburys', delivery_cost=2).save() nescafe.suppliers.connect(tesco) qb = QueryBuilder(NodeSet(source=nescafe).suppliers.match(since__lt=datetime.now())) results = qb.build_ast()._execute() assert 'start' in qb._ast assert 'match' in qb._ast assert qb._ast['return'] == 'suppliers' assert len(results) == 1 assert results[0].name == 'Sainsburys'
def test_filter_exclude_via_labels(): Coffee(name='Java', price=99).save() node_set = NodeSet(Coffee) qb = QueryBuilder(node_set).build_ast() results = qb._execute() assert '(coffee:Coffee)' in qb._ast['match'] assert 'result_class' in qb._ast assert len(results) == 1 assert isinstance(results[0], Coffee) assert results[0].name == 'Java' # with filter and exclude Coffee(name='Kenco', price=3).save() node_set = node_set.filter(price__gt=2).exclude(price__gt=6, name='Java') qb = QueryBuilder(node_set).build_ast() results = qb._execute() assert '(coffee:Coffee)' in qb._ast['match'] assert 'NOT' in qb._ast['where'][0] assert len(results) == 1 assert results[0].name == 'Kenco'
def test_simple_traverse_with_filter(): nescafe = Coffee(name="Nescafe2", price=99).save() tesco = Supplier(name="Sainsburys", delivery_cost=2).save() nescafe.suppliers.connect(tesco) qb = QueryBuilder( NodeSet(source=nescafe).suppliers.match(since__lt=datetime.now())) results = qb.build_ast()._execute() assert "lookup" in qb._ast assert "match" in qb._ast assert qb._ast["return"] == "suppliers" assert len(results) == 1 assert results[0].name == "Sainsburys"
def test_filter_exclude_via_labels(): Coffee(name="Java", price=99).save() node_set = NodeSet(Coffee) qb = QueryBuilder(node_set).build_ast() results = qb._execute() assert "(coffee:Coffee)" in qb._ast["match"] assert "result_class" in qb._ast assert len(results) == 1 assert isinstance(results[0], Coffee) assert results[0].name == "Java" # with filter and exclude Coffee(name="Kenco", price=3).save() node_set = node_set.filter(price__gt=2).exclude(price__gt=6, name="Java") qb = QueryBuilder(node_set).build_ast() results = qb._execute() assert "(coffee:Coffee)" in qb._ast["match"] assert "NOT" in qb._ast["where"][0] assert len(results) == 1 assert results[0].name == "Kenco"
def test_traversal_filter_left_hand_statement(): nescafe = Coffee(name='Nescafe2', price=99).save() nescafe_gold = Coffee(name='Nescafe gold', price=11).save() tesco = Supplier(name='Sainsburys', delivery_cost=3).save() biedronka = Supplier(name='Biedronka', delivery_cost=5).save() lidl = Supplier(name='Lidl', delivery_cost=3).save() nescafe.suppliers.connect(tesco) nescafe_gold.suppliers.connect(biedronka) nescafe_gold.suppliers.connect(lidl) lidl_supplier = NodeSet( Coffee.nodes.filter(price=11).suppliers).filter(delivery_cost=3).all() assert lidl in lidl_supplier
def test_count(): Coffee(name='Nescafe Gold', price=99).save() count = QueryBuilder(NodeSet(source=Coffee)).build_ast()._count() assert count > 0