/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License 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 org.apache.ambari.server.controller.internal;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.expectLastCall;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import org.apache.ambari.server.controller.internal.WorkflowResourceProvider.WorkflowFetcher;
import org.apache.ambari.server.controller.jdbc.ConnectionFactory;
import org.apache.ambari.server.controller.spi.NoSuchParentResourceException;
import org.apache.ambari.server.controller.spi.NoSuchResourceException;
import org.apache.ambari.server.controller.spi.Predicate;
import org.apache.ambari.server.controller.spi.Request;
import org.apache.ambari.server.controller.spi.Resource;
import org.apache.ambari.server.controller.spi.Resource.Type;
import org.apache.ambari.server.controller.spi.ResourceProvider;
import org.apache.ambari.server.controller.spi.SystemException;
import org.apache.ambari.server.controller.spi.UnsupportedPropertyException;
import org.apache.ambari.server.controller.utilities.PredicateBuilder;
import org.apache.ambari.server.controller.utilities.PropertyHelper;
import org.junit.Assert;
import org.junit.Test;
/**
* WorkflowResourceProvider tests.
*/
public class WorkflowResourceProviderTest {
@Test
public void testGetResources() throws SystemException,
UnsupportedPropertyException, NoSuchResourceException,
NoSuchParentResourceException {
Set<Resource> expected = new HashSet<Resource>();
expected.add(createWorkflowResponse("Cluster100", "workflow1"));
expected.add(createWorkflowResponse("Cluster100", "workflow2"));
expected.add(createWorkflowResponse("Cluster100", "workflow3"));
Resource.Type type = Resource.Type.Workflow;
Set<String> propertyIds = PropertyHelper.getPropertyIds(type);
WorkflowFetcher workflowFetcher = createMock(WorkflowFetcher.class);
expect(workflowFetcher.fetchWorkflows(propertyIds, "Cluster100", null))
.andReturn(expected).once();
replay(workflowFetcher);
Map<Resource.Type,String> keyPropertyIds = PropertyHelper
.getKeyPropertyIds(type);
ResourceProvider provider = new WorkflowResourceProvider(propertyIds,
keyPropertyIds, workflowFetcher);
Request request = PropertyHelper.getReadRequest(propertyIds);
Predicate predicate = new PredicateBuilder()
.property(WorkflowResourceProvider.WORKFLOW_CLUSTER_NAME_PROPERTY_ID)
.equals("Cluster100").toPredicate();
Set<Resource> resources = provider.getResources(request, predicate);
Assert.assertEquals(3, resources.size());
Set<String> names = new HashSet<String>();
for (Resource resource : resources) {
String clusterName = (String) resource
.getPropertyValue(WorkflowResourceProvider.WORKFLOW_CLUSTER_NAME_PROPERTY_ID);
Assert.assertEquals("Cluster100", clusterName);
names.add((String) resource
.getPropertyValue(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID));
}
// Make sure that all of the response objects got moved into resources
for (Resource resource : expected) {
Assert.assertTrue(names.contains(resource
.getPropertyValue(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID)));
}
verify(workflowFetcher);
}
@Test
public void testWorkflowFetcher() throws SystemException,
UnsupportedPropertyException, NoSuchResourceException,
NoSuchParentResourceException {
Set<String> requestedIds = new HashSet<String>();
requestedIds.add(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID);
Map<Resource.Type,String> keyPropertyIds = PropertyHelper
.getKeyPropertyIds(Resource.Type.Workflow);
ResourceProvider provider = new TestWorkflowResourceProvider(requestedIds,
keyPropertyIds);
Request request = PropertyHelper.getReadRequest(requestedIds);
Predicate predicate = new PredicateBuilder()
.property(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID)
.equals("workflow1").toPredicate();
Set<Resource> resources = provider.getResources(request, predicate);
Assert.assertEquals(1, resources.size());
for (Resource resource : resources) {
String workflowId = (String) resource
.getPropertyValue(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID);
Assert.assertEquals("workflow1", workflowId);
}
}
private static Resource createWorkflowResponse(String clusterName,
String workflowId) {
Resource r = new ResourceImpl(Resource.Type.Workflow);
r.setProperty(WorkflowResourceProvider.WORKFLOW_CLUSTER_NAME_PROPERTY_ID,
clusterName);
r.setProperty(WorkflowResourceProvider.WORKFLOW_ID_PROPERTY_ID, workflowId);
return r;
}
private static class TestWorkflowResourceProvider extends
WorkflowResourceProvider {
protected TestWorkflowResourceProvider(Set<String> propertyIds,
Map<Type,String> keyPropertyIds) {
super(propertyIds, keyPropertyIds, null);
this.workflowFetcher = new TestWorkflowFetcher();
}
private class TestWorkflowFetcher extends PostgresWorkflowFetcher {
ResultSet rs = null;
public TestWorkflowFetcher() {
super((ConnectionFactory) null);
}
@Override
protected ResultSet getResultSet(Set<String> requestedIds,
String workflowId) throws SQLException {
rs = createMock(ResultSet.class);
expect(rs.next()).andReturn(true).once();
expect(rs.getString(getDBField(WORKFLOW_ID_PROPERTY_ID).toString()))
.andReturn("workflow1").once();
expect(rs.next()).andReturn(false).once();
rs.close();
expectLastCall().once();
replay(rs);
return rs;
}
@Override
protected void close() {
verify(rs);
}
}
}
}