xml/tests/org.eclipse.wst.xml.tests.encoding/src/org/eclipse/wst/xml/tests/encoding/CreateCodedReaderTester.java - sourceediting/webtools.sourceediting - Git at Google

 /*******************************************************************************
  * Copyright (c) 2004, 2017 IBM Corporation and others.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License 2.0
  * which accompanies this distribution, and is available at
  * https://www.eclipse.org/legal/epl-2.0/
  *
  * SPDX-License-Identifier: EPL-2.0
  *
  * Contributors:
  *     IBM Corporation - initial API and implementation
  *******************************************************************************/
 package org.eclipse.wst.xml.tests.encoding;

 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URL;
 import java.nio.charset.IllegalCharsetNameException;
 import java.nio.charset.MalformedInputException;
 import java.util.Iterator;
 import java.util.List;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipOutputStream;

 import junit.framework.TestCase;

 import org.eclipse.core.resources.IProject;
 import org.eclipse.core.resources.IProjectDescription;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.resources.IWorkspace;
 import org.eclipse.core.resources.IWorkspaceRoot;
 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.core.runtime.IPath;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.NullProgressMonitor;
 import org.eclipse.core.runtime.Platform;
 import org.eclipse.wst.sse.core.internal.encoding.CodedReaderCreator;
 import org.eclipse.wst.sse.core.internal.encoding.NonContentBasedEncodingRules;
 import org.eclipse.wst.sse.core.internal.exceptions.UnsupportedCharsetExceptionWithDetail;
 import org.eclipse.wst.xml.tests.encoding.util.ProjectUnzipUtility;

 /**
  * This class is intended to be executed only at development time, and it
  * creates the class to JUnit test all the files found in the testfiles
  * directory.
  */
 public class CreateCodedReaderTester extends TestCase {
 	//	private final String fileDir = "html/";
 	//	private final String fileRoot =
 	// "/builds/Workspaces/HeadWorkspace/org.eclipse.wst.xml.tests.encoding/";
 	//	private final String fileLocation = fileRoot + fileDir;
 	private static final String TESTFILES_ZIPFILE_NAME = "testfiles.zip"; //$NON-NLS-1$
 	private static final boolean DEBUG = false;
 	private static final String TEST_FILE_DIR = "testfiles"; //$NON-NLS-1$
 	// needs to be static, since JUnit creates difference instances for each
 	// test
 	private static boolean fTestProjectInitialized;
 	private static final String TEST_PROJECT_NAME = "org.eclipse.wst.xml.tests.encoding"; //$NON-NLS-1$
 	static IProject fTestProject;
 	private static int nSetups = 0;
 	private static final String currentPlatformCharset = getPlatformDefault();
 	private boolean RECREATE_FILES = false;

 	/**
 	 *
 	 */
 	public CreateCodedReaderTester() {
 		super();
 		//System.out.println(currentPlatformCharset);
 	}

 	/**
 	 * @return
 	 */
 	private static String getPlatformDefault() {
 		String platformDefault = NonContentBasedEncodingRules.useDefaultNameRules(null);
 		return platformDefault;
 	}

 	public static void main(String[] args) {
 		//				try {
 		//					new CreateCodedReaderTester().doCreateAllFiles();
 		//				} catch (CoreException e) {
 		//					e.printStackTrace();
 		//				} catch (IOException e) {
 		//					e.printStackTrace();
 		//				}
 	}

 	private static void createProject(IProject project, IPath locationPath, IProgressMonitor monitor) throws CoreException {
 		if (monitor == null) {
 			monitor = new NullProgressMonitor();
 		}
 		monitor.beginTask("creating test project", 10);
 		// create the project
 		try {
 			if (!project.exists()) {
 				IProjectDescription desc = project.getWorkspace().newProjectDescription(project.getName());
 				if (Platform.getLocation().equals(locationPath)) {
 					locationPath = null;
 				}
 				desc.setLocation(locationPath);
 				project.create(desc, monitor);
 				monitor = null;
 			}
 			if (!project.isOpen()) {
 				project.open(monitor);
 				monitor = null;
 			}
 		}
 		finally {
 			if (monitor != null) {
 				monitor.done();
 			}
 		}
 	}

 	private static void getAndCreateProject() throws CoreException {
 		//TestsPlugin testsPlugin = (TestsPlugin)
 		// Platform.getPlugin("org.eclipse.wst.xml.tests.encoding");
 		IWorkspace workspace = TestsPlugin.getWorkspace();
 		IWorkspaceRoot root = workspace.getRoot();
 		fTestProject = root.getProject(TEST_PROJECT_NAME);
 		// this form creates project as "linked" back to 'fileRoot'
 		//createProject(testProject, new Path(fileRoot), null);
 		createProject(fTestProject, null, null);
 		fTestProject.refreshLocal(IResource.DEPTH_INFINITE, null);
 		assertTrue(fTestProject.exists());
 		//		IContainer testFiles = testProject.getFolder("testfiles");
 		//		assertTrue(testFiles.exists());
 		//		IResource[] allFolders = testFiles.members();
 		//		assertNotNull(allFolders);
 	}

 	private void createTestMethodSource(int count, String filePathAndName, String detectedCharsetName, String javaCharsetName, String expectedException) {
 		String javaCharsetNameOrKey = javaCharsetName;
 		if (null != javaCharsetNameOrKey && javaCharsetNameOrKey.equals(currentPlatformCharset))
 			javaCharsetNameOrKey = "expectPlatformCharset";
 		System.out.println("public void testFile" + count + " () throws CoreException, IOException {" + " doTest( \"" + javaCharsetNameOrKey + "\", \"" + detectedCharsetName + "\",  \"" + filePathAndName + "\", " + expectedException + "); }");
 	}

 	protected void setUp() throws Exception {
 		super.setUp();
 		nSetups++;
 		if (!fTestProjectInitialized) {
 			getAndCreateProject();
 			// unzip files to the root of workspace directory
 			String destinationProjectString = fTestProject.getLocation().toOSString();
 			String destinationFolder = destinationProjectString + "/";
 			// this zip file is sitting in the "root" of test plugin
 			File zipFile = TestsPlugin.getTestFile("testfiles.zip");
 			ProjectUnzipUtility projUtil = new ProjectUnzipUtility();
 			projUtil.unzipAndImport(zipFile, destinationFolder);
 			projUtil.initJavaProject(TEST_PROJECT_NAME);
 			fTestProject.refreshLocal(IResource.DEPTH_INFINITE, null);
 			fTestProjectInitialized = true;
 		}
 	}

 	protected void tearDown() throws Exception {
 		super.tearDown();
 		nSetups--;
 		if (nSetups == 0) {
 			if (!DEBUG) {
 				//				Display display = PlatformUI.getWorkbench().getDisplay();
 				//				display.asyncExec(new Runnable() {
 				//					public void run() {
 				//						ProjectUnzipUtility projUtil = new ProjectUnzipUtility();
 				//						IProject proj = fTestProject;
 				//						fTestProject = null;
 				//						try {
 				//							projUtil.deleteProject(proj);
 				//						} catch (Exception e) {
 				//							e.printStackTrace();
 				//						}
 				//					}
 				//				});
 			}
 		}
 	}

 	public void testCreateAllFiles() throws CoreException, IOException {
 		if (RECREATE_FILES) {
 			List allFiles = TestsPlugin.getAllTestFiles(TEST_FILE_DIR);
 			URL outputDirURL = TestsPlugin.getInstallLocation();
 			File zipoutFile = new File(outputDirURL.getPath(), TESTFILES_ZIPFILE_NAME);
 			java.io.FileOutputStream zipOut = new FileOutputStream(zipoutFile);
 			ZipOutputStream zipOutputStream = new ZipOutputStream(zipOut);
 			int count = 1;
 			for (Iterator iter = allFiles.iterator(); iter.hasNext();) {
 				File file = (File) iter.next();
 				createZipEntry(zipOutputStream, file);
 				CodedReaderCreator codedReaderCreator = new CodedReaderCreator();
 				codedReaderCreator.set(file.getName(), new FileInputStream(file));
 				String detectedCharsetName = null;
 				String javaCharsetName = null;
 				String expectedException = null;
 				try {
 					// just used for debug info, but can throw exception
 					javaCharsetName = codedReaderCreator.getEncodingMemento().getJavaCharsetName();
 					detectedCharsetName = codedReaderCreator.getEncodingMemento().getDetectedCharsetName();
 				}
 				catch (UnsupportedCharsetExceptionWithDetail e) {
 					// ignore for simply creating tests
 					expectedException = e.getClass().getName() + ".class";
 				}
 				catch (MalformedInputException e) {
 					// ignore for simply creating tests
 					expectedException = e.getClass().getName() + ".class";
 				}
 				catch (IllegalCharsetNameException e) {
 					// ignore for simply creating tests
 					expectedException = e.getClass().getName() + ".class";
 				}
 				String subpath = getSubPathName(file);
 				createTestMethodSource(count, subpath, detectedCharsetName, javaCharsetName, expectedException);
 				count++;
 			}
 			zipOutputStream.close();
 			zipOut.close();
 			assertTrue(true);
 		}
 	}

 	private String getSubPathName(File file) {
 		String path = file.getPath();
 		int lastIndex = path.lastIndexOf(TEST_FILE_DIR);
 		String subpath = path.substring(lastIndex);
 		subpath = subpath.replace('\\', '/');
 		return subpath;
 	}

 	/**
 	 * @param zipOutputStream
 	 * @param element
 	 */
 	private void createZipEntry(ZipOutputStream zipOutputStream, File file) throws IOException {
 		String subPathName = getSubPathName(file);
 		ZipEntry zipEntry = new ZipEntry(subPathName);
 		zipOutputStream.putNextEntry(zipEntry);
 		InputStream inputStream = new FileInputStream(file);
 		int nRead = 0;
 		byte[] buffer = new byte[1024 * 8];
 		while (nRead != -1) {
 			nRead = inputStream.read(buffer);
 			if (nRead > 0) {
 				zipOutputStream.write(buffer, 0, nRead);
 			}
 		}
 		zipOutputStream.flush();
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2004, 2017 IBM Corporation and others.
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License 2.0
	* which accompanies this distribution, and is available at
	* https://www.eclipse.org/legal/epl-2.0/
	*
	* SPDX-License-Identifier: EPL-2.0
	*
	* Contributors:
	* IBM Corporation - initial API and implementation
	*******************************************************************************/
	package org.eclipse.wst.xml.tests.encoding;

	import java.io.File;
	import java.io.FileInputStream;
	import java.io.FileOutputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.net.URL;
	import java.nio.charset.IllegalCharsetNameException;
	import java.nio.charset.MalformedInputException;
	import java.util.Iterator;
	import java.util.List;
	import java.util.zip.ZipEntry;
	import java.util.zip.ZipOutputStream;

	import junit.framework.TestCase;

	import org.eclipse.core.resources.IProject;
	import org.eclipse.core.resources.IProjectDescription;
	import org.eclipse.core.resources.IResource;
	import org.eclipse.core.resources.IWorkspace;
	import org.eclipse.core.resources.IWorkspaceRoot;
	import org.eclipse.core.runtime.CoreException;
	import org.eclipse.core.runtime.IPath;
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.core.runtime.NullProgressMonitor;
	import org.eclipse.core.runtime.Platform;
	import org.eclipse.wst.sse.core.internal.encoding.CodedReaderCreator;
	import org.eclipse.wst.sse.core.internal.encoding.NonContentBasedEncodingRules;
	import org.eclipse.wst.sse.core.internal.exceptions.UnsupportedCharsetExceptionWithDetail;
	import org.eclipse.wst.xml.tests.encoding.util.ProjectUnzipUtility;

	/**
	* This class is intended to be executed only at development time, and it
	* creates the class to JUnit test all the files found in the testfiles
	* directory.
	*/
	public class CreateCodedReaderTester extends TestCase {
	// private final String fileDir = "html/";
	// private final String fileRoot =
	// "/builds/Workspaces/HeadWorkspace/org.eclipse.wst.xml.tests.encoding/";
	// private final String fileLocation = fileRoot + fileDir;
	private static final String TESTFILES_ZIPFILE_NAME = "testfiles.zip"; //$NON-NLS-1$
	private static final boolean DEBUG = false;
	private static final String TEST_FILE_DIR = "testfiles"; //$NON-NLS-1$
	// needs to be static, since JUnit creates difference instances for each
	// test
	private static boolean fTestProjectInitialized;
	private static final String TEST_PROJECT_NAME = "org.eclipse.wst.xml.tests.encoding"; //$NON-NLS-1$
	static IProject fTestProject;
	private static int nSetups = 0;
	private static final String currentPlatformCharset = getPlatformDefault();
	private boolean RECREATE_FILES = false;

	/**
	*
	*/
	public CreateCodedReaderTester() {
	super();
	//System.out.println(currentPlatformCharset);
	}

	/**
	* @return
	*/
	private static String getPlatformDefault() {
	String platformDefault = NonContentBasedEncodingRules.useDefaultNameRules(null);
	return platformDefault;
	}

	public static void main(String[] args) {
	// try {
	// new CreateCodedReaderTester().doCreateAllFiles();
	// } catch (CoreException e) {
	// e.printStackTrace();
	// } catch (IOException e) {
	// e.printStackTrace();
	// }
	}

	private static void createProject(IProject project, IPath locationPath, IProgressMonitor monitor) throws CoreException {
	if (monitor == null) {
	monitor = new NullProgressMonitor();
	}
	monitor.beginTask("creating test project", 10);
	// create the project
	try {
	if (!project.exists()) {
	IProjectDescription desc = project.getWorkspace().newProjectDescription(project.getName());
	if (Platform.getLocation().equals(locationPath)) {
	locationPath = null;
	}
	desc.setLocation(locationPath);
	project.create(desc, monitor);
	monitor = null;
	}
	if (!project.isOpen()) {
	project.open(monitor);
	monitor = null;
	}
	}
	finally {
	if (monitor != null) {
	monitor.done();
	}
	}
	}

	private static void getAndCreateProject() throws CoreException {
	//TestsPlugin testsPlugin = (TestsPlugin)
	// Platform.getPlugin("org.eclipse.wst.xml.tests.encoding");
	IWorkspace workspace = TestsPlugin.getWorkspace();
	IWorkspaceRoot root = workspace.getRoot();
	fTestProject = root.getProject(TEST_PROJECT_NAME);
	// this form creates project as "linked" back to 'fileRoot'
	//createProject(testProject, new Path(fileRoot), null);
	createProject(fTestProject, null, null);
	fTestProject.refreshLocal(IResource.DEPTH_INFINITE, null);
	assertTrue(fTestProject.exists());
	// IContainer testFiles = testProject.getFolder("testfiles");
	// assertTrue(testFiles.exists());
	// IResource[] allFolders = testFiles.members();
	// assertNotNull(allFolders);
	}

	private void createTestMethodSource(int count, String filePathAndName, String detectedCharsetName, String javaCharsetName, String expectedException) {
	String javaCharsetNameOrKey = javaCharsetName;
	if (null != javaCharsetNameOrKey && javaCharsetNameOrKey.equals(currentPlatformCharset))
	javaCharsetNameOrKey = "expectPlatformCharset";
	System.out.println("public void testFile" + count + " () throws CoreException, IOException {" + " doTest( \"" + javaCharsetNameOrKey + "\", \"" + detectedCharsetName + "\", \"" + filePathAndName + "\", " + expectedException + "); }");
	}

	protected void setUp() throws Exception {
	super.setUp();
	nSetups++;
	if (!fTestProjectInitialized) {
	getAndCreateProject();
	// unzip files to the root of workspace directory
	String destinationProjectString = fTestProject.getLocation().toOSString();
	String destinationFolder = destinationProjectString + "/";
	// this zip file is sitting in the "root" of test plugin
	File zipFile = TestsPlugin.getTestFile("testfiles.zip");
	ProjectUnzipUtility projUtil = new ProjectUnzipUtility();
	projUtil.unzipAndImport(zipFile, destinationFolder);
	projUtil.initJavaProject(TEST_PROJECT_NAME);
	fTestProject.refreshLocal(IResource.DEPTH_INFINITE, null);
	fTestProjectInitialized = true;
	}
	}

	protected void tearDown() throws Exception {
	super.tearDown();
	nSetups--;
	if (nSetups == 0) {
	if (!DEBUG) {
	// Display display = PlatformUI.getWorkbench().getDisplay();
	// display.asyncExec(new Runnable() {
	// public void run() {
	// ProjectUnzipUtility projUtil = new ProjectUnzipUtility();
	// IProject proj = fTestProject;
	// fTestProject = null;
	// try {
	// projUtil.deleteProject(proj);
	// } catch (Exception e) {
	// e.printStackTrace();
	// }
	// }
	// });
	}
	}
	}

	public void testCreateAllFiles() throws CoreException, IOException {
	if (RECREATE_FILES) {
	List allFiles = TestsPlugin.getAllTestFiles(TEST_FILE_DIR);
	URL outputDirURL = TestsPlugin.getInstallLocation();
	File zipoutFile = new File(outputDirURL.getPath(), TESTFILES_ZIPFILE_NAME);
	java.io.FileOutputStream zipOut = new FileOutputStream(zipoutFile);
	ZipOutputStream zipOutputStream = new ZipOutputStream(zipOut);
	int count = 1;
	for (Iterator iter = allFiles.iterator(); iter.hasNext();) {
	File file = (File) iter.next();
	createZipEntry(zipOutputStream, file);
	CodedReaderCreator codedReaderCreator = new CodedReaderCreator();
	codedReaderCreator.set(file.getName(), new FileInputStream(file));
	String detectedCharsetName = null;
	String javaCharsetName = null;
	String expectedException = null;
	try {
	// just used for debug info, but can throw exception
	javaCharsetName = codedReaderCreator.getEncodingMemento().getJavaCharsetName();
	detectedCharsetName = codedReaderCreator.getEncodingMemento().getDetectedCharsetName();
	}
	catch (UnsupportedCharsetExceptionWithDetail e) {
	// ignore for simply creating tests
	expectedException = e.getClass().getName() + ".class";
	}
	catch (MalformedInputException e) {
	// ignore for simply creating tests
	expectedException = e.getClass().getName() + ".class";
	}
	catch (IllegalCharsetNameException e) {
	// ignore for simply creating tests
	expectedException = e.getClass().getName() + ".class";
	}
	String subpath = getSubPathName(file);
	createTestMethodSource(count, subpath, detectedCharsetName, javaCharsetName, expectedException);
	count++;
	}
	zipOutputStream.close();
	zipOut.close();
	assertTrue(true);
	}
	}

	private String getSubPathName(File file) {
	String path = file.getPath();
	int lastIndex = path.lastIndexOf(TEST_FILE_DIR);
	String subpath = path.substring(lastIndex);
	subpath = subpath.replace('\\', '/');
	return subpath;
	}

	/**
	* @param zipOutputStream
	* @param element
	*/
	private void createZipEntry(ZipOutputStream zipOutputStream, File file) throws IOException {
	String subPathName = getSubPathName(file);
	ZipEntry zipEntry = new ZipEntry(subPathName);
	zipOutputStream.putNextEntry(zipEntry);
	InputStream inputStream = new FileInputStream(file);
	int nRead = 0;
	byte[] buffer = new byte[1024 * 8];
	while (nRead != -1) {
	nRead = inputStream.read(buffer);
	if (nRead > 0) {
	zipOutputStream.write(buffer, 0, nRead);
	}
	}
	zipOutputStream.flush();
	}
	}