source: src/test/java/agents/nastyagent/CheckStoredData.java@ 126

package agents.nastyagent;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import genius.core.AgentID;
import genius.core.Bid;
import genius.core.actions.Action;
import genius.core.actions.ActionWithBid;
import genius.core.list.Tuple;
import genius.core.parties.NegotiationInfo;
import genius.core.persistent.StandardInfo;
import genius.core.persistent.StandardInfoList;

/**
 * Checks if data is retained in persistent storage properly. This uses a hack.
 *
 * 
 */
public class CheckStoredData extends NastyAgent {
        private static final String DATA = "data";
        static boolean runBefore = false;
        static List<ActionWithBid> previousActions = new ArrayList<ActionWithBid>();

        public void init(NegotiationInfo info) {
                super.init(info);
                if (runBefore) {
                        checkStore((StandardInfoList) data.get());
                }
        }

        private void checkStore(StandardInfoList standardInfoList) {
                if (standardInfoList.size() != 1) {
                        throw new IllegalStateException("Expected 1 info but found " + standardInfoList.size());
                }
                StandardInfo info = standardInfoList.get(0);
                if (info.getStartingAgent() == previousActions.get(0).getAgent().toString()) {
                        throw new IllegalStateException("the starting agent is not reported properly");
                }

                List<Tuple<String, Double>> utilities = info.getUtilities();
                if (utilities.size() != previousActions.size()) {
                        throw new IllegalStateException("The number of actions is not the expected " + previousActions.size());
                }

                for (int n = 0; n < previousActions.size(); n++) {
                        if (!utilities.get(n).get1().startsWith(previousActions.get(n).getAgent().toString())) {
                                throw new IllegalStateException("agent at utility " + n + " differs");
                        }
                        if (utilities.get(n).get2() != utilitySpace.getUtility(previousActions.get(n).getBid())) {
                                throw new IllegalStateException("utility " + n + " differs");
                        }
                }

        }

        @Override
        public Action chooseAction(List<Class<? extends Action>> possibleActions) {
                Action action = super.chooseAction(possibleActions);
                if (!runBefore && action instanceof ActionWithBid) {
                        previousActions.add((ActionWithBid) action);
                }
                return action;
        }

        @Override
        public void receiveMessage(AgentID sender, Action action) {
                if (!runBefore && action instanceof ActionWithBid) {
                        previousActions.add((ActionWithBid) action);
                }
        }

        @Override
        public HashMap<String, String> negotiationEnded(Bid acceptedBid) {
                // we actually got here. Report it.
                super.negotiationEnded(acceptedBid);
                runBefore = true;

                return null;
        }

}