/* * 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.api.services; import java.lang.reflect.Method; import java.util.ArrayList; import java.util.List; import java.util.Map; import javax.servlet.http.HttpServletResponse; import javax.ws.rs.core.HttpHeaders; import javax.ws.rs.core.Response; import javax.ws.rs.core.UriInfo; import org.apache.ambari.server.api.resources.ResourceInstance; import org.apache.ambari.server.api.services.parsers.RequestBodyParser; import org.apache.ambari.server.api.services.serializers.ResultSerializer; import org.apache.ambari.server.controller.spi.Resource.Type; import org.easymock.EasyMock; import org.junit.Test; import junit.framework.Assert; /** * Unit tests for GroupService. */ public class GroupPrivilegeServiceTest extends BaseServiceTest { public List<ServiceTestInvocation> getTestInvocations() throws Exception { List<ServiceTestInvocation> listInvocations = new ArrayList<>(); GroupPrivilegeService groupPrivilegeService; Method m; Object[] args; //getPrivilege groupPrivilegeService = new TestGroupPrivilegeService(); m = groupPrivilegeService.getClass().getMethod("getPrivilege", HttpHeaders.class, UriInfo.class, String.class); args = new Object[] {getHttpHeaders(), getUriInfo(), "id"}; listInvocations.add(new ServiceTestInvocation(Request.Type.GET, groupPrivilegeService, m, args, null)); //getPrivileges groupPrivilegeService = new TestGroupPrivilegeService(); m = groupPrivilegeService.getClass().getMethod("getPrivileges", HttpHeaders.class, UriInfo.class); args = new Object[] {getHttpHeaders(), getUriInfo()}; listInvocations.add(new ServiceTestInvocation(Request.Type.GET, groupPrivilegeService, m, args, null)); return listInvocations; } @Test public void testDisabledMethods() { final HttpHeaders headers = EasyMock.createNiceMock(HttpHeaders.class); final UriInfo uriInfo = EasyMock.createNiceMock(UriInfo.class); final GroupPrivilegeService service = new TestGroupPrivilegeService(); final List<Response> disabledMethods = new ArrayList<>(); disabledMethods.add(service.createPrivilege("test", headers, uriInfo)); disabledMethods.add(service.updatePrivilege("test", headers, uriInfo, "test")); disabledMethods.add(service.updatePrivileges("test", headers, uriInfo)); disabledMethods.add(service.deletePrivilege(headers, uriInfo, "test")); disabledMethods.add(service.deletePrivileges("test", headers, uriInfo)); for (Response response: disabledMethods) { Assert.assertEquals(HttpServletResponse.SC_NOT_IMPLEMENTED, response.getStatus()); } } private class TestGroupPrivilegeService extends GroupPrivilegeService { public TestGroupPrivilegeService() { super("group"); } @Override protected ResourceInstance createResource(Type type, Map<Type, String> mapIds) { return getTestResource(); } @Override RequestFactory getRequestFactory() { return getTestRequestFactory(); } @Override protected RequestBodyParser getBodyParser() { return getTestBodyParser(); } @Override protected ResultSerializer getResultSerializer() { return getTestResultSerializer(); } } }