@Test(expected = IllegalStateException.class)
public void rollbackThrowsIfIllegalIdInput() throws Exception {

  rollbackAction.inputStream = yes;
  rollbackActionConfig.name = secret.getDisplayName();
  rollbackActionConfig.id = 1L;
  when(keywhizClient.getSanitizedSecretByName(secret.getName())).thenReturn(sanitizedSecret);
  when(keywhizClient.rollbackSecret(secret.getDisplayName(), 1L)).thenThrow(new IllegalStateException());
  rollbackAction.run();
}