66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Copyright 2021 Memgraph Ltd.
|
||
|
|
#
|
||
|
|
# Use of this software is governed by the Business Source License
|
||
|
|
# included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
||
|
|
# License, and you may not use this file except in compliance with the Business Source License.
|
||
|
|
#
|
||
|
|
# As of the Change Date specified in that file, in accordance with
|
||
|
|
# the Business Source License, use of this software will be governed
|
||
|
|
# by the Apache License, Version 2.0, included in the file
|
||
|
|
# licenses/APL.txt.
|
||
|
|
|
||
|
|
import threading
|
||
|
|
|
||
|
|
import neo4j
|
||
|
|
|
||
|
|
|
||
|
|
def check_md(tx_md):
|
||
|
|
n = 0
|
||
|
|
for record in tx_md:
|
||
|
|
md = record[3]
|
||
|
|
if md["ver"] == "session" and md["str"] == "aha" and md["num"] == 123:
|
||
|
|
n = n + 1
|
||
|
|
elif md["ver"] == "transaction" and md["str"] == "oho" and md["num"] == 456:
|
||
|
|
n = n + 1
|
||
|
|
return n
|
||
|
|
|
||
|
|
|
||
|
|
def session_run(driver):
|
||
|
|
print("Checking implicit transaction metadata...")
|
||
|
|
with driver.session() as session:
|
||
|
|
query = neo4j.Query("SHOW TRANSACTIONS", timeout=2, metadata={"ver": "session", "str": "aha", "num": 123})
|
||
|
|
result = session.run(query).values()
|
||
|
|
assert check_md(result) == 1, "metadata info error!"
|
||
|
|
|
||
|
|
|
||
|
|
def show_tx(driver, tx_md):
|
||
|
|
with driver.session() as session:
|
||
|
|
query = neo4j.Query("SHOW TRANSACTIONS", timeout=2, metadata={"ver": "session", "str": "aha", "num": 123})
|
||
|
|
for t in session.run(query).values():
|
||
|
|
tx_md.append(t)
|
||
|
|
|
||
|
|
|
||
|
|
def transaction_run(driver):
|
||
|
|
print("Checking explicit transaction metadata...")
|
||
|
|
with driver.session() as session:
|
||
|
|
tx = session.begin_transaction(timeout=2, metadata={"ver": "transaction", "str": "oho", "num": 456})
|
||
|
|
tx.run("MATCH (n) RETURN n LIMIT 1").consume()
|
||
|
|
tx_md = []
|
||
|
|
th = threading.Thread(target=show_tx, args=(driver, tx_md))
|
||
|
|
th.start()
|
||
|
|
if th.is_alive():
|
||
|
|
th.join()
|
||
|
|
tx.commit()
|
||
|
|
assert check_md(tx_md) == 2, "metadata info error!"
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
driver = neo4j.GraphDatabase.driver("bolt://localhost:7687", auth=("user", "pass"), encrypted=False)
|
||
|
|
session_run(driver)
|
||
|
|
transaction_run(driver)
|
||
|
|
driver.close()
|
||
|
|
print("All ok!")
|