def remote_connection(request):
try:
if request.param == 'graphbinaryv1':
remote_conn = DriverRemoteConnection(
anonymous_url,
'gmodern',
message_serializer=serializer.GraphBinarySerializersV1())
elif request.param == 'graphsonv2':
remote_conn = DriverRemoteConnection(
anonymous_url,
'gmodern',
message_serializer=serializer.GraphSONSerializersV2d0())
elif request.param == 'graphsonv3':
remote_conn = DriverRemoteConnection(
anonymous_url,
'gmodern',
message_serializer=serializer.GraphSONSerializersV3d0())
else:
raise ValueError("Invalid serializer option - " + request.param)
except OSError:
pytest.skip('Gremlin Server is not running')
else:
def fin():
remote_conn.close()
request.addfinalizer(fin)
return remote_conn
def __create_remote(server_graph_name):
if not("serializer" in world.config.user_data):
raise ValueError('test configuration requires setting of --user-data="serializer={mime-type}"')
if world.config.user_data["serializer"] == "application/vnd.gremlin-v3.0+json":
s = serializer.GraphSONSerializersV3d0()
elif world.config.user_data["serializer"] == "application/vnd.graphbinary-v1.0":
s = serializer.GraphBinarySerializersV1()
else:
raise ValueError('serializer not found - ' + world.config.user_data["serializer"])
return DriverRemoteConnection('ws://localhost:45940/gremlin', server_graph_name, message_serializer=s)
def create_connection_to_gtx():
return DriverRemoteConnection(
anonymous_url,
'gtx',
message_serializer=serializer.GraphBinarySerializersV1())
