Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

import java.util.Iterator;

import java.util.List;

import junit.framework.Assert;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

public abstract class TestDialog 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());

                manager.setRootUserFolder(file.getAbsolutePath());

        }

        public File getFile(String name){

                return new File(manager.getRootUserFolder()+File.separator+name);

        }

        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 void checkFile(String fName){

                File fscript = getFile(fName);

                Assert.assertTrue(fscript.exists());

        }

        public void checkScriptDefaultValues(ScriptingDialog script){

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

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

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

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

        }

        public void checkScriptNewValues(ScriptingDialog script){

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

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

                Assert.assertEquals("Tests sobre dialogs", 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 hola():\n  return 'hola mundo'\n",script.getCode());

        }

        public void testCreation(){

                deleteWorkingFolderContent();

                setWorkingFolder();

                ScriptingDialog script = (ScriptingDialog) manager.createUnit(ScriptingManager.UNIT_DIALOG,manager.getUserFolder(),"test_dialog.py");

                script.save();

                checkFile("test_dialog.py");

                checkFile("test_dialog.dlg");

                checkFile("test_dialog.inf");

                checkScriptDefaultValues(script);

        }

        public void testOpen(){

                testCreation();

                ScriptingDialog script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));

                checkScriptDefaultValues(script);

                deleteWorkingFolderContent();

        }

        public void testSave(){

                testCreation();

                ScriptingDialog script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));

                script.setName("TestDialog");

                script.setDescription("Tests sobre dialogs");

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

                script.setCreatedBy("creator");

                script.setVersion("v1.0");

                script.save();

                File f = script.getDialogFile();

                try {

                        BufferedWriter bw = new BufferedWriter(new FileWriter(f));

                        bw.write("<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>\n" + 

                                        "<panel columns=\"1\">\n" +

                                        "\t<button action=\"hola\" text=\"press\"/>\n" + 

                                        "</panel>");

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));

                checkScriptNewValues(script);

        }

        public void testRename(){

                deleteWorkingFolderContent();

                testCreation();

                ScriptingDialog script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));

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

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

                script = (ScriptingDialog) manager.getScript(getFile("test_dialog2.py"));

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

                File f = getFile("test_dialog.py");

                Assert.assertFalse(f.exists());

                f = getFile("test_dialog2.py");

                Assert.assertTrue(f.exists());

                deleteWorkingFolderContent();

        }

        public void testMove(){

                testCreation();

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

                checkFile("other folder");

                ScriptingDialog script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));        

                script.move(folderDestination);

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

                deleteWorkingFolderContent();

        }

        public void testRun(){

        /*        testSave();

                ScriptingDialog script = (ScriptingDialog) manager.getScript(getFile("test_dialog.py"));                

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

                //setMainName

                //invokeMethod

                //invokeFunction

                deleteWorkingFolderContent();*/

        }

}