com.amazonaws.services.simpleworkflow.flow.junit.GenericWorkflowTest Maven / Gradle / Ivy

Go to download
/*
 * 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.junit;

import com.amazonaws.services.simpleworkflow.flow.generic.ActivityImplementationFactory;
import com.amazonaws.services.simpleworkflow.flow.generic.WorkflowDefinitionFactoryFactory;
import com.amazonaws.services.simpleworkflow.flow.test.TestDecisionContext;
import com.amazonaws.services.simpleworkflow.flow.test.TestGenericActivityClient;
import com.amazonaws.services.simpleworkflow.flow.test.TestGenericWorkflowClient;
import com.amazonaws.services.simpleworkflow.flow.test.TestWorkflowClock;
import com.amazonaws.services.simpleworkflow.flow.test.TestWorkflowContext;

public class GenericWorkflowTest extends WorkflowTestBase {

    private TestGenericActivityClient activityClient;

    public GenericWorkflowTest(WorkflowDefinitionFactoryFactory factoryFactory) {
        super(new TestDecisionContext(new TestGenericActivityClient(),
                new TestGenericWorkflowClient(factoryFactory), new TestWorkflowClock(), new TestWorkflowContext()));
        activityClient = (TestGenericActivityClient) decisionContext.getActivityClient();
    }

    public void addFactory(String taskListToListen, ActivityImplementationFactory factory) {
        activityClient.addFactory(taskListToListen, factory);
    }

    public void addFactory(ActivityImplementationFactory factory) {
        addFactory(defaultActivitiesTaskListToPoll, factory);
    }

}