From a3ea8698452c132ae4a756d617775f40f036b907 Mon Sep 17 00:00:00 2001
From: Ilya Gorbunov <ilya.gorbunov@jetbrains.com>
Date: Fri, 27 Oct 2017 01:55:09 +0300
Subject: Support assertFalse and message in assertTrue/assertFalse

---
 .../kotlin/Samples/KotlinWebsiteSampleProcessingService.kt    | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'core/src/main/kotlin/Samples/KotlinWebsiteSampleProcessingService.kt')

diff --git a/core/src/main/kotlin/Samples/KotlinWebsiteSampleProcessingService.kt b/core/src/main/kotlin/Samples/KotlinWebsiteSampleProcessingService.kt
index 4dd0bc7f..665a687f 100644
--- a/core/src/main/kotlin/Samples/KotlinWebsiteSampleProcessingService.kt
+++ b/core/src/main/kotlin/Samples/KotlinWebsiteSampleProcessingService.kt
@@ -35,14 +35,18 @@ open class KotlinWebsiteSampleProcessingService
             }
         }
 
-        fun convertAssertTrue(expression: KtCallExpression) {
+        fun convertAssertTrueFalse(expression: KtCallExpression, expectedResult: Boolean) {
             val (argument) = expression.valueArguments
             builder.apply {
+                expression.valueArguments.getOrNull(1)?.let {
+                    appendln("// ${it.extractStringArgumentValue()}")
+                    // TODO: append same amount of whitespace indentation
+                }
                 append("println(\"")
                 append(argument.text)
                 append(" is \${")
                 append(argument.text)
-                append("}\") // true")
+                append("}\") // $expectedResult")
             }
         }
 
@@ -77,7 +81,8 @@ open class KotlinWebsiteSampleProcessingService
         override fun visitCallExpression(expression: KtCallExpression) {
             when (expression.calleeExpression?.text) {
                 "assertPrints" -> convertAssertPrints(expression)
-                "assertTrue" -> convertAssertTrue(expression)
+                "assertTrue" -> convertAssertTrueFalse(expression, expectedResult = true)
+                "assertFalse" -> convertAssertTrueFalse(expression, expectedResult = false)
                 "assertFails" -> convertAssertFails(expression)
                 "assertFailsWith" -> convertAssertFailsWith(expression)
                 else -> super.visitCallExpression(expression)
-- 
cgit