Add "Replace 'associate' with 'associateBy' or 'associateWith'" inspection

#KT-26269 Fixed

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/array.kt

@@ -0,0 +1,7 @@
+// PROBLEM: none
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    arrayOf(1).<caret>associate { it to getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).<caret>associate { it to getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith { getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic2.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associate<caret> { i -> i to getValue(i) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic2.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith { i -> getValue(i) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic3.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).<caret>associate { Pair(it, getValue(it)) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic3.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith { getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic4.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).<caret>associate({ it to getValue(it) })
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic4.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith({ getValue(it) })
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic5.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).<caret>associate { it to "$it" }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/basic5.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith { "$it" }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/intArray.kt

@@ -0,0 +1,7 @@
+// PROBLEM: none
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    intArrayOf(1).<caret>associate { it to getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/labeledReturn.kt

@@ -0,0 +1,12 @@
+// PROBLEM: none
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test(b: Boolean) {
+    listOf(1).<caret>associate {
+        if (b) {
+            return@associate it to ""
+        }
+        it to getValue(it)
+    }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/multiLine.kt

@@ -0,0 +1,11 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).<caret>associate {
+        val value = getValue(it)
+        it to value
+    }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/multiLine.kt.after

@@ -0,0 +1,11 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    listOf(1).associateWith {
+        val value = getValue(it)
+        value
+    }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/notPair.kt

@@ -0,0 +1,13 @@
+// PROBLEM: none
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test(b: Boolean) {
+    listOf(1).<caret>associate {
+        if (b) {
+            it to getValue(it)
+        } else {
+            it to ""
+        }
+    }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/sequence.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    sequenceOf(1).<caret>associate { it to getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/sequence.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    sequenceOf(1).associateWith { getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/set.kt

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    setOf(1).<caret>associate { it to getValue(it) }
+}

idea/testData/inspectionsLocal/replaceAssociateFunction/associateWith/set.kt.after

@@ -0,0 +1,8 @@
+// PROBLEM: Replace 'associate' with 'associateWith'
+// FIX: Replace with 'associateWith'
+// WITH_RUNTIME
+fun getValue(i: Int): String = ""
+
+fun test() {
+    setOf(1).associateWith { getValue(it) }
+}