X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.modeling.tests%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Ftests%2Fcommands%2FAssertIssueCount.java;fp=bundles%2Forg.simantics.modeling.tests%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Ftests%2Fcommands%2FAssertIssueCount.java;h=c88a9cc8a5ef9658aa193994914b716f66be3d2d;hb=ca6fcd858a0b82eb1b5a8f12c4d5df30e84393e0;hp=0000000000000000000000000000000000000000;hpb=67fd62f9c742337ec80eef658192db198a0efaac;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.modeling.tests/src/org/simantics/modeling/tests/commands/AssertIssueCount.java b/bundles/org.simantics.modeling.tests/src/org/simantics/modeling/tests/commands/AssertIssueCount.java
new file mode 100644
index 000000000..c88a9cc8a
--- /dev/null
+++ b/bundles/org.simantics.modeling.tests/src/org/simantics/modeling/tests/commands/AssertIssueCount.java
@@ -0,0 +1,38 @@
+package org.simantics.modeling.tests.commands;
+
+import java.util.Set;
+
+import junit.framework.Assert;
+
+import org.simantics.db.Resource;
+import org.simantics.db.WriteGraph;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.variable.Variable;
+import org.simantics.db.testing.common.WriteCommand;
+import org.simantics.issues.common.IssuesWithContext;
+import org.simantics.modeling.ModelingResources;
+import org.simantics.modeling.tests.cases.ModelingCommandSequenceTest;
+
+
+public class AssertIssueCount extends WriteCommand<ModelingCommandSequenceTest> {
+
+	private AddModule module;
+	private int expected;
+	
+	public AssertIssueCount(AddModule module, int expected) {
+		this.module = module;
+		this.expected = expected;
+	}
+	
+	@Override
+	protected void run(WriteGraph graph, ModelingCommandSequenceTest environment) throws DatabaseException {
+		
+        Resource element = module.getResource();
+        Resource component = graph.getSingleObject(element, ModelingResources.getInstance(graph).ElementToComponent);
+        Set<Variable> issues = graph.sync(new IssuesWithContext(graph, component));
+
+        Assert.assertEquals(expected, issues.size());
+        
+	}
+
+}
\ No newline at end of file