source: fedd/abac-src/rbtm/test/Test.java @ 4adc697

package edu.stanford.peer.rbtm;

import java.io.*;
import java.util.*;
import edu.stanford.peer.rbtm.engine.*;
import edu.stanford.peer.rbtm.credential.*;
import edu.stanford.peer.rbtm.util.*;

public class Test 
{
    public static void main(String args[]) throws Exception {
                //test1();
                test3();
        //test3(Integer.parseInt(args[0]), Integer.parseInt(args[1]));
    }

    static void test1() throws Exception {
        CredentialManager cs = new CredentialManager();
        cs.addCredential(new StaticCredential("university", "professors", "university.depts.professors"));
        cs.addCredential(new StaticCredential("dept_cs", "professors", "mitchell"));
        cs.addCredential(new StaticCredential("university", "depts", "dept_cs"));
        System.out.println(cs.toString());
        System.out.println();
        cs.removeCredential(new StaticCredential("university", "depts", "dept_cs"));
        System.out.println(cs.toString());
        System.out.println();

        Role r = new Role("university", "professors");
        System.out.println(cs.backwardSearch(r));
        System.out.println(cs.forwardSearch(new SimpleEntity("mitchell")));

    }

    static void test2() throws Exception {
            CredentialManager cs = new CredentialManager();
            SimpleEntity        tony = new SimpleEntity("Tony"),
                                        carmela = new SimpleEntity("Carmela"),
                                        melfi = new SimpleEntity("Jennifer Melfi"),
                                        livia = new SimpleEntity("Livia"),
                                        junior = new SimpleEntity("Junior"),
                                        chris = new SimpleEntity("Chris Moltisanti"),
                                        meadow = new SimpleEntity("Meadow"),
                                        anthony = new SimpleEntity("Anthony Jr."),
                                        paulie = new SimpleEntity("Paulie Walnuts"),
                                        jackie = new SimpleEntity("Jackie Aprile");

            Role        tonyWifeRole = new Role(tony, "Wife"),
                        tonyFamilyRole = new Role(tony, "Family Members"),
                        tonyChildrenRole = new Role(tony, "Children"),
                        tonyCrewRole = new Role(tony, "Crew"),
                        tonyMotherRole = new Role(tony, "Mother"),
                        tonyBossRole = new Role(tony, "Boss"),
                        tonyUncleRole = new Role(tony, "Uncle"),
                        melfiPatientRole = new Role(melfi, "Patients"),
                        melfiPreferredPatientRole = new Role(melfi, "Preferred Patients"),
                        chrisBossRole = new Role(chris, "Boss"),
                        chrisActingBossRole = new Role(chris, "ActingBoss"),
                        chrisGangRole = new Role(chris, "Gang");
            
            ArrayList tonyCreds = new ArrayList(20);
            
            tonyCreds.add(new StaticCredential(tonyWifeRole, carmela));
            tonyCreds.add(new StaticCredential(tonyFamilyRole, tonyWifeRole));
            tonyCreds.add(new StaticCredential(tonyFamilyRole, tonyChildrenRole));
            tonyCreds.add(new StaticCredential(tonyChildrenRole, meadow));
            tonyCreds.add(new StaticCredential(tonyChildrenRole, anthony));
            tonyCreds.add(new StaticCredential(tonyCrewRole, chris));
            tonyCreds.add(new StaticCredential(tonyCrewRole, paulie));
            tonyCreds.add(new StaticCredential(tonyMotherRole, livia));
            tonyCreds.add(new StaticCredential(tonyBossRole, jackie));
            tonyCreds.add(new StaticCredential(tonyUncleRole, junior));
            // Setting up credentials for toni
            cs.addCredentials(tonyCreds); 

            ArrayList melfiCreds = new ArrayList(5);
            melfiCreds.add(new StaticCredential(melfiPatientRole, melfiPreferredPatientRole));
            melfiCreds.add(new StaticCredential(melfiPreferredPatientRole, tony));
            cs.addCredentials(melfiCreds); 

            
            ArrayList chrisCreds = new ArrayList(5);
            cs.addCredential(new StaticCredential(chrisBossRole, tony));
            cs.addCredential(new StaticCredential(chrisActingBossRole, tonyBossRole));
            cs.addCredential(new StaticCredential(chrisGangRole, new LinkedRole(chrisBossRole, "Crew")));

                System.out.println(cs.toString());
                System.out.println();
                System.out.println(cs.forwardSearch(carmela));
                System.out.println();

                cs.removeCredential(new StaticCredential(chrisBossRole, tony));
                cs.removeCredential(new StaticCredential(chrisActingBossRole, tonyBossRole));

                cs.addCredential(new StaticCredential(chrisGangRole, new Role("Tony", "Wife")));
                cs.addCredential(new StaticCredential(new Role("Meadow", "Mother"), new Role("Tony", "Wife")));

                System.out.println(cs.toString());
                System.out.println();
                System.out.println(cs.forwardSearch(carmela));
                System.out.println();

                //System.out.println(cs.forwardSearch(jackie));
                //System.out.println();
        }
        
        
        static void test3() throws Exception
        {
                CredentialManager store = new CredentialManager();

                store.addCredential(new StaticCredential("sbose", "picturesRead", "sbose.family"));
                store.addCredential(new StaticCredential("sbose", "picturesWrite", "sbose.family"));
                store.addCredential(new StaticCredential("sbose", "picturesRead", "sbose.family.friends"));
                store.addCredential(new StaticCredential("sbose", "picturesRead", "patrick"));
                store.addCredential(new StaticCredential("sbose", "picturesWrite", "andrew"));
                store.addCredential(new StaticCredential("sbose", "picturesWrite", "patrick"));

                System.out.println(store.toString() + "\n\n");

                ResultEvidenceMap result = store.forwardSearch(new SimpleEntity("andrew"));
                Set resultSet = result.resultSet();
                System.out.println("forward andrew is: " + resultSet);
                Iterator it = resultSet.iterator();
                while (it.hasNext()) {
                    System.out.println(it.next().toString());
                }

                //add a credential
                store.addCredential(new StaticCredential("sbose", "picturesRead", "andrew"));

                System.out.println("\n\n added credential sbose.picturesRead <- andrew");

                System.out.println("\n" + store.toString() + "\n\n" + "forward search on entity andrew:");

                result = store.forwardSearch(new SimpleEntity("andrew"));
                resultSet = result.resultSet();
                System.out.println("forward andrew is: " + resultSet);
                it = resultSet.iterator();
                while (it.hasNext()) {
                    System.out.println(it.next().toString());
                }


                //remove a credential
                //
                //

                store.removeCredential(new StaticCredential("sbose", "picturesWrite", "patrick"));

                System.out.println("\n\n removed credential sbose.picturesWrite <- patrick");

                System.out.println("\n" + store.toString() + "\n\n" + "forward search on patrick:");

                result = store.forwardSearch(new SimpleEntity("patrick"));
                resultSet = result.resultSet();
                it = resultSet.iterator();
                while (it.hasNext()) {
                    System.out.println(it.next().toString());
                }
        }



    static void test4(int l, int n) throws Exception {
        CredentialManager cs = new CredentialManager();
        for (int j=0; j<n; j++) {
            cs.addCredential(new StaticCredential("K", "C", "K.C"+l+".B"+j));
        }
            cs.addCredential(new StaticCredential("K", "C1", "K0.A"));
        for (int i=1; i<l; i++) {
            int j = i+1;
            cs.addCredential(new StaticCredential("K", "C"+j, "K.C"+i+".A"));
        }
        for (int i=0; i<n; i++) {
            cs.addCredential(new StaticCredential("K0", "A", "K"+i));
        }
        for (int i=0; i<n; i++) {
            int j = (i+1) % n;
            cs.addCredential(new StaticCredential("K"+i, "A", "K"+j+".A"));
        }
        System.out.println(cs.toString());
        cs.forwardSearch(new SimpleEntity("K0"));

        cs.backwardSearch(new Role("K", "C"));

        //engine.bidirectionSearch(new Role("K0", "B0"), new Role("K", "C"));
    }

}