TestConfiguredPolicy.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.security.authorize;
  19. import java.security.Permission;
  20. import javax.security.auth.Subject;
  21. import org.apache.hadoop.conf.Configuration;
  22. import org.apache.hadoop.security.SecurityUtil;
  23. import org.apache.hadoop.security.UnixUserGroupInformation;
  24. import org.apache.hadoop.security.SecurityUtil.AccessControlList;
  25. import junit.framework.TestCase;
  26. public class TestConfiguredPolicy extends TestCase {
  27.   private static final String USER1 = "drwho";
  28.   private static final String USER2 = "joe";
  29.   private static final String[] GROUPS1 = new String[]{"tardis"};
  30.   private static final String[] GROUPS2 = new String[]{"users"};
  31.   
  32.   private static final String KEY_1 = "test.policy.1";
  33.   private static final String KEY_2 = "test.policy.2";
  34.   
  35.   public static class Protocol1 {
  36.     int i;
  37.   }
  38.   public static class Protocol2 {
  39.     int j;
  40.   }
  41.   
  42.   private static class TestPolicyProvider extends PolicyProvider {
  43.     @Override
  44.     public Service[] getServices() {
  45.       return new Service[] {
  46.           new Service(KEY_1, Protocol1.class),
  47.           new Service(KEY_2, Protocol2.class),
  48.           };
  49.     }
  50.   }
  51.   
  52.   public void testConfiguredPolicy() throws Exception {
  53.     Configuration conf = new Configuration();
  54.     conf.set(KEY_1, AccessControlList.WILDCARD_ACL_VALUE);
  55.     conf.set(KEY_2, USER1 + " " + GROUPS1[0]);
  56.     
  57.     ConfiguredPolicy policy = new ConfiguredPolicy(conf, new TestPolicyProvider());
  58.     SecurityUtil.setPolicy(policy);
  59.     
  60.     Subject user1 = 
  61.       SecurityUtil.getSubject(new UnixUserGroupInformation(USER1, GROUPS1));
  62.     // Should succeed
  63.     ServiceAuthorizationManager.authorize(user1, Protocol1.class);
  64.     
  65.     // Should fail
  66.     Subject user2 = 
  67.       SecurityUtil.getSubject(new UnixUserGroupInformation(USER2, GROUPS2));
  68.     boolean failed = false;
  69.     try {
  70.       ServiceAuthorizationManager.authorize(user2, Protocol2.class);
  71.     } catch (AuthorizationException ae) {
  72.       failed = true;
  73.     }
  74.     assertTrue(failed);
  75.   }
  76. }