Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting;

import java.io.File;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Set;

import junit.framework.Assert;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

public abstract class TestManager 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 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 testUserFolderPath(){

                setWorkingFolder();

                Assert.assertEquals(manager.getRootUserFolder(), manager.getUserFolder().getPath());

        }

        public void testExistsScriptId(){

                setWorkingFolder();

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

                Assert.assertEquals(false, manager.validateUnitId(manager.getUserFolder(),"test_script"));

                Assert.assertEquals(true, manager.validateUnitId(manager.getUserFolder(),"test_script2"));

                deleteWorkingFolderContent();

        }

        public void testExtensionsAndLanguages(){

        Map<String, String> langByExt =

            manager.getSupportedLanguagesByExtension();

        Map<String, String> extByLang = manager.getExtensionsByLanguages();

        Set<String> keyExt = langByExt.keySet();

        Iterator<String> it = keyExt.iterator();

                while(it.hasNext()){

                        Object ext = it.next();

                        Object lang = langByExt.get(ext);

                        Assert.assertEquals(extByLang.get(lang), ext);

                }

        }

        public void checkUnitValues(ScriptingUnit unit1, ScriptingUnit unit2){

                Assert.assertEquals(unit1.getId(),unit2.getId());

                Assert.assertEquals(unit1.getName(),unit2.getName());

        }

        public void testRegisteringSystemFolders(){

                ScriptingFolder folder_parent = (ScriptingFolder) manager.createUnit(ScriptingManager.UNIT_FOLDER,manager.getUserFolder(), "test");

                ScriptingFolder folder = (ScriptingFolder) manager.createUnit(ScriptingManager.UNIT_FOLDER,folder_parent, "test");

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

                manager.registerSystemFolder("test", new File(folder.getPath()));

                List<ScriptingUnit> units = manager.getSystemFolder().getUnits();

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

                ScriptingUnit unit = null;

                while(it.hasNext()){

                        unit = it.next();

                        checkUnitValues(unit, folder);

                }

                Assert.assertNotNull(unit);

                ScriptingScript script2 = (ScriptingScript) manager.getScript(new File(((ScriptingFolder)unit).getPath()+File.separator+script.getId()+".py"));

                checkUnitValues(script,script2);

                deleteWorkingFolderContent();

        }

}