/** * 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.msi; import junit.framework.Assert; import org.apache.ambari.scom.TestClusterDefinitionProvider; import org.apache.ambari.scom.TestHostInfoProvider; import org.apache.ambari.server.controller.internal.ResourceImpl; import org.apache.ambari.server.controller.spi.Resource; import org.apache.ambari.server.controller.utilities.PropertyHelper; import org.junit.Test; import java.util.HashMap; import java.util.Set; /** * Tests for RequestProvider */ public class RequestProviderTest { @Test public void testGetResources() throws Exception { ClusterDefinition clusterDefinition = new ClusterDefinition(new TestStateProvider(), new TestClusterDefinitionProvider(), new TestHostInfoProvider()); clusterDefinition.getRequestResources().add(new ResourceImpl(Resource.Type.Request)); RequestProvider provider = new RequestProvider(clusterDefinition); Set<Resource> resources = provider.getResources(PropertyHelper.getReadRequest(), null); Assert.assertEquals(1, resources.size()); } @Test public void testCreateResources() throws Exception { ClusterDefinition clusterDefinition = new ClusterDefinition(new TestStateProvider(), new TestClusterDefinitionProvider(), new TestHostInfoProvider()); RequestProvider provider = new RequestProvider(clusterDefinition); try { provider.createResources(PropertyHelper.getReadRequest()); Assert.fail("Expected UnsupportedOperationException."); } catch (UnsupportedOperationException e) { //expected } } @Test public void testUpdateResources() throws Exception { ClusterDefinition clusterDefinition = new ClusterDefinition(new TestStateProvider(), new TestClusterDefinitionProvider(), new TestHostInfoProvider()); RequestProvider provider = new RequestProvider(clusterDefinition); provider.updateResources(PropertyHelper.getUpdateRequest(new HashMap<String, Object>(), null), null); } @Test public void testDeleteResources() throws Exception { ClusterDefinition clusterDefinition = new ClusterDefinition(new TestStateProvider(), new TestClusterDefinitionProvider(), new TestHostInfoProvider()); RequestProvider provider = new RequestProvider(clusterDefinition); try { provider.deleteResources(null); Assert.fail("Expected UnsupportedOperationException."); } catch (UnsupportedOperationException e) { //expected } } }