Examples with MockClusterMap com.github.ambry.clustermap.MockClusterMap used on opensource projects

Example 36 with MockClusterMap

use of com.github.ambry.clustermap.MockClusterMap in project ambry by linkedin.

the class RequestResponseTest method adminRequestResponseTest.

/**
 * Tests the ser/de of {@link AdminRequest} and {@link AdminResponse} and checks for equality of fields with
 * reference data.
 * @throws IOException
 */
@Test
public void adminRequestResponseTest() throws IOException {
    int correlationId = 1234;
    String clientId = "client";
    for (AdminRequestOrResponseType type : AdminRequestOrResponseType.values()) {
        MockClusterMap clusterMap = new MockClusterMap();
        PartitionId id = clusterMap.getWritablePartitionIds(MockClusterMap.DEFAULT_PARTITION_CLASS).get(0);
        // with a valid partition id
        AdminRequest adminRequest = new AdminRequest(type, id, correlationId, clientId);
        DataInputStream requestStream = serAndPrepForRead(adminRequest, -1, true);
        deserAdminRequestAndVerify(requestStream, clusterMap, correlationId, clientId, type, id);
        adminRequest.release();
        // with a null partition id
        adminRequest = new AdminRequest(type, null, correlationId, clientId);
        requestStream = serAndPrepForRead(adminRequest, -1, true);
        deserAdminRequestAndVerify(requestStream, clusterMap, correlationId, clientId, type, null);
        adminRequest.release();
        // response
        ServerErrorCode[] values = ServerErrorCode.values();
        int indexToPick = TestUtils.RANDOM.nextInt(values.length);
        ServerErrorCode responseErrorCode = values[indexToPick];
        AdminResponse response = new AdminResponse(correlationId, clientId, responseErrorCode);
        DataInputStream responseStream = serAndPrepForRead(response, -1, false);
        AdminResponse deserializedAdminResponse = AdminResponse.readFrom(responseStream);
        Assert.assertEquals(deserializedAdminResponse.getCorrelationId(), correlationId);
        Assert.assertEquals(deserializedAdminResponse.getClientId(), clientId);
        Assert.assertEquals(deserializedAdminResponse.getError(), responseErrorCode);
        response.release();
    }
}

Example 37 with MockClusterMap

use of com.github.ambry.clustermap.MockClusterMap in project ambry by linkedin.

the class RequestResponseTest method undeleteRequestResponseTest.

/**
 * Test for undelete request and response.
 * @throws IOException
 */
@Test
public void undeleteRequestResponseTest() throws IOException {
    MockClusterMap clusterMap = new MockClusterMap();
    short accountId = Utils.getRandomShort(TestUtils.RANDOM);
    short containerId = Utils.getRandomShort(TestUtils.RANDOM);
    BlobId id1 = new BlobId(CommonTestUtils.getCurrentBlobIdVersion(), BlobId.BlobIdType.NATIVE, ClusterMap.UNKNOWN_DATACENTER_ID, accountId, containerId, clusterMap.getWritablePartitionIds(MockClusterMap.DEFAULT_PARTITION_CLASS).get(0), false, BlobId.BlobDataType.DATACHUNK);
    int correlationId = TestUtils.RANDOM.nextInt();
    long operationTimeMs = SystemTime.getInstance().milliseconds() + TestUtils.RANDOM.nextInt();
    UndeleteRequest undeleteRequest = new UndeleteRequest(correlationId, "client", id1, operationTimeMs);
    DataInputStream requestStream = serAndPrepForRead(undeleteRequest, -1, true);
    UndeleteRequest deserializedUndeleteRequest = UndeleteRequest.readFrom(requestStream, clusterMap);
    Assert.assertEquals(deserializedUndeleteRequest.getClientId(), "client");
    Assert.assertEquals(deserializedUndeleteRequest.getBlobId(), id1);
    Assert.assertEquals("AccountId mismatch ", id1.getAccountId(), deserializedUndeleteRequest.getAccountId());
    Assert.assertEquals("ContainerId mismatch ", id1.getContainerId(), deserializedUndeleteRequest.getContainerId());
    Assert.assertEquals("OperationTimeMs mismatch ", operationTimeMs, deserializedUndeleteRequest.getOperationTimeMs());
    undeleteRequest.release();
    UndeleteResponse response = null;
    try {
        response = new UndeleteResponse(correlationId, "client", ServerErrorCode.No_Error);
        Assert.fail("No Error is not a valid error node for this response");
    } catch (IllegalArgumentException e) {
    // do nothing
    }
    response = new UndeleteResponse(correlationId, "client", ServerErrorCode.Blob_Deleted);
    requestStream = serAndPrepForRead(response, -1, false);
    UndeleteResponse deserializedUndeleteResponse = UndeleteResponse.readFrom(requestStream);
    Assert.assertEquals(deserializedUndeleteResponse.getCorrelationId(), correlationId);
    Assert.assertEquals(deserializedUndeleteResponse.getError(), ServerErrorCode.Blob_Deleted);
    Assert.assertEquals(deserializedUndeleteResponse.getLifeVersion(), UndeleteResponse.INVALID_LIFE_VERSION);
    response.release();
    try {
        response = new UndeleteResponse(correlationId, "client", UndeleteResponse.INVALID_LIFE_VERSION);
        Assert.fail("Not an valid life version");
    } catch (IllegalArgumentException e) {
    // do nothing
    }
    short lifeVersion = 1;
    response = new UndeleteResponse(correlationId, "client", lifeVersion);
    requestStream = serAndPrepForRead(response, -1, false);
    deserializedUndeleteResponse = UndeleteResponse.readFrom(requestStream);
    Assert.assertEquals(deserializedUndeleteResponse.getCorrelationId(), correlationId);
    Assert.assertEquals(deserializedUndeleteResponse.getError(), ServerErrorCode.No_Error);
    Assert.assertEquals(deserializedUndeleteResponse.getLifeVersion(), lifeVersion);
    response.release();
}

Example 38 with MockClusterMap

use of com.github.ambry.clustermap.MockClusterMap in project ambry by linkedin.

the class RequestResponseTest method doRequestControlAdminRequestTest.

/**
 * Does the actual test of ser/de of {@link RequestControlAdminRequest} and checks for equality of fields with
 * reference data.
 * @param requestOrResponseType the {@link RequestOrResponseType} to be used in the {@link RequestControlAdminRequest}
 * @param enable the value for the enable field in {@link RequestControlAdminRequest}.
 * @throws IOException
 */
private void doRequestControlAdminRequestTest(RequestOrResponseType requestOrResponseType, boolean enable) throws IOException {
    MockClusterMap clusterMap = new MockClusterMap();
    PartitionId id = clusterMap.getWritablePartitionIds(MockClusterMap.DEFAULT_PARTITION_CLASS).get(0);
    int correlationId = 1234;
    String clientId = "client";
    AdminRequest adminRequest = new AdminRequest(AdminRequestOrResponseType.RequestControl, id, correlationId, clientId);
    RequestControlAdminRequest controlRequest = new RequestControlAdminRequest(requestOrResponseType, enable, adminRequest);
    DataInputStream requestStream = serAndPrepForRead(controlRequest, -1, true);
    AdminRequest deserializedAdminRequest = deserAdminRequestAndVerify(requestStream, clusterMap, correlationId, clientId, AdminRequestOrResponseType.RequestControl, id);
    RequestControlAdminRequest deserializedControlRequest = RequestControlAdminRequest.readFrom(requestStream, deserializedAdminRequest);
    Assert.assertEquals(requestOrResponseType, deserializedControlRequest.getRequestTypeToControl());
    Assert.assertEquals(enable, deserializedControlRequest.shouldEnable());
    controlRequest.release();
}

Example 39 with MockClusterMap

use of com.github.ambry.clustermap.MockClusterMap in project ambry by linkedin.

the class DiskTokenPersistorTest method setup.

/**
 * Create the one time setup for the tests.
 * @throws Exception if Exception happens during setup.
 */
@BeforeClass
public static void setup() throws Exception {
    clusterMap = new MockClusterMap();
    mountPathToPartitionInfoList = new HashMap<>();
    mountPathToReplicaTokenInfos = new HashMap<>();
    BlobIdFactory blobIdFactory = new BlobIdFactory(clusterMap);
    StoreFindTokenFactory factory = new StoreFindTokenFactory(blobIdFactory);
    DataNodeId dataNodeId = clusterMap.getDataNodeIds().get(0);
    List<? extends ReplicaId> localReplicas = clusterMap.getReplicaIds(dataNodeId);
    replicaId = localReplicas.get(0);
    for (ReplicaId replicaId : localReplicas) {
        List<? extends ReplicaId> peerReplicas = replicaId.getPeerReplicaIds();
        List<RemoteReplicaInfo> remoteReplicas = new ArrayList<>();
        for (ReplicaId remoteReplica : peerReplicas) {
            RemoteReplicaInfo remoteReplicaInfo = new RemoteReplicaInfo(remoteReplica, replicaId, null, factory.getNewFindToken(), 10, SystemTime.getInstance(), remoteReplica.getDataNodeId().getPortToConnectTo());
            remoteReplicas.add(remoteReplicaInfo);
            mountPathToReplicaTokenInfos.computeIfAbsent(replicaId.getMountPath(), k -> new ArrayList<>()).add(new RemoteReplicaInfo.ReplicaTokenInfo(remoteReplicaInfo));
        }
        PartitionInfo partitionInfo = new PartitionInfo(remoteReplicas, replicaId.getPartitionId(), null, replicaId);
        mountPathToPartitionInfoList.computeIfAbsent(replicaId.getMountPath(), key -> ConcurrentHashMap.newKeySet()).add(partitionInfo);
    }
    Properties replicationProperties = new Properties();
    replicationProperties.setProperty("replication.cloud.token.factory", MockFindTokenFactory.class.getName());
    ReplicationConfig replicationConfig = new ReplicationConfig(new VerifiableProperties(replicationProperties));
    findTokenHelper = new FindTokenHelper(blobIdFactory, replicationConfig);
    mockStoreManager = Mockito.mock(StoreManager.class);
    Mockito.when(mockStoreManager.checkLocalPartitionStatus(any(), any())).thenReturn(ServerErrorCode.No_Error);
}

Example 40 with MockClusterMap

use of com.github.ambry.clustermap.MockClusterMap in project ambry by linkedin.

the class LeaderBasedReplicationTest method setUp.

public void setUp() throws IOException {
    properties.setProperty("replication.model.across.datacenters", "LEADER_BASED");
    replicationConfig = new ReplicationConfig(new VerifiableProperties(properties));
    clusterMap = new MockClusterMap();
    clusterMapConfig = new ClusterMapConfig(verifiableProperties);
    MockHelixParticipant.metricRegistry = new MetricRegistry();
    mockHelixParticipant = new MockHelixParticipant(clusterMapConfig);
    /*
      Setup:
      we have 3 nodes that have replicas belonging to same partitions:
      a) localNode (local node that hosts partitions)
      b) remoteNodeInLocalDC (remote node in local data center that shares the partitions)
      c) remoteNodeInRemoteDC (remote node in remote data center that shares the partitions)

      Each node have few of its partitions as leaders and others are standby. They are randomly assigned during creation
      of replicas for mock partitions.
     */
    DataNodeId localNode = clusterMap.getDataNodeIds().get(0);
    List<DataNodeId> remoteNodes = getRemoteNodesFromLocalAndRemoteDCs(clusterMap, localNode);
    remoteNodeInLocalDC = remoteNodes.get(0);
    remoteNodeInRemoteDC = remoteNodes.get(1);
    // mock hosts for remote nodes
    localHost = new MockHost(localNode, clusterMap);
    remoteHostInLocalDC = new MockHost(remoteNodeInLocalDC, clusterMap);
    remoteHostInRemoteDC = new MockHost(remoteNodeInRemoteDC, clusterMap);
}