Application: gvSIG desktop » Library: org.gvsig.tools

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.tools.persistence;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Set;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.exception.PersistenceClassNotRegistered;

import org.gvsig.tools.persistence.exception.PersistenceException;

import org.gvsig.tools.persistence.exception.PersistenceTypeNotSupportedException;

import org.gvsig.tools.persistence.exception.PersistenceValidateExceptions;

public class BasicSetTest         extends AbstractBasicTest {

        public static  class MyObject2 implements Persistent {

                String name;

                public MyObject2() {

                        this.name = "(unknow)";

                }

                MyObject2(String name){

                        this.name = name;

                }

                public String getName() {

                        return this.name;

                }

                public void loadFromState(PersistentState state)

                                throws PersistenceException {

                        this.name = state.getString("name");

                }

                public void saveToState(PersistentState state)

                                throws PersistenceException {

                        state.set("name",this.name);

                }

        }

        public static  class MyObject implements Persistent {

                Set values;

                Set persistentValues;

                public MyObject() {

                        this.values = new HashSet();

                        this.persistentValues = new HashSet();

                        for( int i=0; i<10; i++ ) {

                                this.values.add( "valor-" + i);

                        }

                        for( int i=0; i<10; i++ ) {

                                this.persistentValues.add( new MyObject2("valor-" + i));

                        }

                }

                void check() {

                        assertNotNull(this.values);

                        assertNotNull(this.persistentValues);

                        assertEquals("size of list of strings mistmach",10, this.values.size());

                        assertEquals("size of list of complex object mistmach",10, this.persistentValues.size());

                        Set values = new HashSet();

                        for( int i=0; i<10; i++ ) {

                                String value = "valor-"+i;

                                values.add(value);

                                assertTrue("value "+ i + " mistmach in list of strings.", this.values.contains(value));

                        }

                        Iterator it = this.persistentValues.iterator();

                        while( it.hasNext() ) {

                                MyObject2 obj = (MyObject2) it.next();

                                if( ! values.contains(obj.getName()) ) {

                                        fail("value '"+ obj.getName() + "' mistmach in list of complex objects.");

                                }

                                values.remove(obj.getName());

                        }

                        if( values.size() != 0 ) {

                                fail("Unspected values in list of complex objects ");

                        }

                }

                public void loadFromState(PersistentState state)

                                throws PersistenceException {

                        this.values = new HashSet(state.getSet("values"));

                        this.persistentValues = new HashSet(state.getSet("persistentValues"));

                }

                public void saveToState(PersistentState state)

                                throws PersistenceException {

                        state.set("values",this.values);

                        state.set("persistentValues",this.persistentValues);

                }

        }

        protected void register() {

                PersistenceManager manager = ToolsLocator.getPersistenceManager();

                DynStruct definition2 = manager.addDefinition(

                                MyObject2.class,

                                "MyObject2BasicSet",

                                "MyObject2BasicSet Persistence definition",

                                null,

                                null

                );

                definition2.addDynFieldString("name").setMandatory(true);

                DynStruct definition = manager.addDefinition(

                                MyObject.class,

                                "MyObjectBasicSet",

                                "MyObjectBasicSet Persistence definition",

                                null,

                                null

                );

                definition.addDynFieldSet("values").setMandatory(true).setElementsType(DataTypes.STRING);

                definition.addDynFieldSet("persistentValues").setMandatory(true).setElementsType(definition2);

        }

        protected void unregister() throws PersistenceClassNotRegistered {

                manager.unregisterClass("MyObjectBasicSet");

                manager.unregisterClass("MyObject2BasicSet");

        }

        public void doSetUp() throws Exception {

                super.doSetUp();

                register();

        }

        protected void tearDown() throws Exception {

                super.tearDown();

                unregister();

        }

        //

        // =================================================

        //

        protected String getTestId() {

                return "basiclist";

        }

        public void testRegister() {

                DynStruct definition = manager.getDefinition(MyObject.class);

                assertNotNull("Can't register class", definition);

                assertEquals(

                                "Registration don't work, name incorrect",

                                "MyObjectBasicSet",

                                definition.getName()

                );

        }

        public void testGetState() throws PersistenceTypeNotSupportedException, PersistenceClassNotRegistered, PersistenceException, PersistenceValidateExceptions {

                try {

                        MyObject obj = new MyObject();

                        PersistentState state = manager.getState(obj);

                        assertNotNull("Can't retrieve state", state);

                        DynStruct definition = state.getDefinition();

                        assertNotNull("Can't register class", definition);

                        assertEquals(

                                        "Registration don't work, name incorrect",

                                        "MyObjectBasicSet",

                                        definition.getName()

                        );

                }catch(PersistenceValidateExceptions ex) {

                        fail(ex.getMessage());

                }

         }

        public void testSetState() throws PersistenceTypeNotSupportedException, PersistenceClassNotRegistered, PersistenceException, PersistenceValidateExceptions {

                MyObject obj1 = new MyObject();

                PersistentState state = manager.getState(obj1);

                assertNotNull("Can't retrieve state", state);

                MyObject obj2 = (MyObject) manager.create(state);

                obj2.check();

                obj1.check();

        }

        public void testListSaveState() throws Exception {

                MyObject obj = new MyObject();

                saveState(obj);

        }

        public void testListLoadState() throws Exception {

                MyObject obj1 = new MyObject();

                MyObject obj2 = (MyObject) manager.create(this.saveAndRestoreState(obj1));

                obj2.check();

                obj1.check();

        }

}