Exemplos de ClusterNodesManager em Python

Exemplo n.º 1

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_all_slots_covered():
    decoded_node_info_ok = list(parse_cluster_nodes(
        RAW_NODE_INFO_DATA_OK.decode('utf-8'), encoding='utf-8'
    ))
    manager = ClusterNodesManager.create(decoded_node_info_ok)
    assert manager.all_slots_covered

    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert not manager.all_slots_covered

    decoded_node_info_ok[2]['slots'] = list(decoded_node_info_ok[2]['slots'])
    del decoded_node_info_ok[2]['slots'][0]

    manager = ClusterNodesManager.create(decoded_node_info_ok)
    assert not manager.all_slots_covered

Exemplo n.º 2

def test_all_slots_covered():
    decoded_node_info_ok = list(
        parse_cluster_nodes(RAW_NODE_INFO_DATA_OK.decode('utf-8'),
                            encoding='utf-8'))
    manager = ClusterNodesManager.create(decoded_node_info_ok)
    assert manager.all_slots_covered

    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert not manager.all_slots_covered

    decoded_node_info_ok[2]['slots'] = list(decoded_node_info_ok[2]['slots'])
    del decoded_node_info_ok[2]['slots'][0]

    manager = ClusterNodesManager.create(decoded_node_info_ok)
    assert not manager.all_slots_covered

Exemplo n.º 3

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_get_node_by_address():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    node = manager.get_node_by_address(('127.0.0.1', 30004))
    assert node.id == '07c37dfeb235213a872192d90877d0cd55635b91'

    no_node = manager.get_node_by_address('xxx')
    assert no_node is None

Exemplo n.º 4

def test_get_node_by_address():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    node = manager.get_node_by_address(('127.0.0.1', 30004))
    assert node.id == '07c37dfeb235213a872192d90877d0cd55635b91'

    no_node = manager.get_node_by_address('xxx')
    assert no_node is None

Exemplo n.º 5

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_in_range():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    master = manager.nodes[5]
    assert master.in_range(0)
    assert master.in_range(5460)
    assert not master.in_range(5461)

    master.slots = tuple()

    assert not master.in_range(0)

Exemplo n.º 6

def test_in_range():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    master = manager.nodes[5]
    assert master.in_range(0)
    assert master.in_range(5460)
    assert not master.in_range(5461)

    master.slots = tuple()

    assert not master.in_range(0)

Exemplo n.º 7

def test_cluster_node():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    node1 = manager.nodes[0]
    assert not node1.is_master
    assert node1.is_slave
    assert node1.address == ('127.0.0.1', 30004)
    assert not node1.is_alive

    node2 = manager.nodes[2]
    assert node2.is_master
    assert not node2.is_slave
    assert node2.is_alive

    assert node1 == manager.get_node_by_id(node1.id)

Exemplo n.º 8

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_cluster_node():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    node1 = manager.nodes[0]
    assert not node1.is_master
    assert node1.is_slave
    assert node1.address == ('127.0.0.1', 30004)
    assert not node1.is_alive

    node2 = manager.nodes[2]
    assert node2.is_master
    assert not node2.is_slave
    assert node2.is_alive

    assert node1 == manager.get_node_by_id(node1.id)

Exemplo n.º 9

def test_determine_slot_multiple():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    assert manager.determine_slot('{key}:1', '{key}:2') == 12539

Exemplo n.º 10

def test_determine_slot():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    assert manager.determine_slot('key') == 12539

    with pytest.raises(TypeError):
        manager.determine_slot(None)

Exemplo n.º 11

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_node_count():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert manager.nodes_count == 4
    assert manager.masters_count == 2
    assert manager.slaves_count == 2

Exemplo n.º 12

def test_alive_nodes():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert manager.alive_nodes == manager.nodes[2:]

Exemplo n.º 13

def test_node_count():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert manager.nodes_count == 4
    assert manager.masters_count == 2
    assert manager.slaves_count == 2

Exemplo n.º 14

def test_create():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert len(manager.nodes) == 6
    assert all(isinstance(node, ClusterNode) for node in manager.nodes)

Exemplo n.º 15

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_create():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert len(manager.nodes) == 6
    assert all(isinstance(node, ClusterNode) for node in manager.nodes)

Exemplo n.º 16

def test_determine_slot_multiple_different():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    with pytest.raises(RedisClusterError):
        manager.determine_slot('key:1', 'key:2')

Exemplo n.º 17

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_determine_slot_multiple_different():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    with pytest.raises(RedisClusterError):
        manager.determine_slot('key:1', 'key:2')

Exemplo n.º 18

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_determine_slot_multiple():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    assert manager.determine_slot('{key}:1', '{key}:2') == 12539

Exemplo n.º 19

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_determine_slot():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_OK)
    assert manager.determine_slot('key') == 12539

    with pytest.raises(TypeError):
        manager.determine_slot(None)

Exemplo n.º 20

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_alive_nodes():
    manager = ClusterNodesManager.create(NODE_INFO_DATA_FAIL)
    assert manager.alive_nodes == manager.nodes[2:]

Exemplo n.º 21

def test_key_slot():
    assert ClusterNodesManager.key_slot(SLOT_ZERO_KEY) == 0
    assert ClusterNodesManager.key_slot('key') == KEY_KEY_SLOT
    assert ClusterNodesManager.key_slot(b'key') == KEY_KEY_SLOT

Exemplo n.º 22

Arquivo: cluster_test.py Projeto: trezorg/aioredis

def test_key_slot():
    assert ClusterNodesManager.key_slot(SLOT_ZERO_KEY) == 0
    assert ClusterNodesManager.key_slot('key') == KEY_KEY_SLOT
    assert ClusterNodesManager.key_slot(b'key') == KEY_KEY_SLOT