def test_disabled_query_isolation(self): conn = self.get_socket() session_msg = SessionRequest(username=self.username, password=self.password) conn.send_message(session_msg) response = conn.get_response() self.assertIsInstance(response, SessionResponse) session_key = response.session_key out_msg = ScriptRequest(""" x = 5 x """, isolate=False, session_key=session_key) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertEqual(response.results, 5) out_msg = ScriptRequest(""" y = 5 + x y """, session_key=session_key) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertEqual(response.results, 10)
def test_query_value_isolation(self): conn = self.get_socket() session_msg = SessionRequest() conn.send_message(session_msg) response = conn.get_response() self.assertIsInstance(response, SessionResponse) session_key = response.session_key out_msg = ScriptRequest( """ x = 5 x """, session_key=session_key ) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertEqual(response.results, 5) out_msg = ScriptRequest( """ y = 5 + x y """, session_key=session_key ) conn.send_message(out_msg) response = conn.get_response() self.assertErrorResponse(response)
def test_omitting_session_key_in_sessioned_message_returns_proper_error(self): conn = self.get_socket() out_msg = ScriptRequest( """ g.getFeatures().toMap() """, ) conn.send_message(out_msg) response = conn.get_response() self.assertErrorResponse(response) self.assertEqual(response.meta.get('flag'), ErrorResponse.SCRIPT_FAILURE_ERROR)
def test_graph_definition(self): conn = self.get_socket() out_msg = ScriptRequest(""" g.getFeatures().toMap() """, isolate=False, in_transaction=False, graph_name=self.graphname, in_session=False) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertIsInstance(response.results, dict)
def test_sessionless_message(self): conn = self.get_socket() out_msg = ScriptRequest(""" v = g.addVertex([xyz:5]) v """, in_session=False, graph_name=self.graphname) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertIsInstance(response.results, dict) self.assertIn('_properties', response.results) self.assertIn('xyz', response.results['_properties']) self.assertEqual(response.results['_properties']['xyz'], 5)
def test_non_transactional_sessionless_message(self): conn = self.get_socket() out_msg = ScriptRequest( """ v = g.addVertex([xyz:5]) v """, in_session=False, graph_name=self.graphname, in_transaction=False ) conn.send_message(out_msg) response = conn.get_response() self.assertNotErrorResponse(response) self.assertIsInstance(response, MsgPackScriptResponse) self.assertIsInstance(response.results, dict) assert '_properties' in response.results assert 'xyz' in response.results['_properties'] assert response.results['_properties']['xyz'] == 5