bundles/org.simantics.modeling.tests/src/org/simantics/modeling/tests/commands/AssertIssueCount.java

   1 package org.simantics.modeling.tests.commands;
   2
   3 import java.util.Set;
   4
   5 import junit.framework.Assert;
   6
   7 import org.simantics.db.Resource;
   8 import org.simantics.db.WriteGraph;
   9 import org.simantics.db.exception.DatabaseException;
  10 import org.simantics.db.layer0.variable.Variable;
  11 import org.simantics.db.testing.common.WriteCommand;
  12 import org.simantics.issues.common.IssuesWithContext;
  13 import org.simantics.modeling.ModelingResources;
  14 import org.simantics.modeling.tests.cases.ModelingCommandSequenceTest;
  15
  16
  17 public class AssertIssueCount extends WriteCommand<ModelingCommandSequenceTest> {
  18
  19         private AddModule module;
  20         private int expected;
  21
  22         public AssertIssueCount(AddModule module, int expected) {
  23                 this.module = module;
  24                 this.expected = expected;
  25         }
  26
  27         @Override
  28         protected void run(WriteGraph graph, ModelingCommandSequenceTest environment) throws DatabaseException {
  29
  30         Resource element = module.getResource();
  31         Resource component = graph.getSingleObject(element, ModelingResources.getInstance(graph).ElementToComponent);
  32         Set<Variable> issues = graph.sync(new IssuesWithContext(graph, component));
  33
  34         Assert.assertEquals(expected, issues.size());
  35
  36         }
  37
  38 }