43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
import logging, datetime, time, json
|
||
|
|
from steps.test_parameters import TestParameters
|
||
|
|
from neo4j.v1 import GraphDatabase, basic_auth
|
||
|
|
from steps.graph_properties import GraphProperties
|
||
|
|
from test_results import TestResults
|
||
|
|
|
||
|
|
test_results = TestResults()
|
||
|
|
|
||
|
|
def before_scenario(context, step):
|
||
|
|
context.test_parameters = TestParameters()
|
||
|
|
context.graph_properties = GraphProperties()
|
||
|
|
context.exception = None
|
||
|
|
|
||
|
|
def after_scenario(context, scenario):
|
||
|
|
test_results.add_test(scenario.status)
|
||
|
|
|
||
|
|
def before_all(context):
|
||
|
|
set_logging(context)
|
||
|
|
context.driver = create_db_driver(context)
|
||
|
|
|
||
|
|
def after_all(context):
|
||
|
|
ts = time.time()
|
||
|
|
timestamp = datetime.datetime.fromtimestamp(ts).strftime("%Y_%m_%d__%H_%M")
|
||
|
|
file_name = context.config.output_folder + context.config.database + "_" + timestamp + ".json"
|
||
|
|
js = {"total": test_results.num_total(), "passed": test_results.num_passed(), "test_suite": context.config.root}
|
||
|
|
with open(file_name, 'w') as f:
|
||
|
|
json.dump(js, f)
|
||
|
|
|
||
|
|
def set_logging(context):
|
||
|
|
logging.basicConfig(level="DEBUG")
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
context.log = log
|
||
|
|
|
||
|
|
def create_db_driver(context):
|
||
|
|
uri = context.config.database_uri
|
||
|
|
auth_token = basic_auth(context.config.database_username, context.config.database_password)
|
||
|
|
if context.config.database == "neo4j":
|
||
|
|
driver = GraphDatabase.driver(uri, auth=auth_token)
|
||
|
|
else:
|
||
|
|
#Memgraph
|
||
|
|
pass
|
||
|
|
return driver
|