source: java/net/deterlab/abac/regression/RocketsTest.java

package net.deterlab.abac.regression;

import java.io.*;
import java.util.*;

import net.deterlab.abac.*;

public class RocketsTest extends RegressionTest {
    /**
     * Put out an identity and key into dir named by the name of the Identity
     * @param i the Identity to write
     * @param dir a File holding the destination directory
     * @throws IOException if the file writing fails
     */
    public void writeCombinedIdentity(Identity i, File dir) 
            throws IOException {
        FileOutputStream f = new FileOutputStream(
                new File(dir,  i.getName() + ".pem"));

        i.write(f);
        i.writePrivateKey(f);
    }
    /**
     * Create a new Credential/Identity writing test for the given class of
     * credentials.
     * @param cn a String containing the binary name of the class to test
     */
     public RocketsTest(String name) { 
         super(name);
     }

    /**
     * Carry out the test.  Create credentials for the create_experiment
     * example, run the proof and make sure the correct proof is generated.
     * @param data a File pointing to a directory that contains files the test
     * may need
     * @param scratch a File pointing to a directory that the test can use to
     * store data
     * @return a boolean, true if the test is passed
     */
    public boolean runTest(File data, File scratch) {
        try {
            Context ctxt = new Context();
            Identity acme = new Identity("Acme");
            Identity warnerBros = new Identity("WarnerBros");
            Identity batman = new Identity("Batman");
            Identity coyote = new Identity("Coyote");
            ArrayList<Credential> inProof = new ArrayList<Credential>();

            ctxt.load_id_chunk(acme);
            ctxt.load_id_chunk(warnerBros);
            ctxt.load_id_chunk(batman);
            ctxt.load_id_chunk(coyote);

            Credential c = ctxt.newCredential(
                    new Role(acme.getKeyID() + ".buy_rockets"),
                    new Role(acme.getKeyID() + ".preferred_customer & " +
                        warnerBros.getKeyID() + ".character"));
            c.make_cert(acme);
            ctxt.load_attribute_chunk(c);
            inProof.add(c);
            c = ctxt.newCredential(
                    new Role(acme.getKeyID() + ".preferred_customer"),
                    new Role(coyote.getKeyID()));
            c.make_cert(acme);
            ctxt.load_attribute_chunk(c);
            inProof.add(c);
            c = ctxt.newCredential(
                    new Role(acme.getKeyID() + ".preferred_customer"),
                    new Role(batman.getKeyID()));
            c.make_cert(acme);
            ctxt.load_attribute_chunk(c);
            c = ctxt.newCredential(
                    new Role(warnerBros.getKeyID() + ".character"),
                    new Role(coyote.getKeyID()));
            c.make_cert(warnerBros);
            ctxt.load_attribute_chunk(c);
            inProof.add(c);

            /* For checking only.  An internal credential of this form will
             * come out of the Context in the proof. */
            c = new InternalCredential(
                    new Role(acme.getKeyID() + ".preferred_customer & " +
                        warnerBros.getKeyID() + ".character"), 
                    new Role(coyote.getKeyID()));
            inProof.add(c);

            Context.QueryResult q = ctxt.query(
                    acme.getKeyID() + ".buy_rockets",
                    coyote.getKeyID());

            if ( !q.getSuccess()) {
                setReason("Coyote can buy rockets proof failed?");
                return false;
            }
            if ( inProof.size() != q.getCredentials().size()) {
                setReason("Coyote can buy rockets proof is the wrong size");
                return false;
            }

            for (Credential cc: inProof)
                if ( !q.getCredentials().contains(cc)) {
                    setReason("Coyote can buy rockets proof missing cred "+cc);
                    return false;
                }

            q = ctxt.query(acme.getKeyID() + ".buy_rockets",
                    batman.getKeyID());
            if ( q.getSuccess()) {
                setReason("Batman can buy rockets ?");
                return false;
            }
        }
        catch (Exception e) {
            setReason(e.getMessage());
            return false;
        }
        return true;
    }
}