Bug 333848 - [Tests] Provide unit tests for New Model/Project Wizards

1 files changed, 48 insertions, 0 deletions

diff --git a/tests/junit/plugins/core/org.eclipse.papyrus.core.tests/test/org/eclipse/papyrus/core/wizard/TestNewSysMLProjectWizard.java b/tests/junit/plugins/core/org.eclipse.papyrus.core.tests/test/org/eclipse/papyrus/core/wizard/TestNewSysMLProjectWizard.java
new file mode 100644
index 00000000000..59dc572c0c3
--- /dev/null
+++ b/tests/junit/plugins/core/org.eclipse.papyrus.core.tests/test/org/eclipse/papyrus/core/wizard/TestNewSysMLProjectWizard.java
@@ -0,0 +1,48 @@
+package org.eclipse.papyrus.core.wizard;
+
+import org.eclipse.papyrus.sysml.diagram.common.commands.CreateSysMLModelCommand;
+import org.eclipse.papyrus.sysml.diagram.ui.NewSysMLModelWizard;
+import org.eclipse.papyrus.sysml.diagram.ui.NewSysMLProjectWizard;
+import org.eclipse.papyrus.wizards.NewPapyrusProjectWizard;
+import org.eclipse.papyrus.wizards.pages.NewModelFilePage;
+import org.eclipse.papyrus.wizards.pages.SelectDiagramCategoryPage;
+import org.eclipse.papyrus.wizards.pages.SelectDiagramKindPage;
+import org.eclipse.ui.IWorkbenchWizard;
+import org.eclipse.ui.dialogs.WizardNewProjectCreationPage;
+
+
+public class TestNewSysMLProjectWizard extends TestNewModelWizardBase {
+	
+	
+	@Override
+	protected IWorkbenchWizard createWizard() {
+		return new NewSysMLProjectWizard();
+	}
+
+	public void testOrderOfPages() {
+		
+		Class[] expectedPages = new Class[]{
+			WizardNewProjectCreationPage.class,
+			SelectDiagramKindPage.class,
+		};
+
+		IWorkbenchWizard wizard = initWizardDialog();
+		testOrderOfPages(wizard, expectedPages);
+	}
+
+	public void testDiagramCategory() {
+		class NewSysMLProjectWizardWithPublicAccess extends NewSysMLProjectWizard {
+			// open access to protected method
+			@Override
+			public String getDiagramCategoryId() {
+				return super.getDiagramCategoryId();
+			}
+		}
+		
+		NewSysMLProjectWizardWithPublicAccess wizard = new NewSysMLProjectWizardWithPublicAccess();
+		initWizardDialog(wizard);
+		String actualCategory = wizard.getDiagramCategoryId();
+		assertEquals(CreateSysMLModelCommand.COMMAND_ID, actualCategory);
+	}
+
+}