View GitHub Repository aiogremlin

Schedule Inspection
Subscribe

test_aliases.test_alias_serialization()   A
last analyzed

↳ Parent: test_aliases

Complexity

Conditions 1

Size

Total Lines 18
Code Lines 17

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
eloc 17
dl 0
loc 18
rs 9.55
c 0
b 0
f 0
cc 1
nop 1
1 
import json
2 
import uuid
3 

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    4
                

                                    
                                                    
                
                
                
import pytest

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    5
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    6
                

                                    
                                                    
                
                
                
from aiogremlin import driver

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                
from gremlin_python.driver import request, serializer

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    9
                

                                    
                                                    
                
                
                
@pytest.mark.asyncio

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    10
                

                                    
                                                    
                
                
                
async def test_gremlin_query(event_loop, cluster):

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                
    alias = { 'g': 'g' }

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                
    cluster = await driver.Cluster.open(event_loop, aliases=alias, hosts=['gremlin-server'])

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    13
                

                                    
                                                    
                
                
                
    client = await cluster.connect()

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    14
                

                                    
                                                    
                
                
                
    assert client.aliases == alias

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    15
                

                                    
                                                    
                
                
                
    resp = await client.submit("1 + 1")

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                
    async for msg in resp:

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    17
                

                                    
                                                    
                
                
                
        print(msg)

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    18
                

                                    
                                                    
                
                
                
    await cluster.close()

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    19
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    20
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    21
                

                                    
                                                    
                
                
                
@pytest.mark.asyncio

            


            
                                                        
            
                                    
            
            

                
                    22
                

                                    
                                                    
                
                
                
async def test_alias_serialization(event_loop):

            


            
                                                        
            
                                    
            
            

                
                    23
                

                                    
                                                    
                
                
                
    alias = { 'g': 'g' }

            


            
                                                        
            
                                    
            
            

                
                    24
                

                                    
                                                    
                
                
                
    message = '1 + 1'

            


            
                                                        
            
                                    
            
            

                
                    25
                

                                    
                                                    
                
                
                
    cluster = await driver.Cluster.open(event_loop, aliases=alias, hosts=['gremlin-server'])

            


            
                                                        
            
                                    
            
            

                
                    26
                

                                    
                                                    
                
                
                
    client = await cluster.connect()

            


            
                                                        
            
                                    
            
            

                
                    27
                

                                    
                                                    
                
                
                
    # This is the code client/conn uses on submit

            


            
                                                        
            
                                    
            
            

                
                    28
                

                                    
                                                    
                
                
                
    message = request.RequestMessage(

            


            
                                                        
            
                                    
            
            

                
                    29
                

                                    
                                                    
                
                
                
        processor='', op='eval',

            


            
                                                        
            
                                    
            
            

                
                    30
                

                                    
                                                    
                
                
                
        args={'gremlin': message,

            


            
                                                        
            
                                    
            
            

                
                    31
                

                                    
                                                    
                
                
                
              'aliases': client._aliases})

            


            
                                                        
            
                                    
            
            

                
                    32
                

                                    
                                                    
                
                
                
    request_id = str(uuid.uuid4())

            


            
                                                        
            
                                    
            
            

                
                    33
                

                                    
                                                    
                
                
                
    message = serializer.GraphSONMessageSerializer().serialize_message(

            


            
                                                        
            
                                    
            
            

                
                    34
                

                                    
                                                    
                
                
                
        request_id, message)

            


            
                                                        
            
                                    
            
            

                
                    35
                

                                    
                                                    
                
                
                
    message = message.decode('utf-8')[34:]

            


            
                                                        
            
                                    
            
            

                
                    36
                

                                    
                                                    
                
                
                
    aliases = json.loads(message)['args']['aliases']

            


            
                                                        
            
                                    
            
            

                
                    37
                

                                    
                                                    
                
                
                
    assert aliases == alias

            


            
                                                        
            
                                    
            
            

                
                    38
                

                                    
                                                    
                
                
                
    await cluster.close()

            


            
                                                        
            
                                    
            
            

                
                    39