Skip to content

Commit a46b21b

Browse files
Add code to handle testkit sending null access mode
1 parent 6a9ab5b commit a46b21b

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

testkitbackend/_async/requests.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -395,17 +395,17 @@ def __init__(self, session):
395395

396396
async def NewSession(backend, data):
397397
driver = backend.drivers[data["driverId"]]
398-
access_mode = data["accessMode"]
399-
if access_mode == "r":
400-
access_mode = neo4j.READ_ACCESS
401-
elif access_mode == "w":
402-
access_mode = neo4j.WRITE_ACCESS
403-
else:
404-
raise ValueError("Unknown access mode:" + access_mode)
405398
config = {
406-
"default_access_mode": access_mode,
407399
"database": data["database"],
408400
}
401+
access_mode = data["accessMode"]
402+
if access_mode is not None:
403+
if access_mode == "r":
404+
config["default_access_mode"] = neo4j.WRITE_ACCESS
405+
elif access_mode == "w":
406+
config["default_access_mode"] = neo4j.WRITE_ACCESS
407+
else:
408+
raise ValueError("Unknown access mode:" + access_mode)
409409
if data.get("bookmarks") is not None:
410410
config["bookmarks"] = neo4j.Bookmarks.from_raw_values(
411411
data["bookmarks"]

testkitbackend/_sync/requests.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -395,17 +395,17 @@ def __init__(self, session):
395395

396396
def NewSession(backend, data):
397397
driver = backend.drivers[data["driverId"]]
398-
access_mode = data["accessMode"]
399-
if access_mode == "r":
400-
access_mode = neo4j.READ_ACCESS
401-
elif access_mode == "w":
402-
access_mode = neo4j.WRITE_ACCESS
403-
else:
404-
raise ValueError("Unknown access mode:" + access_mode)
405398
config = {
406-
"default_access_mode": access_mode,
407399
"database": data["database"],
408400
}
401+
access_mode = data["accessMode"]
402+
if access_mode is not None:
403+
if access_mode == "r":
404+
config["default_access_mode"] = neo4j.READ_ACCESS
405+
elif access_mode == "w":
406+
config["default_access_mode"] = neo4j.WRITE_ACCESS
407+
else:
408+
raise ValueError("Unknown access mode:" + access_mode)
409409
if data.get("bookmarks") is not None:
410410
config["bookmarks"] = neo4j.Bookmarks.from_raw_values(
411411
data["bookmarks"]

0 commit comments

Comments
 (0)