Test305892.java example

Explorer
emf.texo-master
/**
 * <copyright>
 *
 * Copyright (c) 2012 Springsite BV (The Netherlands) and others
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Martin Taal - Initial API and implementation
 *
 * </copyright>
 *
 * $Id: TestBidirectionalAssociationSupport.java,v 1.2 2011/08/25 12:42:19 mtaal Exp $
 */
package org.eclipse.emf.texo.test.emfissues;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import junit.framework.TestCase;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EPackage.Registry;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.xmi.XMLResource;
import org.eclipse.emf.ecore.xmi.impl.EcoreResourceFactoryImpl;
import org.eclipse.emf.ecore.xmi.impl.XMIResourceImpl;
import org.eclipse.emf.texo.datagenerator.ModelDataGenerator;

public class Test305892 extends TestCase {

  public void test305892() throws Exception {
    final EPackage ePackage = readRegisterEPackage();
    assertNotNull(ePackage);
    // final List<EObject> testData = generateTestSet(ePackage, 1, 1, 1, 10000);

    final Map<Object, Object> options = new HashMap<Object, Object>();
    options.put(XMLResource.OPTION_ENCODING, "UTF-8"); //$NON-NLS-1$
    options.put(XMLResource.OPTION_EXTENDED_META_DATA, true);
    options.put(XMLResource.OPTION_SUPPRESS_DOCUMENT_ROOT, true);
    options.put(XMLResource.OPTION_SCHEMA_LOCATION, true);
    options.put(XMLResource.OPTION_USE_ENCODED_ATTRIBUTE_STYLE, true);
    options.put(XMLResource.OPTION_USE_LEXICAL_HANDLER, true);

    //      final Resource res = new XMIResourceImpl(URI.createURI("accounting.xmi")); //$NON-NLS-1$
    //      final FileOutputStream fos = new FileOutputStream("/tmp/acounting_1.xmi"); //$NON-NLS-1$
    // res.getContents().addAll(testData);
    // res.save(fos, options);
    // fos.close();

    final XMLResource res = new XMIResourceImpl(URI.createURI("accounting.xmi")); //$NON-NLS-1$

    res.load(this.getClass().getResourceAsStream("accounting.xmi"), Collections.emptyMap()); //$NON-NLS-1$
    res.save(new FileOutputStream("/tmp/accounting_out.xmi"), options); //$NON-NLS-1$
  }

  protected List<EObject> generateTestSet(EPackage ePackage, final int dataSize, final int collectionSize,
      final int maxDepth, final int maxObjects) {
    final List<EClass> eClasses = new ArrayList<EClass>();
    for (EClassifier eClassifier : ePackage.getEClassifiers()) {
      if (eClassifier instanceof EClass) {
        eClasses.add((EClass) eClassifier);
      }
    }
    final ModelDataGenerator modelDataGenerator = new ModelDataGenerator();
    modelDataGenerator.setStartEClasses(eClasses);
    modelDataGenerator.setMaxDepth(maxDepth);
    modelDataGenerator.setCollectionSize(collectionSize);
    modelDataGenerator.setDataSize(dataSize);
    modelDataGenerator.setMaxObjects(maxObjects);
    modelDataGenerator.setEPackages(Collections.singletonList(ePackage));
    modelDataGenerator.generateTestData();
    System.err.println(modelDataGenerator.getTotalObjectCount());
    final List<EObject> list = modelDataGenerator.getResult();
    return list;
  }

  private EPackage readRegisterEPackage() {
    final String ecoreFileName = "accounting.ecore"; //$NON-NLS-1$
    try {

      // note the resource uri is the same as by which it is saved in the
      // GenEPackage.getECoreFileContent
      final Resource res = new EcoreResourceFactoryImpl().createResource(URI.createURI(ecoreFileName));
      final InputStream is = this.getClass().getResourceAsStream(ecoreFileName);
      res.load(is, Collections.EMPTY_MAP);
      is.close();

      final Iterator<?> it = res.getAllContents();
      while (it.hasNext()) {
        final Object obj = it.next();
        if (obj instanceof EPackage) {
          final EPackage epack = (EPackage) obj;
          return registerEPackage(epack);
        }
      }
    } catch (final IOException e) {
      throw new IllegalStateException("Exception while loading models from ecorefile " + ecoreFileName);//$NON-NLS-1$
    }
    return null;
  }

  private EPackage registerEPackage(EPackage ePackage) {
    final Registry registry = EPackage.Registry.INSTANCE;
    if (registry.containsKey(ePackage.getNsURI())) {
      //      log.warn("EPackage with this \"" + ePackage.getNsURI() //$NON-NLS-1$
      //          + "\" already registered. Returning the registered one if it is an epackage");//$NON-NLS-1$
      final Object packageObject = registry.get(ePackage.getNsURI());
      if (packageObject instanceof EPackage) {
        return (EPackage) packageObject;
      }
    }

    registry.put(ePackage.getNsURI(), ePackage);
    return ePackage;
  }
}