38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
from behave import *
|
||
|
|
import os
|
||
|
|
import database
|
||
|
|
|
||
|
|
@given(u'the binary-tree-1 graph')
|
||
|
|
def step_impl(context):
|
||
|
|
create_given_graph('binary-tree-1', context)
|
||
|
|
|
||
|
|
@given(u'the binary-tree-2 graph')
|
||
|
|
def step_impl(context):
|
||
|
|
create_given_graph('binary-tree-2', context)
|
||
|
|
|
||
|
|
def create_given_graph(name, context):
|
||
|
|
"""
|
||
|
|
Function deletes everything from database and creates a new
|
||
|
|
graph. Graph file name is an argument of function. Function
|
||
|
|
executes query written in a .cypher file and sets graph
|
||
|
|
properties to beginning values.
|
||
|
|
"""
|
||
|
|
database.query("MATCH (n) DETACH DELETE n", context)
|
||
|
|
path = find_graph_path(name, context.config.graphs_root)
|
||
|
|
with open(path, 'r') as f:
|
||
|
|
#TODO not good solution for all queries
|
||
|
|
q = f.read().replace('\n', ' ')
|
||
|
|
database.query(q, context)
|
||
|
|
context.graph_properties.set_beginning_parameters()
|
||
|
|
|
||
|
|
|
||
|
|
def find_graph_path(name, path):
|
||
|
|
"""
|
||
|
|
Function returns path to .cypher file with given name in
|
||
|
|
given folder or subfolders. Argument path is path to a given
|
||
|
|
folder.
|
||
|
|
"""
|
||
|
|
for root, dirs, files in os.walk(path):
|
||
|
|
if name + '.cypher' in files:
|
||
|
|
return root + '/' + name + '.cypher'
|