Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting;

import java.io.File;

import java.util.Iterator;

import java.util.List;

import junit.framework.Assert;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

public abstract class TestFolder extends AbstractLibraryAutoInitTestCase {

        protected ScriptingManager manager;

        @Override

        protected void doSetUp() throws Exception {

                manager = ScriptingLocator.getManager();

        }

        public void setWorkingFolder(){

                File file =new File(this.getClass().getClassLoader().getResource("org/gvsig/scripting/home").getFile());

                if(file.exists()){

                        file.mkdir();

                }

                manager.setRootUserFolder(file.getAbsolutePath());

        }

        public void deleteFolderContent(ScriptingFolder folder){

                List<ScriptingUnit> units = folder.getUnits();

        Iterator<ScriptingUnit> it = units.iterator();

                while(it.hasNext()){

                        ScriptingUnit unit = it.next();

                        if(unit instanceof ScriptingFolder){

                                deleteFolderContent((ScriptingFolder)unit);

                                folder.remove(unit);

                        }else{

                                folder.remove(unit);

                        }

                }

        }

        public void deleteWorkingFolderContent(){

                setWorkingFolder();

                ScriptingFolder folder = manager.getUserFolder();

                deleteFolderContent(folder);

        }

        public File getFile(ScriptingFolder folder, String name){

                return new File(folder.getPath()+File.separator+File.separator+name);

        }

        public void checkFolder(ScriptingFolder folder, String fName){

                File ffolder = getFile(folder, fName);

                Assert.assertTrue(ffolder.exists());

                Assert.assertTrue(ffolder.isDirectory());

        }

        public void checkScriptDefaultValues(ScriptingScript script){

                Assert.assertEquals("test_script", script.getId());

                Assert.assertEquals("test_script", script.getName());

                Assert.assertEquals("python", script.getLangName());

                Assert.assertEquals("main", script.getMainName());

        }

        public void testCreation(){

                deleteWorkingFolderContent();

                ScriptingFolder folder = (ScriptingFolder) manager.createUnit(ScriptingManager.UNIT_FOLDER,manager.getUserFolder(),"test_folder");

                checkFolder(manager.getUserFolder(),folder.getId());

        }

        public void testList(){

                testCreation();

                ScriptingFolder folder = manager.getFolder(getFile(manager.getUserFolder(),"test_folder"));

                ScriptingScript script = (ScriptingScript) manager.createUnit(ScriptingManager.UNIT_SCRIPT,folder, "test_script.py");

        List<ScriptingUnit> units = folder.getUnits();

        Iterator<ScriptingUnit> it = units.iterator();

                while(it.hasNext()){

                        ScriptingUnit unit = it.next();

                        checkScriptDefaultValues((ScriptingScript) unit);

                }

                folder.remove(script);

                Assert.assertTrue(folder.getUnits().isEmpty());

        }

        public void testRename(){

                testCreation();

                ScriptingFolder folder = manager.getFolder(getFile(manager.getUserFolder(),"test_folder"));

                Assert.assertFalse(folder.rename("test_folder"));

                Assert.assertTrue(folder.rename("test_folder2"));

                Assert.assertEquals("test_folder2", folder.getName());

                deleteWorkingFolderContent();

        }

        public void testMove(){

                testCreation();

                ScriptingScript script = (ScriptingScript) manager.createUnit(ScriptingManager.UNIT_SCRIPT,manager.getUserFolder(), "test_script.py");        

                ScriptingFolder folder = manager.getFolder(getFile(manager.getUserFolder(),"test_folder"));

                Assert.assertEquals(script.getParent().getPath(),manager.getUserFolder().getPath());

                folder.add(script);

                Assert.assertEquals(script.getParent(),folder);

                deleteWorkingFolderContent();

        }

}