/**
* Copyright (C) 2009-2013 FoundationDB, LLC
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.foundationdb.server.test;
import com.foundationdb.server.service.config.ConfigurationService;
import com.foundationdb.server.service.servicemanager.GuicedServiceManager;
import com.foundationdb.sql.LayerInfoInterface;
import com.foundationdb.sql.Main;
import com.google.inject.Inject;
import org.junit.Test;
import java.util.Map;
import static org.junit.Assert.assertFalse;
public final class FailureOnStartupIT extends ApiTestBase {
@Override
protected GuicedServiceManager.BindingsConfigurationProvider serviceBindingsProvider() {
return super.serviceBindingsProvider()
.bind(LayerInfoInterface.class, BadLayerInfoInterface.class);
}
@Override
protected Map<String, String> startupConfigProperties() {
return uniqueStartupConfigProperties(getClass());
}
@Test
public void one() {
// failure would have happened in @Before
assertFalse(stage == Stage.FIRST_START);
}
@Test
public void two() {
// failure would have happened in @Before, but hopefully not at all!
// we can't assert that stage is SUBSEQUENT, because JUnit doesn't guarantee ordering
assertFalse(stage == Stage.FIRST_START);
}
public FailureOnStartupIT() {
super("IT");
}
@Override
void handleStartupFailure(Exception e) throws Exception {
// eat only the first failure
if (stage == Stage.FIRST_FAILURE) {
stage = Stage.SUBSEQUENT;
return;
}
throw e;
}
private static Stage stage = Stage.FIRST_START;
private enum Stage {
FIRST_START,
FIRST_FAILURE,
SUBSEQUENT
}
public static class BadLayerInfoInterface extends Main
{
@Inject
public BadLayerInfoInterface(ConfigurationService config) {
super(config);
}
@Override
public void start() {
if (stage == Stage.FIRST_START) {
stage = Stage.FIRST_FAILURE;
throw new RuntimeException();
}
super.start();
}
}
}