/*
* Copyright 2012-2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package com.amazonaws.services.simpleworkflow.flow.examples.deployment;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import junit.framework.Assert;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import com.amazonaws.services.simpleworkflow.flow.core.TryCatchFinally;
import com.amazonaws.services.simpleworkflow.flow.junit.WorkflowTestBase;
import com.amazonaws.services.simpleworkflow.flow.junit.spring.FlowSpringJUnit4ClassRunner;
@RunWith(FlowSpringJUnit4ClassRunner.class)
@ContextConfiguration("DeploymentTest-context.xml")
public class DeploymentTest {
public static class TestDeploymentActivities implements DeploymentActivities {
private final String host;
public TestDeploymentActivities(String host) {
this.host = host;
}
@Override
public String deployDatabase() {
trace.add("Database-" + host);
return "dbUrl-" + host;
}
@Override
public String deployAppServer(List<String> dataSources) {
trace.add("appServer-" + host);
return "appServerUrl-" + host;
}
@Override
public String deployWebServer(List<String> appServerUrls, List<String> dataSources) {
trace.add("WebServer-" + host);
return "webServerUrl-" + host;
}
@Override
public String deployLoadBalancer(List<String> frontendUrls) {
trace.add("LoadBalancer-" + host);
return "loadBalancerUrl-" + host;
}
}
@Rule
@Autowired
public WorkflowTestBase workflowTest;
@Autowired
public DeploymentWorkflowClientFactory workflowClientFactory;
private static final List<String> trace = new ArrayList<String>();
@Test
public void test() throws IOException {
new TryCatchFinally() {
Throwable failure;
@Override
protected void doTry() throws Throwable {
String template = getTemplate("AppStack1.xml");
DeploymentWorkflowClient workflow = workflowClientFactory.getClient("AppStack1");
workflow.deploy(template);
}
@Override
protected void doCatch(Throwable e) throws Throwable {
failure = e;
throw e;
}
@Override
protected void doFinally() throws Throwable {
// Without this check assertEquals fails overwriting
// original exception
if (failure == null) {
List<String> expected = new ArrayList<String>();
expected.add("Database-host1");
expected.add("Database-host2");
expected.add("appServer-host3");
expected.add("appServer-host2");
expected.add("appServer-host3");
expected.add("WebServer-host2");
expected.add("WebServer-host1");
expected.add("LoadBalancer-host2");
Assert.assertEquals(expected, trace);
}
}
};
}
private String getTemplate(String resourceName) throws IOException {
URL resource = DeploymentTest.class.getResource(resourceName);
String template = new Scanner(resource.openStream()).useDelimiter("\\A").next();
return template;
}
}