[Feature] Map Collection Support

CHANGELOG.MD

@@ -1,3 +1,7 @@
+## [0.9.0] - Not released yet
+
+* Support for map collection type
+
 ## [0.8.0] - December 7th 2021
 
 * Auto generation for additional types:

README.md

@@ -183,6 +183,7 @@ Model Forge currently supports the auto generation for the following types:
 ### Collections
 
 * List
+* Map (_SNAPSHOT_)
 * Set
 
 _Can't find your data type? Feel free to create a pull request or open an issue_ :parachute:

forge-core/src/main/kotlin/io/github/hellocuriosity/ModelForge.kt

@@ -61,6 +61,7 @@ open class ModelForge {
                     field.isAccessible = true
                     when (field.type) {
                         List::class.java -> field.set(model, field.getValues())
+                        Map::class.java -> field.set(model, field.getValues())
                         Set::class.java -> field.set(model, field.getValues())
                         else -> realBuild(field.type)?.also {
                             field.set(model, it)
@@ -89,6 +90,26 @@ open class ModelForge {
         return list
     }
 
+    /**
+     *  Creates an automatically generated model map with
+     *  a specified size.
+     *
+     *  @param type ParameterizedType to instantiate
+     *  @param size Int number of models to create
+     *
+     *  @return Instance of clazz
+     */
+    private fun buildMap(type: ParameterizedType, size: Int): Map<Any, Any> {
+        val key = type.actualTypeArguments[0] as Class<*>
+        val value = type.actualTypeArguments[1] as Class<*>
+
+        val map = (0 until size).map {
+            build(key) to build(value)
+        }.toMap()
+
+        return map
+    }
+
     /**
      *  Creates an automatically generated model set with
      *  or without a specified size.
@@ -173,6 +194,7 @@ open class ModelForge {
 
         return when (this.type) {
             List::class.java -> buildList(clazz, size)
+            Map::class.java -> buildMap(type, size)
             Set::class.java -> buildSet(clazz, size)
             else -> throw ModelForgeException("Could not create values for: ${clazz.name}")
         }

forge-test-utils/src/main/kotlin/io/github/hellocuriosity/TestObject.kt

@@ -40,6 +40,8 @@ data class TestObject(
     val listOptional: List<ComplexObject>?,
     val longValue: Long,
     val longOptional: Long?,
+    val mapValues: Map<String, ComplexObject>,
+    val mapOptional: Map<String, ComplexObject>?,
     val setValues: Set<ComplexObject>,
     val setOptional: Set<ComplexObject>?,
     val shortValue: Short,
@@ -114,6 +116,10 @@ fun TestObject.assert() {
     assertEquals(10, listValues.size)
     assertNotNull(listOptional)
     assertEquals(10, listOptional.size)
+    assertNotNull(mapValues)
+    assertTrue(mapValues.size in 0 until 11)
+    assertNotNull(mapOptional)
+    assertTrue(mapOptional.size in 0 until 11)
     assertNotNull(setValues)
     assertEquals(10, setValues.size)
     assertNotNull(setOptional)