BlockStructureBraces assertion failed (#461)

* BlockStructureBraces assertion failed

### What's done:
    Fixed bug and added tests

diktat-rules/src/main/kotlin/org/cqfn/diktat/ruleset/rules/BlockStructureBraces.kt

@@ -1,6 +1,5 @@
 package org.cqfn.diktat.ruleset.rules
 
-import com.pinterest.ktlint.core.KtLint
 import com.pinterest.ktlint.core.Rule
 import com.pinterest.ktlint.core.ast.ElementType.BLOCK
 import com.pinterest.ktlint.core.ast.ElementType.BODY
@@ -29,16 +28,11 @@ import com.pinterest.ktlint.core.ast.ElementType.WHILE
 import com.pinterest.ktlint.core.ast.ElementType.WHILE_KEYWORD
 import com.pinterest.ktlint.core.ast.ElementType.WHITE_SPACE
 import com.pinterest.ktlint.core.ast.isWhiteSpaceWithNewline
-import java.lang.Exception
 import org.cqfn.diktat.common.config.rules.RuleConfiguration
 import org.cqfn.diktat.common.config.rules.RulesConfig
 import org.cqfn.diktat.common.config.rules.getRuleConfig
 import org.cqfn.diktat.ruleset.constants.Warnings.BRACES_BLOCK_STRUCTURE_ERROR
-import org.cqfn.diktat.ruleset.utils.emptyBlockList
-import org.cqfn.diktat.ruleset.utils.findAllNodesWithSpecificType
-import org.cqfn.diktat.ruleset.utils.findLBrace
-import org.cqfn.diktat.ruleset.utils.hasChildOfType
-import org.cqfn.diktat.ruleset.utils.isBlockEmpty
+import org.cqfn.diktat.ruleset.utils.*
 import org.jetbrains.kotlin.com.intellij.lang.ASTNode
 import org.jetbrains.kotlin.com.intellij.psi.impl.source.tree.LeafPsiElement
 import org.jetbrains.kotlin.com.intellij.psi.impl.source.tree.PsiWhiteSpaceImpl
@@ -200,7 +194,7 @@ class BlockStructureBraces(private val configRules: List<RulesConfig>) : Rule("b
             BRACES_BLOCK_STRUCTURE_ERROR.warnAndFix(configRules, emitWarn, isFixMode, "incorrect same line after opening brace",
                     newNode.startOffset, newNode) {
                 if (newNode.elementType != WHITE_SPACE) {
-                    node.addChild(PsiWhiteSpaceImpl("\n"), newNode)
+                    newNode.treeParent.addChild(PsiWhiteSpaceImpl("\n"), newNode)
                 } else {
                     (newNode as LeafPsiElement).replaceWithText("\n")
                 }

diktat-rules/src/test/kotlin/org/cqfn/diktat/ruleset/smoke/DiktatSmokeTest.kt

@@ -76,4 +76,10 @@ class DiktatSmokeTest : FixTestBase("test/smoke/src/main/kotlin",
     fun `regression - shouldn't throw exception in cases similar to #371`() {
         fixAndCompare("Bug1Expected.kt", "Bug1Test.kt")
     }
+
+    @Test
+    @Tag("DiktatRuleSetProvider")
+    fun `smoke test #3`() {
+        fixAndCompare("Example3Expected.kt", "Example3Test.kt")
+    }
 }

diktat-rules/src/test/resources/test/paragraph3/block_brace/ClassBracesExpected.kt

@@ -13,3 +13,7 @@ class C{
         private val id = 11
 }
 }
+
+class D {
+val x = 0
+}

diktat-rules/src/test/resources/test/paragraph3/block_brace/ClassBracesTest.kt

@@ -11,3 +11,5 @@ class C{
     companion object{
         private val id = 11}
 }
+
+class D {val x = 0}

diktat-rules/src/test/resources/test/smoke/src/main/kotlin/Bug1Expected.kt

@@ -7,3 +7,7 @@ fun readFile(foo: Foo) {
     var bar: Bar
 }
 
+class D {
+    val x = 0
+}
+

diktat-rules/src/test/resources/test/smoke/src/main/kotlin/Bug1Test.kt

@@ -2,4 +2,6 @@ package test.smoke.src.main.kotlin
 
 fun readFile(foo: Foo) {
     var bar: Bar
-}
+}
+
+class D {val x = 0}

diktat-rules/src/test/resources/test/smoke/src/main/kotlin/Example3Expected.kt

@@ -0,0 +1,23 @@
+package org.cqfn.diktat
+
+/**
+ * @property name
+ */
+class HttpClient(var name: String) {
+    var url: String = ""
+    var port: String = ""
+    var timeout = 0
+
+    fun doRequest() {}
+}
+
+fun mains() {
+    val httpClient = HttpClient("myConnection")
+            .apply {
+                url = "http://pushkin.com"
+                port = "8080"
+                timeout = 100
+            }
+            .doRequest()
+}
+

diktat-rules/src/test/resources/test/smoke/src/main/kotlin/Example3Test.kt

@@ -0,0 +1,19 @@
+package test.smoke.src.main.kotlin
+
+class HttpClient(var name: String) {
+    var url: String = ""
+    var port: String = ""
+    var timeout = 0
+
+    fun doRequest() {}
+}
+
+fun mains() {
+    val httpClient = HttpClient("myConnection")
+            .apply {
+                url = "http://pushkin.com"
+                port = "8080"
+                timeout = 100
+            }
+            .doRequest()
+}