implemented missing methods in TaintedMapGene

Author: arcuri82

core/src/main/kotlin/org/evomaster/core/search/gene/Gene.kt

@@ -1210,8 +1210,7 @@ abstract class Gene(
             //revert back
             val success = unsafeCopyValueFrom(current)
             //reversion should always work... if fails, it is a bug
-            // FIXME put back once all are implemented, eg, TaintedMapGene currently missing
-            //assert(success)
+            assert(success)
             return false
         }
         return true

core/src/main/kotlin/org/evomaster/core/search/gene/collection/TaintedMapGene.kt

@@ -220,16 +220,44 @@ class TaintedMapGene(
 
 
     override fun containsSameValueAs(other: Gene): Boolean {
-        //TODO
 
-        return false
+        if(other !is TaintedMapGene){
+            return false
+        }
+
+        if(this.elements.size != other.elements.size){
+            return false
+        }
+
+        for(e in this.elements){
+            val x = other.elements.find { it.first.name == e.first.name }
+                ?: return false
+            if(! e.second.containsSameValueAs(x.second)){
+                return false
+            }
+        }
+
+        return true
     }
 
     override fun unsafeCopyValueFrom(other: Gene): Boolean {
-        //TODO
-        LoggingUtil.uniqueWarn(log,"unsafeCopyValueFrom() not implemented for TaintedMapGene")
 
-        return false
+        if(other !is TaintedMapGene){
+            return false
+        }
+
+        killAllChildren()
+        learnedKeys.clear()
+        learnedKeys.addAll(other.learnedKeys)
+        learnedTypes.clear()
+        learnedTypes.putAll(other.learnedTypes)
+
+        other.elements.forEach {
+            addElement(it.copy() as PairGene<StringGene, Gene>)
+        }
+        taintId = other.taintId
+
+        return true
     }
 
 

core/src/test/kotlin/org/evomaster/core/search/gene/collection/TaintedMapGeneTest.kt

@@ -62,6 +62,36 @@ class TaintedMapGeneTest{
     }
 
 
+    @Test
+    fun testCopyFrom() {
+
+        val foo = TaintedMapGene("foo", TaintInputName.getTaintName(42))
+        foo.doInitialize()
+        assertEquals(1, foo.getSizeOfElements())
+
+        //x
+        foo.registerKey("x")
+        foo.evolve()
+        foo.registerNewType("x", "java/lang/Integer")
+        foo.evolve()
+
+        val bar = TaintedMapGene("bar", TaintInputName.getTaintName(66))
+        bar.doInitialize()
+
+        //y
+        bar.registerKey("y")
+        bar.evolve()
+        bar.registerNewType("y", "java/lang/Boolean")
+        bar.evolve()
+
+        assertFalse(foo.containsSameValueAs(bar))
+
+        val copied = foo.copyValueFrom(bar)
+        assertTrue(copied)
+
+        assertTrue(foo.containsSameValueAs(bar))
+    }
+
     private fun verifyCollection(key: String, type: String, lambda: (ArrayDto) -> Any?){
 
         val gene = TaintedMapGene("foo", TaintInputName.getTaintName(42))