示例#1
0
def test_replica_delete(grpc_controller_client):  # NOQA
    f1 = create_backend_file()
    f2 = create_backend_file()
    f3 = create_backend_file()
    r1 = grpc_controller_client.replica_create(address='file://' + f1)
    grpc_controller_client.replica_update(r1.address, mode='RW')
    r2 = grpc_controller_client.replica_create(address='file://' + f2)
    grpc_controller_client.replica_update(r2.address, mode='RW')
    r3 = grpc_controller_client.replica_create(address='file://' + f3)
    grpc_controller_client.replica_update(r3.address, mode='RW')

    rs = grpc_controller_client.replica_list()
    assert len(rs) == 3

    grpc_controller_client.replica_delete(r1.address)
    rs = grpc_controller_client.replica_list()
    assert len(rs) == 2

    grpc_controller_client.replica_delete(r1.address)
    rs = grpc_controller_client.replica_list()
    assert len(rs) == 2

    grpc_controller_client.replica_delete(r2.address)
    rs = grpc_controller_client.replica_list()
    assert len(rs) == 1

    grpc_controller_client.replica_delete(r3.address)
    rs = grpc_controller_client.replica_list()
    assert len(rs) == 0

    cleanup_backend_file([f1, f2, f3])
示例#2
0
def test_replica_create(grpc_controller_client):  # NOQA
    f1 = create_backend_file()
    r1 = 'file://' + f1
    replica = grpc_controller_client.replica_create(address=r1)
    assert replica.address == r1

    grpc_controller_client.replica_create(address=r1)
    grpc_controller_client.replica_create(address=r1)

    rs = grpc_controller_client.replica_list()
    assert len(rs) == 1
    assert rs[0].address == r1
    assert rs[0].mode == 'WO'

    f2 = create_backend_file()
    r2 = 'file://' + f2
    with pytest.raises(grpc.RpcError) as e:
        grpc_controller_client.replica_create(address=r2)
    assert 'Can only have one WO replica at a time' in str(e.value)

    r = grpc_controller_client.replica_update(rs[0].address, mode='RW')
    assert r.mode == 'RW'

    replica2 = grpc_controller_client.replica_create(address=r2)
    assert replica2.address == r2

    rs = grpc_controller_client.replica_list()
    assert len(rs) == 2

    cleanup_backend_file([f1, f2])
示例#3
0
def test_shutdown(grpc_controller_client):  # NOQA
    v = grpc_controller_client.volume_get()
    assert v.replicaCount == 0

    f1 = create_backend_file()
    f2 = create_backend_file()
    addresses = ['file://' + f1, 'file://' + f2]
    v = grpc_controller_client.volume_start(replicas=addresses)
    assert v.replicaCount == 2

    v = grpc_controller_client.volume_shutdown()
    assert v.replicaCount == 0

    rs = grpc_controller_client.replica_list()
    assert len(rs) == 0

    cleanup_backend_file([f1, f2])
示例#4
0
def test_start(grpc_controller_client):  # NOQA
    v = grpc_controller_client.volume_get()
    assert v.replicaCount == 0

    f1 = create_backend_file()
    f2 = create_backend_file()
    addresses = ['file://' + f1, 'file://' + f2]
    v = grpc_controller_client.volume_start(replicas=addresses)

    rs = grpc_controller_client.replica_list()
    assert len(rs) == 2
    assert v.replicaCount == 2

    found_addresses = [r.address for r in rs]
    assert set(found_addresses) == set(addresses)

    cleanup_backend_file([f1, f2])
示例#5
0
def test_controller_expand(grpc_controller_client):  # NOQA
    v = grpc_controller_client.volume_get()
    assert v.replicaCount == 0

    f1 = create_backend_file()
    f2 = create_backend_file()
    addresses = ['file://' + f1, 'file://' + f2]
    v = grpc_controller_client.volume_start(replicas=addresses)
    assert v.replicaCount == 2

    expand_volume_with_frontend(grpc_controller_client, EXPANDED_SIZE)
    wait_and_check_volume_expansion(grpc_controller_client, EXPANDED_SIZE)
    f1_size = os.path.getsize(f1)
    f2_size = os.path.getsize(f2)
    assert f1_size == f2_size == EXPANDED_SIZE

    v = grpc_controller_client.volume_shutdown()
    assert v.replicaCount == 0
示例#6
0
def test_replica_change(grpc_controller_client):  # NOQA
    f = create_backend_file()
    r1 = grpc_controller_client.replica_create(address='file://' + f)
    assert r1.mode == 'WO'

    r1 = grpc_controller_client.replica_update(r1.address, mode='RW')
    assert r1.mode == 'RW'

    r1 = grpc_controller_client.replica_get(r1.address)
    assert r1.mode == 'RW'

    cleanup_backend_file([f])