smartrics.rest.test.fitnesse.fixture.Resources Maven / Gradle / Ivy

Go to download
/*  Copyright 2015 Fabrizio Cannizzo
 *
 *  This file is part of RestFixture.
 *
 *  RestFixture (http://code.google.com/p/rest-fixture/) is free software:
 *  you can redistribute it and/or modify it under the terms of the
 *  GNU Lesser General Public License as published by the Free Software Foundation,
 *  either version 3 of the License, or (at your option) any later version.
 *
 *  RestFixture is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with RestFixture.  If not, see .
 *
 *  If you want to contact the author please leave a comment here
 *  http://smartrics.blogspot.com/2008/08/get-fitnesse-with-some-rest.html
 */
package smartrics.rest.test.fitnesse.fixture;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Vector;

/**
 * The database of resources to support RestFixture CATs.
 * 
 * @author smartrics
 * 
 */
public class Resources {
    private final Map> resourceDb = Collections.synchronizedMap(new HashMap>());
    private static Resources instance = new Resources();
    private int counter = 0;

    public static Resources getInstance() {
        return instance;
    }

    public void clear() {
        for (String c : resourceDb.keySet()) {
            resourceDb.get(c).clear();
        }
        resourceDb.clear();
    }

    public Collection asCollection(String context) {
        Collection c = new Vector();
        Map m = resourceDb.get(context);
        if (m != null) {
            for (Entry e : m.entrySet()) {
                String s = e.getKey();
                c.add(m.get(s));
            }
        }
        return c;
    }

    public List contexts() {
        List ctxKeys = new ArrayList();
        ctxKeys.addAll(resourceDb.keySet());
        return ctxKeys;
    }

    public void add(String context, Resource r) {
        if (r.getId() == null) {
            r.setId(Integer.toString(newCounter()));
        }
        Map m = getMapForContext(context);
        m.put(r.getId(), r);
    }

    private Map getMapForContext(String context) {
        Map m = resourceDb.get(context);
        if (m == null) {
            m = new HashMap();
            resourceDb.put(context, m);
        }
        return m;
    }

    public Resource get(String context, String i) {
        return getMapForContext(context).get(i);
    }

    public int size(String context) {
        return getMapForContext(context).size();
    }

    public Resource remove(String context, String index) {
        return getMapForContext(context).remove(index);
    }

    public void remove(String context, Resource o) {
        remove(context, o.getId());
    }

    public void reset() {
        clear();
        counter = 0;
        add("/resources", new Resource("\n    a funky name\n    an important message"
                + "\n    \n        3\n    " + "\n"));
        add("/resources", new Resource("{ \"resource\" : { \"name\" : \"a funky name\", " + "\"data\" : \"an important message\" } }"));
        StringBuffer sb = new StringBuffer();
        sb.append("\n");
        sb.append("   giant bob\n");
        sb.append("   large content\n");
        sb.append("   \n");
        sb.append("       \n");
        sb.append("            Regent Street\n");
        sb.append("       \n");
        sb.append("       \n");
        sb.append("            12345\n");
        sb.append("       \n");
        sb.append("   \n");
        sb.append("   \n");
        sb.append("       \n");
        sb.append("           \n");
        sb.append("           \n");
        sb.append("Inferno (Italian for 'Hell') is the first part of Dante Alighieri's 14th-century epic poem Divine Comedy. \n");
        sb.append("It is followed by Purgatorio and Paradiso. It is an allegory telling of the journey of Dante through what is \n");
        sb.append("largely the medieval concept of Hell, guided by the Roman poet Virgil. In the poem, Hell is depicted as nine \n");
        sb.append("circles of suffering located within the Earth. Allegorically, the Divine Comedy represents the journey of the soul");
        sb.append("towards God, with the Inferno describing the recognition and rejection of sin.\n");
        sb.append("           \n");
        sb.append("       \n");
        sb.append("   \n");
        sb.append("\n");

        add("/resources", new Resource("100", sb.toString()));
    }

    private synchronized int newCounter() {
        return counter++;
    }

    @Override
    public String toString() {
        StringBuffer b = new StringBuffer();
        String nl = System.getProperty("line.separator");
        b.append("Resources:[").append(nl);
        for (String c : resourceDb.keySet()) {
            b.append(" Context(").append(c).append("):[").append(nl);
            for (Resource r : asCollection(c)) {
                b.append(r).append(nl);
            }
            b.append(" ]").append(nl);
        }
        b.append("]").append(nl);
        return b.toString();
    }

}