120 lines
4.1 KiB
Python
Executable File
120 lines
4.1 KiB
Python
Executable File
#!/usr/bin/python3 -u
|
|
import argparse
|
|
import atexit
|
|
import csv
|
|
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
PROJECT_DIR = os.path.normpath(os.path.join(SCRIPT_DIR, "..", "..", ".."))
|
|
|
|
QUERIES = [
|
|
("MATCH (n) DELETE n", {}),
|
|
("MATCH (n) DETACH DELETE n", {}),
|
|
("CREATE (n)", {}),
|
|
("CREATE (n {name: $name})", {"name": True}),
|
|
("CREATE (n {name: $name})", {"name": 5}),
|
|
("CREATE (n {name: $name})", {"name": 3.14}),
|
|
("CREATE (n {name: $name})", {"name": "nandare"}),
|
|
("CREATE (n {name: $name})", {"name": ["nandare", "hai hai hai"]}),
|
|
("CREATE (n {name: $name})", {"name": {"test": "ho ho ho"}}),
|
|
("CREATE (n {name: $name})", {"name": 5, "leftover": 42}),
|
|
("MATCH (n), (m) CREATE (n)-[:e {when: $when}]->(m)", {"when": 42}),
|
|
("MATCH (n) RETURN n", {}),
|
|
(
|
|
"MATCH (n), (m {type: $type}) RETURN count(n), count(m)",
|
|
{"type": "dadada"}
|
|
),
|
|
(
|
|
"MERGE (n) ON CREATE SET n.created = timestamp() "
|
|
"ON MATCH SET n.lastSeen = timestamp() "
|
|
"RETURN n.name, n.created, n.lastSeen",
|
|
{}
|
|
),
|
|
(
|
|
"MATCH (n {value: $value}) SET n.value = 0 RETURN n",
|
|
{"value": "nandare!"}
|
|
),
|
|
("MATCH (n), (m) SET n.value = m.value", {}),
|
|
("MATCH (n {test: $test}) REMOVE n.value", {"test": 48}),
|
|
("MATCH (n), (m) REMOVE n.value, m.value", {}),
|
|
("CREATE INDEX ON :User (id)", {}),
|
|
]
|
|
|
|
|
|
def wait_for_server(port, delay=0.1):
|
|
cmd = ["nc", "-z", "-w", "1", "127.0.0.1", str(port)]
|
|
while subprocess.call(cmd) != 0:
|
|
time.sleep(0.01)
|
|
time.sleep(delay)
|
|
|
|
|
|
def execute_test(memgraph_binary, tester_binary):
|
|
storage_directory = tempfile.TemporaryDirectory()
|
|
memgraph_args = [memgraph_binary,
|
|
"--storage-properties-on-edges",
|
|
"--data-directory", storage_directory.name,
|
|
"--audit-enabled"]
|
|
|
|
# Start the memgraph binary
|
|
memgraph = subprocess.Popen(list(map(str, memgraph_args)))
|
|
time.sleep(0.1)
|
|
assert memgraph.poll() is None, "Memgraph process died prematurely!"
|
|
wait_for_server(7687)
|
|
|
|
# Register cleanup function
|
|
@atexit.register
|
|
def cleanup():
|
|
if memgraph.poll() is None:
|
|
memgraph.terminate()
|
|
assert memgraph.wait() == 0, "Memgraph process didn't exit cleanly!"
|
|
|
|
# Execute all queries
|
|
print("\033[1;36m~~ Starting query execution ~~\033[0m")
|
|
for query, params in QUERIES:
|
|
print(query, params)
|
|
args = [tester_binary, "--query", query,
|
|
"--params-json", json.dumps(params)]
|
|
subprocess.run(args).check_returncode()
|
|
print("\033[1;36m~~ Finished query execution ~~\033[0m\n")
|
|
|
|
# Shutdown the memgraph binary
|
|
memgraph.terminate()
|
|
assert memgraph.wait() == 0, "Memgraph process didn't exit cleanly!"
|
|
|
|
# Verify the written log
|
|
print("\033[1;36m~~ Starting log verification ~~\033[0m")
|
|
with open(os.path.join(storage_directory.name, "audit", "audit.log")) as f:
|
|
reader = csv.reader(f, delimiter=',', doublequote=False,
|
|
escapechar='\\', lineterminator='\n',
|
|
quotechar='"', quoting=csv.QUOTE_MINIMAL,
|
|
skipinitialspace=False, strict=True)
|
|
queries = []
|
|
for line in reader:
|
|
timestamp, address, username, query, params = line
|
|
params = json.loads(params)
|
|
queries.append((query, params))
|
|
print(query, params)
|
|
assert queries == QUERIES, "Logged queries don't match " \
|
|
"executed queries!"
|
|
print("\033[1;36m~~ Finished log verification ~~\033[0m\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
memgraph_binary = os.path.join(PROJECT_DIR, "build", "memgraph")
|
|
tester_binary = os.path.join(PROJECT_DIR, "build", "tests",
|
|
"integration", "audit", "tester")
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--memgraph", default=memgraph_binary)
|
|
parser.add_argument("--tester", default=tester_binary)
|
|
args = parser.parse_args()
|
|
|
|
execute_test(args.memgraph, args.tester)
|
|
|
|
sys.exit(0)
|