use of com.palantir.atlasdb.timelock.api.DisableNamespacesResponse in project atlasdb by palantir.
the class AtlasRestoreServiceTest method prepareBackupFailsIfDisableFails.
@Test
public void prepareBackupFailsIfDisableFails() {
DisableNamespacesResponse failedDisable = DisableNamespacesResponse.unsuccessful(UnsuccessfulDisableNamespacesResponse.of(ImmutableSet.of(WITH_BACKUP), ImmutableSet.of()));
DisableNamespacesRequest request = DisableNamespacesRequest.of(ImmutableSet.of(WITH_BACKUP), BACKUP_ID);
when(timeLockManagementService.disableTimelock(authHeader, request)).thenReturn(failedDisable);
Set<Namespace> disabledNamespaces = atlasRestoreService.prepareRestore(ImmutableSet.of(restoreRequest(WITH_BACKUP), restoreRequest(NO_BACKUP)), BACKUP_ID);
assertThat(disabledNamespaces).isEmpty();
}
use of com.palantir.atlasdb.timelock.api.DisableNamespacesResponse in project atlasdb by palantir.
the class AtlasRestoreServiceTest method restoresToNewNamespaceCorrectly.
@Test
public void restoresToNewNamespaceCorrectly() {
RestoreRequest restoreRequest = RestoreRequest.builder().oldNamespace(WITH_BACKUP).newNamespace(NO_BACKUP).build();
// prepare
DisableNamespacesResponse successfulDisable = DisableNamespacesResponse.successful(SuccessfulDisableNamespacesResponse.of(BACKUP_ID));
DisableNamespacesRequest request = DisableNamespacesRequest.of(ImmutableSet.of(NO_BACKUP), BACKUP_ID);
when(timeLockManagementService.disableTimelock(authHeader, request)).thenReturn(successfulDisable);
Set<Namespace> disabledNamespaces = atlasRestoreService.prepareRestore(ImmutableSet.of(restoreRequest), BACKUP_ID);
assertThat(disabledNamespaces).containsExactly(NO_BACKUP);
// repair
BiConsumer<String, RangesForRepair> doNothingConsumer = (_unused1, _unused2) -> {
};
Set<Namespace> repairedNamespaces = atlasRestoreService.repairInternalTables(ImmutableSet.of(restoreRequest), doNothingConsumer);
assertThat(repairedNamespaces).containsExactly(NO_BACKUP);
verify(cassandraRepairHelper).repairInternalTables(NO_BACKUP, doNothingConsumer);
verify(cassandraRepairHelper).repairTransactionsTables(eq(NO_BACKUP), anyList(), eq(doNothingConsumer));
verify(cassandraRepairHelper).cleanTransactionsTables(eq(NO_BACKUP), eq(BACKUP_START_TIMESTAMP), anyList());
verifyNoMoreInteractions(cassandraRepairHelper);
// complete
CompletedBackup completedBackup = backupPersister.getCompletedBackup(WITH_BACKUP).orElseThrow();
CompleteRestoreRequest completeRestoreRequest = CompleteRestoreRequest.of(ImmutableMap.of(NO_BACKUP, completedBackup));
when(atlasRestoreClient.completeRestore(authHeader, completeRestoreRequest)).thenReturn(CompleteRestoreResponse.of(ImmutableSet.of(NO_BACKUP)));
ReenableNamespacesRequest reenableRequest = ReenableNamespacesRequest.of(ImmutableSet.of(NO_BACKUP), BACKUP_ID);
Set<Namespace> completedNamespaces = atlasRestoreService.completeRestore(ImmutableSet.of(restoreRequest), BACKUP_ID);
assertThat(completedNamespaces).containsExactly(NO_BACKUP);
verify(atlasRestoreClient).completeRestore(authHeader, completeRestoreRequest);
verify(timeLockManagementService).reenableTimelock(authHeader, reenableRequest);
}
use of com.palantir.atlasdb.timelock.api.DisableNamespacesResponse in project atlasdb by palantir.
the class AllNodesDisabledNamespacesUpdaterTest method doesNotDisableIfPingFailsOnOneNode.
@Test
public void doesNotDisableIfPingFailsOnOneNode() {
when(remote2.ping(any())).thenThrow(new SafeRuntimeException("unreachable"));
DisableNamespacesResponse response = updater.disableOnAllNodes(AUTH_HEADER, disableNamespacesRequest(ImmutableSet.of(NAMESPACE)));
assertThat(response).isEqualTo(DISABLE_FAILED_SUCCESSFULLY);
verify(remote1, never()).disable(any(), any());
verify(remote2, never()).disable(any(), any());
verify(localUpdater, never()).disable(any());
}
use of com.palantir.atlasdb.timelock.api.DisableNamespacesResponse in project atlasdb by palantir.
the class AllNodesDisabledNamespacesUpdaterTest method disableDoesNotReportConsistentStateWhenNamespacesAreLockedWithDifferentIds.
@Test
public void disableDoesNotReportConsistentStateWhenNamespacesAreLockedWithDifferentIds() {
Set<Namespace> namespaces = ImmutableSet.of(NAMESPACE);
String otherLockId = "otherLockId";
String yetAnotherLockId = "yetAnotherLockId";
SingleNodeUpdateResponse lockedWithOtherLock = ImmutableSingleNodeUpdateResponse.builder().isSuccessful(false).putLockedNamespaces(NAMESPACE, otherLockId).build();
SingleNodeUpdateResponse lockedWithYetAnotherLock = ImmutableSingleNodeUpdateResponse.builder().isSuccessful(false).putLockedNamespaces(NAMESPACE, yetAnotherLockId).build();
when(remote1.disable(any(), any())).thenReturn(lockedWithOtherLock);
when(remote2.disable(any(), any())).thenReturn(SingleNodeUpdateResponse.successful());
when(remote2.reenable(any(), any())).thenReturn(SingleNodeUpdateResponse.successful());
when(localUpdater.getNamespacesLockedWithDifferentLockId(any(), any())).thenReturn(lockedWithYetAnotherLock.lockedNamespaces());
DisableNamespacesResponse response = updater.disableOnAllNodes(AUTH_HEADER, disableNamespacesRequest(namespaces));
verify(remote1, never()).reenable(any(), any());
verify(remote2).reenable(any(), any());
assertThat(response).isEqualTo(partiallyDisabled(namespaces));
}
use of com.palantir.atlasdb.timelock.api.DisableNamespacesResponse in project atlasdb by palantir.
the class AllNodesDisabledNamespacesUpdaterTest method rollsBackIfLocalUpdateFails.
@Test
public void rollsBackIfLocalUpdateFails() {
when(remote1.disable(any(), any())).thenReturn(SUCCESSFUL_SINGLE_NODE_UPDATE);
when(remote2.disable(any(), any())).thenReturn(SUCCESSFUL_SINGLE_NODE_UPDATE);
Set<Namespace> failedNamespaces = ImmutableSet.of(OTHER_NAMESPACE);
when(localUpdater.disable(DisableNamespacesRequest.of(BOTH_NAMESPACES, LOCK_ID))).thenReturn(singleNodeUpdateFailure(failedNamespaces));
DisableNamespacesResponse response = updater.disableOnAllNodes(AUTH_HEADER, disableNamespacesRequest(BOTH_NAMESPACES));
assertThat(response).isEqualTo(partiallyDisabled(BOTH_NAMESPACES));
ReenableNamespacesRequest rollbackRequest = ReenableNamespacesRequest.of(BOTH_NAMESPACES, LOCK_ID);
verify(remote1).reenable(AUTH_HEADER, rollbackRequest);
verify(remote2).reenable(AUTH_HEADER, rollbackRequest);
// local update failed, so no need to roll back
verify(localUpdater, never()).reEnable(any());
}
Aggregations