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 TestScript extends AbstractLibraryAutoInitTestCase {

        protected ScriptingManager manager;

        @Override

        protected void doSetUp() throws Exception {

                ScriptingLocator locator = new ScriptingLocator();

                Assert.assertNotNull(locator);

                manager = ScriptingLocator.getManager();

                Assert.assertNotNull(manager);

        }

        public void setWorkingFolder(){

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

                if(file.exists()){

                        file.mkdir();

                }

//                FIXME: hay que sustituirlo por manager.setHomeFolder(home) 

//                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.getFile(),name);

        }

        public void checkFile(ScriptingFolder folder, String fName){

                File fscript = getFile(folder, fName);

                Assert.assertTrue(fscript.exists());

        }

        public void checkScriptDefaultValues(ScriptingScript script){

                Assert.assertEquals(manager.getUserFolder().getFile().getAbsolutePath(), script.getParent().getFile().getAbsolutePath());

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

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

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

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

        }

        public void checkScriptNewValues(ScriptingScript script){

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

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

                Assert.assertEquals("Tests sobre scripts", script.getDescription());

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

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

                Assert.assertEquals("creator", script.getCreatedBy());

                Assert.assertEquals("v1.0", script.getVersion());

                Assert.assertEquals("def main():\n  return 'hola mundo'\n",script.getCode());

        }

        public void testCreation(){

                setWorkingFolder();

                ScriptingScript previous = (ScriptingScript) manager.getScript(new File(manager.getRootUserFolder()+File.separator+"test_script.py"));

                if (previous!=null){

                        manager.getUserFolder().remove(previous);

                }

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

                script.save();

                checkFile(manager.getUserFolder(),"test_script.py");

                checkFile(manager.getUserFolder(),"test_script.inf");

                checkScriptDefaultValues(script);

        }

        public void testOpen(){

                testCreation();

                ScriptingScript script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));

                checkScriptDefaultValues(script);

                deleteWorkingFolderContent();

        }

        public void testSave(){

                testCreation();

                ScriptingScript script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));

                script.setName("TestScript");

                script.setDescription("Tests sobre scripts");

                script.setCode("def main():\n  return 'hola mundo'\n");

                script.setCreatedBy("creator");

                script.setVersion("v1.0");

                script.save();

                script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));

                checkScriptNewValues(script);

        }

        public void testRename(){

                testSave();

                ScriptingScript script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));

                Assert.assertFalse(script.rename("test_script"));

                Assert.assertTrue(script.rename("test_script2"));

                script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script2.py"));

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

                deleteWorkingFolderContent();

        }

        public void testMove(){

                setWorkingFolder();

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

                checkFile(manager.getUserFolder(),"other folder");

                ScriptingScript previous = (ScriptingScript) manager.getScript(new File(folderDestination.getFile(),"test_script.py"));

                if (previous!=null){

                        folderDestination.remove(previous);

                }

                testSave();

                ScriptingScript script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));        

                script.move(folderDestination);

                System.out.println(script.getParent() + " " + folderDestination);

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

                deleteWorkingFolderContent();

        }

        public void testRun() throws NoSuchMethodException{

                testSave();

                ScriptingScript script = (ScriptingScript) manager.getScript(getFile(manager.getUserFolder(),"test_script.py"));                

                Assert.assertEquals("hola mundo", script.run());

                script.setCode("def hola():\n  return 'hola mundo'\n");

                //setMainName

                script.setMainName("hola");

                Assert.assertEquals("hola mundo", script.run());

                //invokeMethod

                //invokeFunction

                Assert.assertEquals("hola mundo", script.invokeFunction("hola", null));

                deleteWorkingFolderContent();

        }

}