PolicyContextTest.java

//========================================================================
//Copyright (c) Webtide LLC
//------------------------------------------------------------------------
//All rights reserved. This program and the accompanying materials
//are made available under the terms of the Eclipse Public License v1.0
//and Apache License v2.0 which accompanies this distribution.
//
//The Eclipse Public License is available at
//http://www.eclipse.org/legal/epl-v10.html
//
//The Apache License v2.0 is available at
//http://www.apache.org/licenses/LICENSE-2.0.txt
//
//You may elect to redistribute this code under either of these licenses.
//========================================================================

package org.eclipse.jetty.policy;

import static org.junit.Assert.assertEquals;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.security.Permission;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.jetty.policy.entry.GrantEntry;
import org.eclipse.jetty.policy.entry.KeystoreEntry;
import org.eclipse.jetty.policy.loader.PolicyFileScanner;
import org.eclipse.jetty.toolchain.test.MavenTestingUtils;
import org.eclipse.jetty.toolchain.test.OS;
import org.junit.Before;
import org.junit.Test;

public class PolicyContextTest
{
    public static final String __PRINCIPAL = "javax.security.auth.x500.X500Principal \"CN=Jetty Policy,OU=Artifact,O=Jetty Project,L=Earth,ST=Internet,C=US\"";

    @Before
    public void init() throws Exception
    {
        System.setProperty( "basedir", MavenTestingUtils.getBaseURI().toASCIIString() );
    }

    @Test
    public void testSelfPropertyExpansion() throws Exception
    {
        PolicyContext context = new PolicyContext();
        PolicyFileScanner loader = new PolicyFileScanner();
        List<GrantEntry> grantEntries = new ArrayList<GrantEntry>();
        List<KeystoreEntry> keystoreEntries = new ArrayList<KeystoreEntry>();

        File policyFile = MavenTestingUtils.getTestResourceFile("context/jetty-certificate.policy");

        loader.scanStream( new InputStreamReader( new FileInputStream( policyFile ) ), grantEntries, keystoreEntries );

        if ( !OS.IS_WINDOWS ) //temporary, create alternate file to load for windows
        {
            for (KeystoreEntry node : keystoreEntries)
            {
                node.expand(context);

                context.setKeystore(node.toKeyStore());
            }

            GrantEntry grant = grantEntries.get( 0 );
            grant.expand( context );

            Permission perm = grant.getPermissions().elements().nextElement();

            assertEquals( __PRINCIPAL, perm.getName() );
        }
    }

    @Test
    public void testAliasPropertyExpansion() throws Exception
    {
        PolicyContext context = new PolicyContext();
        PolicyFileScanner loader = new PolicyFileScanner();
        List<GrantEntry> grantEntries = new ArrayList<GrantEntry>();
        List<KeystoreEntry> keystoreEntries = new ArrayList<KeystoreEntry>();

        File policyFile = MavenTestingUtils.getTestResourceFile("context/jetty-certificate-alias.policy");

        loader.scanStream( new InputStreamReader( new FileInputStream( policyFile ) ), grantEntries, keystoreEntries );

        if ( !OS.IS_WINDOWS ) //temporary, create alternate file to load for windows
        {
            for (KeystoreEntry node : keystoreEntries)
            {
                node.expand(context);

                context.setKeystore(node.toKeyStore());
            }

            GrantEntry grant = grantEntries.get( 0 );
            grant.expand( context );

            Permission perm = grant.getPermissions().elements().nextElement();

            assertEquals( __PRINCIPAL, perm.getName() );
        }
    }

    @Test
    public void testFileSeparatorExpansion() throws Exception
    {
        PolicyContext context = new PolicyContext();
        context.addProperty( "foo", "bar" );

        assertEquals(File.separator, context.evaluate( "${/}" ) );

        assertEquals(File.separator + "bar" + File.separator, context.evaluate( "${/}${foo}${/}" ) );

        assertEquals(File.separator + File.separator, context.evaluate( "${/}${/}" ) );
    }
}