ModelPrintoutFakeTest.java

/* (c) 2014, 2015 Open Source Geospatial Foundation - all rights reserved
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */

package org.geoserver.geofence.services.rest.model;

import org.geoserver.geofence.core.model.LayerAttribute;
import org.geoserver.geofence.core.model.enums.AccessType;
import org.geoserver.geofence.core.model.enums.GrantType;
import org.geoserver.geofence.core.model.enums.LayerType;
import org.geoserver.geofence.services.dto.ShortGroup;
import org.geoserver.geofence.services.dto.ShortInstance;
import org.geoserver.geofence.services.rest.model.config.RESTFullUserGroupList;
import org.geoserver.geofence.services.rest.model.util.IdName;

import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import javax.xml.bind.JAXB;
import org.geoserver.geofence.services.rest.model.RESTRulePosition.RulePosition;

import org.junit.Test;

/**
 *
 * @author ETj (etj at geo-solutions.it)
 */
public class ModelPrintoutFakeTest {

    public ModelPrintoutFakeTest() {
        System.out.println("RESTShortUser sample");
        RESTShortUser user = createShortUser("01");
        System.out.println(marshal(user));

        System.out.println("RESTShortUserList sample");
        RESTShortUserList userList = new RESTShortUserList();
        userList.add(createShortUser("01"));
        userList.add(createShortUser("02"));
        System.out.println(marshal(userList));

        System.out.println("RESTInputUser sample");
        RESTInputUser inputUser = createInputUser("02");
        System.out.println(marshal(inputUser));

        System.out.println("RESTInputRule sample");
        RESTInputRule inputRule = createInputRule("02");
        System.out.println(marshal(inputRule));

        {
            System.out.println("RESTRuleList sample");
            RESTOutputRuleList ruleList = new RESTOutputRuleList();

            RESTOutputRule r1 = createOutputRule("01");
            r1.setConstraints(null);
            ruleList.add(r1);

            r1 = createOutputRule("02");
            r1.setGrant(GrantType.DENY);
            r1.setConstraints(null);
            ruleList.add(r1);

            System.out.println(marshal(ruleList));
        }

        {
            System.out.println("RESTInputGroup sample");
            RESTInputGroup inputGroup = new RESTInputGroup();
            inputGroup.setEnabled(Boolean.TRUE);
            inputGroup.setName("sample group");
            inputGroup.setExtId("external_id_here");
            System.out.println(marshal(inputGroup));
        }
        {
            System.out.println("RESTInputGroup sample (field enable not set)");
            RESTInputGroup inputGroup = new RESTInputGroup();
            inputGroup.setName("sample group");
            System.out.println(marshal(inputGroup));
        }
        {
            System.out.println("RESTFullUserGroupList sample");
            RESTFullUserGroupList list = new RESTFullUserGroupList();
            list.add(createShortGroup("group1"));
            list.add(createShortGroup("group2"));
            System.out.println(marshal(list));
        }
        {
            System.out.println("RESTInputInstance sample");
            RESTInputInstance i = new RESTInputInstance();
            i.setName("sample instance");
            i.setDescription("sample descr");
            i.setBaseURL("http://yourgeoserver/geoserver");
            i.setUsername("admin");
            i.setPassword("clearpw");
            System.out.println(marshal(i));
        }

        {
            System.out.println("RESTShortInstanceList sample");
            RESTShortInstanceList list = new RESTShortInstanceList();
            {
                ShortInstance i1 = new ShortInstance();
                i1.setName("instance_01");
                i1.setId(100);
                i1.setUrl("http://test/geoserver");
                list.add(i1);
            }
            {
                ShortInstance i1 = new ShortInstance();
                i1.setName("instance_02");
                i1.setId(101);
                i1.setUrl("http://othertest/geoserver");
                list.add(i1);
            }
            System.out.println(marshal(list));
        }
    }

    protected ShortGroup createShortGroup(String base) {
        ShortGroup g1 = new ShortGroup();
        g1.setName(base);
        g1.setId(base.hashCode());
        g1.setExtId("ext_"+base);
        g1.setEnabled(Boolean.TRUE);
        return g1;
    }

    private String marshal(Object o) {

        StringWriter w = new StringWriter();
        JAXB.marshal(o, w);
        w.flush();
        return w.getBuffer().toString();
    }

    @Test
    public void testGetId() {
    }

    private RESTShortUser createShortUser(String base) {
        RESTShortUser ret = new RESTShortUser();
        ret.setId(base.hashCode() % 1000l);
        ret.setExtId("ext_"+base);
        ret.setUserName("user_"+base);
        ret.setEnabled(true);
        return ret;
    }

    private RESTInputUser createInputUser(String base) {
        RESTInputUser ret = new RESTInputUser();       
        ret.setExtId("ext_"+base);
        ret.setName("name_"+base);
        ret.setFullName("fullname_"+base);
        ret.setPassword("pw_"+base);
        ret.setEmailAddress("email_"+base);
        ret.setEnabled(true);
        ret.setAdmin(false);

        List<IdName> groups = new ArrayList<IdName>();
        groups.add(new IdName((long)base.hashCode()));
        groups.add(new IdName("grp_" + base));

        ret.setGroups(groups);

        return ret;
    }

    private RESTInputRule createInputRule(String base) {
        RESTInputRule ret = new RESTInputRule();

        ret.setUsername("user_"+base);
        ret.setRolename("role_"+base);
        ret.setInstanceId((long)base.hashCode());
        ret.setService("WMS_"+base);
        ret.setRequest("getMap_"+base);
        ret.setWorkspace("wsp_"+base);
        ret.setLayer("layer_"+base);
        ret.setPosition(new RESTRulePosition(RulePosition.offsetFromBottom, 1));
        ret.setGrant(GrantType.ALLOW);

        RESTLayerConstraints constraints = new RESTLayerConstraints();
        constraints.setType(LayerType.VECTOR);
        constraints.setAllowedStyles(new HashSet(Arrays.asList("teststyle1","teststyle2","Style_"+base)));
        constraints.setCqlFilterRead("CQL_READ");
        constraints.setCqlFilterWrite("CQL_WRITE");
        constraints.setDefaultStyle("Style_"+base);
        constraints.setRestrictedAreaWkt("wkt_"+base);

        Set<LayerAttribute> attrs = new HashSet<LayerAttribute>();
        attrs.add(new LayerAttribute("attr1", "java.lang.String", AccessType.NONE));
        attrs.add(new LayerAttribute("attr2", "java.lang.String", AccessType.READONLY));
        attrs.add(new LayerAttribute("attr3", "java.lang.String", AccessType.READWRITE));
        constraints.setAttributes(attrs);
        
        ret.setConstraints(constraints);

        return ret;
    }

    private static long rulePri = 100;

    private RESTOutputRule createOutputRule(String base) {
        RESTOutputRule ret = new RESTOutputRule();

        ret.setPriority(rulePri++);
        ret.setUsername("user_"+base);
        ret.setRolename("role_"+base);
        ret.setInstance(new IdName((long)(Math.random()*10000), "gs_"+base));
        ret.setService("WMS_"+base);
        ret.setRequest("getMap_"+base);
        ret.setWorkspace("wsp_"+base);
        ret.setLayer("layer_"+base);
        
        ret.setGrant(GrantType.ALLOW);

//        RESTLayerConstraints constraints = new RESTLayerConstraints();
//        constraints.setType(LayerType.VECTOR);
//        constraints.setAllowedStyles(new HashSet(Arrays.asList("teststyle1","teststyle2","Style_"+base)));
//        constraints.setCqlFilterRead("CQL_READ");
//        constraints.setCqlFilterWrite("CQL_WRITE");
//        constraints.setDefaultStyle("Style_"+base);
//        constraints.setRestrictedAreaWkt("wkt_"+base);
//
//        Set<LayerAttribute> attrs = new HashSet<LayerAttribute>();
//        attrs.add(new LayerAttribute("attr1", "java.lang.String", AccessType.NONE));
//        attrs.add(new LayerAttribute("attr2", "java.lang.String", AccessType.READONLY));
//        attrs.add(new LayerAttribute("attr3", "java.lang.String", AccessType.READWRITE));
//        constraints.setAttributes(attrs);
//
//        ret.setConstraints(constraints);

        return ret;
    }

}