change internal process method, which introducing ExtractionResultSet

[stigmata/stigmata.git] / src / test / java / jp / naist / se / stigmata / birthmarks / uc / UCBirthmarkExtractorTest.java

diff --git a/src/test/java/jp/naist/se/stigmata/birthmarks/uc/UCBirthmarkExtractorTest.java b/src/test/java/jp/naist/se/stigmata/birthmarks/uc/UCBirthmarkExtractorTest.java
index 60c1969..c642b2f 100755 (executable)
--- a/src/test/java/jp/naist/se/stigmata/birthmarks/uc/UCBirthmarkExtractorTest.java
+++ b/src/test/java/jp/naist/se/stigmata/birthmarks/uc/UCBirthmarkExtractorTest.java
@@ -4,15 +4,17 @@ package jp.naist.se.stigmata.birthmarks.uc;
  * $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.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
@@ -20,59 +22,52 @@ import jp.naist.se.stigmata.Stigmata;
  */\r
 public class UCBirthmarkExtractorTest{\r
     private Stigmata stigmata;\r
+    private BirthmarkContext context;\r
 \r
     @Before\r
     public void setup(){\r
         stigmata = Stigmata.getInstance();\r
+        context = stigmata.createContext();\r
+        context.addExtractionType("uc");\r
     }\r
 \r
     @Test\r
     public void checkUCBirthmark() throws Exception{\r
-        BirthmarkSet[] array = stigmata.extract(\r
-            new String[] { "uc", },\r
-            new String[] { "target/classes/jp/naist/se/stigmata/Stigmata.class", }\r
+        ExtractionResultSet ers = stigmata.createEngine().extract(\r
+            new String[] { "target/classes/jp/naist/se/stigmata/Stigmata.class", },\r
+            context\r
         );\r
+        BirthmarkSet[] array = ers.getBirthmarkSets();\r
 \r
         Assert.assertEquals(1, array.length);\r
         Assert.assertNotNull(array[0].getBirthmark("uc"));\r
 \r
         Birthmark birthmark = array[0].getBirthmark("uc");\r
         Assert.assertEquals("uc", birthmark.getType());\r
-        Assert.assertEquals(23, birthmark.getElementCount());\r
+        Assert.assertEquals(11, birthmark.getElementCount());\r
 \r
         BirthmarkElement[] elements = birthmark.getElements();\r
         int index = 0;\r
-        Assert.assertEquals("java.io.ByteArrayInputStream",                elements[index++].getValue());\r
-        Assert.assertEquals("java.io.ByteArrayOutputStream",               elements[index++].getValue());\r
-        Assert.assertEquals("java.io.File",                                elements[index++].getValue());\r
-        Assert.assertEquals("java.io.FileInputStream",                     elements[index++].getValue());\r
-        Assert.assertEquals("java.io.InputStream",                         elements[index++].getValue());\r
-        Assert.assertEquals("java.lang.Class",                             elements[index++].getValue());\r
-        Assert.assertEquals("java.lang.Object",                            elements[index++].getValue());\r
-        Assert.assertEquals("java.lang.String",                            elements[index++].getValue());\r
-        Assert.assertEquals("java.lang.StringBuilder",                     elements[index++].getValue());\r
-        Assert.assertEquals("java.lang.System",                            elements[index++].getValue());\r
-        Assert.assertEquals("java.net.URI",                                elements[index++].getValue());\r
-        Assert.assertEquals("java.net.URL",                                elements[index++].getValue());\r
-        Assert.assertEquals("java.util.ArrayList",                         elements[index++].getValue());\r
-        Assert.assertEquals("java.util.Collection",                        elements[index++].getValue());\r
-        Assert.assertEquals("java.util.HashMap",                           elements[index++].getValue());\r
-        Assert.assertEquals("java.util.Iterator",                          elements[index++].getValue());\r
-        Assert.assertEquals("java.util.List",                              elements[index++].getValue());\r
-        Assert.assertEquals("java.util.Map",                               elements[index++].getValue());\r
-        Assert.assertEquals("java.util.Set",                               elements[index++].getValue());\r
-        Assert.assertEquals("java.util.Stack",                             elements[index++].getValue());\r
-        Assert.assertEquals("java.util.logging.Logger",                    elements[index++].getValue());\r
-        Assert.assertEquals("javax.imageio.spi.ServiceRegistry",           elements[index++].getValue());\r
-        Assert.assertEquals("org.apache.commons.beanutils.BeanUtils",      elements[index++].getValue());\r
+        Assert.assertEquals("java.io.File",                      elements[index++].getValue());\r
+        Assert.assertEquals("java.io.FileInputStream",           elements[index++].getValue());\r
+        Assert.assertEquals("java.io.InputStream",               elements[index++].getValue());\r
+        Assert.assertEquals("java.lang.Class",                   elements[index++].getValue());\r
+        Assert.assertEquals("java.lang.Object",                  elements[index++].getValue());\r
+        Assert.assertEquals("java.lang.String",                  elements[index++].getValue());\r
+        Assert.assertEquals("java.lang.System",                  elements[index++].getValue());\r
+        Assert.assertEquals("java.util.ArrayList",               elements[index++].getValue());\r
+        Assert.assertEquals("java.util.Iterator",                elements[index++].getValue());\r
+        Assert.assertEquals("java.util.List",                    elements[index++].getValue());\r
+        Assert.assertEquals("javax.imageio.spi.ServiceRegistry", elements[index++].getValue());\r
     }\r
 \r
     @Test\r
     public void checkUCBirthmark2() throws Exception{\r
-        BirthmarkSet[] array = stigmata.extract(\r
-            new String[] { "uc", },\r
-            new String[] { "target/classes/jp/naist/se/stigmata/RoundRobinComparisonResultSet.class", }\r
+        ExtractionResultSet ers = stigmata.createEngine().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("uc"));\r
@@ -82,14 +77,15 @@ public class UCBirthmarkExtractorTest{
         Assert.assertEquals(9, birthmark.getElementCount());\r
 \r
         BirthmarkElement[] elements = birthmark.getElements();\r
-        Assert.assertEquals("java.lang.Object",    elements[0].getValue());\r
-        Assert.assertEquals("java.net.URL",        elements[1].getValue());\r
-        Assert.assertEquals("java.util.Arrays",    elements[2].getValue());\r
-        Assert.assertEquals("java.util.HashMap",   elements[3].getValue());\r
-        Assert.assertEquals("java.util.Iterator",  elements[4].getValue());\r
-        Assert.assertEquals("java.util.List",      elements[5].getValue());\r
-        Assert.assertEquals("java.util.Map",       elements[6].getValue());\r
-        Assert.assertEquals("java.util.Map$Entry", elements[7].getValue());\r
-        Assert.assertEquals("java.util.Set",       elements[8].getValue());\r
+        int index = 0;\r
+        Assert.assertEquals("java.lang.Object",     elements[index++].getValue());\r
+        Assert.assertEquals("java.lang.String",     elements[index++].getValue());\r
+        Assert.assertEquals("java.net.URL",         elements[index++].getValue());\r
+        Assert.assertEquals("java.util.ArrayList",  elements[index++].getValue());\r
+        Assert.assertEquals("java.util.Collection", elements[index++].getValue());\r
+        Assert.assertEquals("java.util.HashMap",    elements[index++].getValue());\r
+        Assert.assertEquals("java.util.Iterator",   elements[index++].getValue());\r
+        Assert.assertEquals("java.util.List",       elements[index++].getValue());\r
+        Assert.assertEquals("java.util.Map",        elements[index++].getValue());\r
     }\r
 }\r