hdfs/src/test/unit/org/apache/hadoop/hdfs/server/namenode/TestGetImageServlet.java

/**
 * 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.hadoop.hdfs.server.namenode;

import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_KRB_HTTPS_USER_NAME_KEY;
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_USER_NAME_KEY;
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_SECONDARY_NAMENODE_KRB_HTTPS_USER_NAME_KEY;
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_SECONDARY_NAMENODE_USER_NAME_KEY;
import static org.junit.Assert.*;

import java.io.IOException;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.hdfs.DFSConfigKeys;
import org.apache.hadoop.security.SecurityUtil;
import org.junit.Test;

public class TestGetImageServlet {
  private static final String HOST = "foo.com";
  private static final String KERBEROS_DOMAIN = "@HADOOP.ORG";
  
  private static Configuration getConf() {
    Configuration conf = new Configuration();
    FileSystem.setDefaultUri(conf, "hdfs://" + HOST);
    conf.set(DFSConfigKeys.DFS_NAMENODE_SECONDARY_HTTP_ADDRESS_KEY, HOST
        + ":50090");
    return conf;
  }
  
  // Worker class to poke the isValidRequestor method with verifying it accepts
  // or rejects with these standard allowed principals
  private void verifyIsValidReqBehavior(GetImageServlet gim, 
                                        boolean shouldSucceed, String msg) 
      throws IOException {
    final String [] validRequestors = {DFS_NAMENODE_KRB_HTTPS_USER_NAME_KEY,
                                       DFS_NAMENODE_USER_NAME_KEY,
                                       DFS_SECONDARY_NAMENODE_KRB_HTTPS_USER_NAME_KEY,
                                       DFS_SECONDARY_NAMENODE_USER_NAME_KEY };
    
    Configuration conf = getConf();
    for(String v : validRequestors) {
      conf.set(v, "a/" + SecurityUtil.HOSTNAME_PATTERN + KERBEROS_DOMAIN);
      assertEquals(msg + v, gim.isValidRequestor(shouldSucceed ? "a/" + HOST
          + KERBEROS_DOMAIN : "b/" + HOST + KERBEROS_DOMAIN, conf),
          shouldSucceed);
    }
  }
  
  @Test
  public void IsValidRequestorAcceptsCorrectly() throws IOException {
    GetImageServlet gim = new GetImageServlet();

    verifyIsValidReqBehavior(gim, true, 
        "isValidRequestor has rejected a valid requestor: ");
  }
  
  @Test
  public void IsValidRequestorRejectsCorrectly() throws IOException {
    GetImageServlet gim = new GetImageServlet();
    
    // Don't set any valid requestors
    assertFalse("isValidRequestor allowed a requestor despite no values being set",
                gim.isValidRequestor("not set", getConf()));
    
    verifyIsValidReqBehavior(gim, false, 
        "isValidRequestor has allowed an invalid requestor: ");
  }

}