change internal process method, which introducing ExtractionResultSet

[stigmata/stigmata.git] / src / test / java / jp / naist / se / stigmata / birthmarks / is / ISBirthmarkExtractorTest.java

diff --git a/src/test/java/jp/naist/se/stigmata/birthmarks/is/ISBirthmarkExtractorTest.java b/src/test/java/jp/naist/se/stigmata/birthmarks/is/ISBirthmarkExtractorTest.java
index 8ec496c..5bce15b 100755 (executable)
--- a/src/test/java/jp/naist/se/stigmata/birthmarks/is/ISBirthmarkExtractorTest.java
+++ b/src/test/java/jp/naist/se/stigmata/birthmarks/is/ISBirthmarkExtractorTest.java
@@ -4,35 +4,43 @@ package jp.naist.se.stigmata.birthmarks.is;
  * $Id$\r
  */\r
 \r
-import org.junit.Assert;\r
-import org.junit.Before;\r
-import org.junit.Test;\r
-\r
 import jp.naist.se.stigmata.Birthmark;\r
+import jp.naist.se.stigmata.BirthmarkContext;\r
 import jp.naist.se.stigmata.BirthmarkElement;\r
+import jp.naist.se.stigmata.BirthmarkEngine;\r
 import jp.naist.se.stigmata.BirthmarkSet;\r
+import jp.naist.se.stigmata.ExtractionResultSet;\r
 import jp.naist.se.stigmata.Stigmata;\r
 \r
+import org.junit.Assert;\r
+import org.junit.Before;\r
+import org.junit.Test;\r
+\r
 /**\r
  *\r
  * @author Haruaki TAMADA\r
  * @version $Revision$ $Date$\r
  */\r
 public class ISBirthmarkExtractorTest{\r
-    private Stigmata stigmata;\r
+    private BirthmarkEngine engine;\r
+    private BirthmarkContext context;\r
 \r
     @Before\r
     public void setup(){\r
-        stigmata = Stigmata.getInstance();\r
+        engine = Stigmata.getInstance().createEngine();\r
+        context = Stigmata.getInstance().createContext();\r
+        context.addExtractionType("is");\r
     }\r
 \r
     @Test\r
     public void checkISBirthmark() throws Exception{\r
-        BirthmarkSet[] array = stigmata.extract(\r
-            new String[] { "is", },\r
-            new String[] { "target/classes/jp/naist/se/stigmata/Stigmata.class", }\r
+        ExtractionResultSet ers = engine.extract(\r
+            new String[] { "target/classes/jp/naist/se/stigmata/Stigmata.class", },\r
+            context\r
         );\r
 \r
+        BirthmarkSet[] array = ers.getBirthmarkSets();\r
+\r
         Assert.assertEquals(1, array.length);\r
         Assert.assertNotNull(array[0].getBirthmark("is"));\r
 \r
@@ -47,20 +55,22 @@ public class ISBirthmarkExtractorTest{
 \r
     @Test\r
     public void checkISBirthmark2() throws Exception{\r
-        BirthmarkSet[] array = stigmata.extract(\r
-            new String[] { "is", },\r
-            new String[] { "target/classes/jp/naist/se/stigmata/RoundRobinComparisonResultSet.class", }\r
+        ExtractionResultSet ers = engine.extract(\r
+            new String[] { "target/classes/jp/naist/se/stigmata/result/RoundRobinComparisonResultSet.class", },\r
+            context\r
         );\r
+        BirthmarkSet[] array = ers.getBirthmarkSets();\r
 \r
         Assert.assertEquals(1, array.length);\r
         Assert.assertNotNull(array[0].getBirthmark("is"));\r
 \r
         Birthmark birthmark = array[0].getBirthmark("is");\r
         Assert.assertEquals("is", birthmark.getType());\r
-        Assert.assertEquals(2, birthmark.getElementCount());\r
+        Assert.assertEquals(3, birthmark.getElementCount());\r
 \r
         BirthmarkElement[] elements = birthmark.getElements();\r
         Assert.assertNull(elements[0].getValue());\r
-        Assert.assertEquals("java.lang.Object", elements[1].getValue());\r
+        Assert.assertNull(elements[1].getValue());\r
+        Assert.assertEquals("java.lang.Object", elements[2].getValue());\r
     }\r
 }\r