ielementtest.java

是高效开发JAVA的XML工具

    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("Java Attribute");

			TransactionManager.endTransaction();

	        boolean isStereotypeAdded =
	        	getString(element.getStereotypes(), "Java Attribute") != null;
			assertTrue(isStereotypeAdded);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testJavaClassStereotypeInAttribute() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("Java Class");

			TransactionManager.endTransaction();
			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testJavaMethodStereotypeInAttribute() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("Java Method");

			TransactionManager.endTransaction();
			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testJavaMethodStereotypeInMethod() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getOperations(), "operation0");
	        element.addStereotype("Java Method");

			TransactionManager.endTransaction();

	        boolean isStereotypeAdded =
	        	getString(element.getStereotypes(), "Java Method") != null;
			assertTrue(isStereotypeAdded);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testCSharpClassStereotypeInClass() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
	        IElement element = getElement(project.getOwnedElements(), "Class0");
	        element.addStereotype("C# Class");

			TransactionManager.endTransaction();

	        boolean isStereotypeAdded =
	        	getString(element.getStereotypes(), "C# Class") != null;
			assertTrue(isStereotypeAdded);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testCSharpAttributeStereotypeInClass() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
	        IElement element = getElement(project.getOwnedElements(), "Class0");
	        element.addStereotype("C# Attribute");

			TransactionManager.endTransaction();
			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testCSharpMethodStereotypeInClass() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
	        IElement element = getElement(project.getOwnedElements(), "Class0");
	        element.addStereotype("C# Method");

			TransactionManager.endTransaction();

			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testCSharpAttributeStereotypeInAttribute() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("C# Attribute");

			TransactionManager.endTransaction();

	        boolean isStereotypeAdded =
	        	getString(element.getStereotypes(), "C# Attribute") != null;
			assertTrue(isStereotypeAdded);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testCSharpClassStereotypeInAttribute() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("C# Class");

			TransactionManager.endTransaction();

			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testCSharpMethodStereotypeInAttribute() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getAttributes(), "attribute0");
	        element.addStereotype("C# Method");

			TransactionManager.endTransaction();

			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testCSharpMethodStereotypeInOperation() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getOperations(), "operation0");
	        element.addStereotype("C# Method");

			TransactionManager.endTransaction();

	        boolean isStereotypeAdded =
	        	getString(element.getStereotypes(), "C# Method") != null;
			assertTrue(isStereotypeAdded);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testCSharpClassStereotypeInOperation() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getOperations(), "operation0");
	        element.addStereotype("C# Class");

			TransactionManager.endTransaction();

			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }

    public void testCSharpAttributeStereotypeInOperation() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageCSharp(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class1");
	        IElement element = getElement(cls.getOperations(), "operation0");
	        element.addStereotype("C# Attribute");

			TransactionManager.endTransaction();
			fail();
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    	}
    }
    //----------------------------------------------------------------------
    //stereotype order test
    //----------------------------------------------------------------------
    public void testStereotypeOrder() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class3");
			cls.addStereotype("Java Class");
			cls.addStereotype("interface");
			cls.addStereotype("stereotype");

			TransactionManager.endTransaction();

			String[] stereotypes = cls.getStereotypes();
			assertEquals("interface", stereotypes[0]);
			assertEquals("stereotype", stereotypes[1]);
			assertEquals("Java Class", stereotypes[2]);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }

    public void testStereotypeOrder2() throws InvalidEditingException, ClassNotFoundException {
    	BasicModelEditor bme = ModelEditorFactory.getBasicModelEditor();
    	try {
			TransactionManager.beginTransaction();

			bme.setLanguageJava(project, true);
			IClass cls = (IClass) getElement(project.getOwnedElements(), "Class3");
			cls.removeStereotype("stereotype");

			TransactionManager.endTransaction();

			String[] stereotypes = cls.getStereotypes();
			assertEquals("interface", stereotypes[0]);
			assertEquals("Java Class", stereotypes[1]);
    	} catch (InvalidEditingException e) {
    		TransactionManager.abortTransaction();
    		fail();
    	}
    }


    //----------------------------------------------------------------------
    //common
    //----------------------------------------------------------------------
    private String getString(String[] names, String target) {
    	for (int i = 0; i < names.length; i++) {
    		if (target.equals(names[i])) {
    			return names[i];
    		}
    	}
    	return null;
    }
}