Fixed lazy var bug

nicklockwood · Mar 14, 2017 · b96b0dd · b96b0dd
1 parent 2d10536
commit b96b0dd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Sources/Rules.swift b/Sources/Rules.swift
@@ -1808,6 +1808,7 @@ extension FormatRules {
                         }
                         index = nextIndex
                     }
+                    continue
                 case let .keyword(name):
                     lastKeyword = name
                 case .startOfScope("("):

diff --git a/Tests/RulesTests.swift b/Tests/RulesTests.swift
@@ -3654,6 +3654,13 @@ class RulesTests: XCTestCase {
         XCTAssertEqual(try format(input + "\n", rules: FormatRules.default), output + "\n")
     }
 
+    func testNoRemoveSelfFromLazyVarClosure2() {
+        let input = "lazy var foo = { let bar = self.baz }()"
+        let output = "lazy var foo = { let bar = self.baz }()"
+        XCTAssertEqual(try format(input, rules: [FormatRules.redundantSelf]), output)
+        XCTAssertEqual(try format(input + "\n", rules: FormatRules.default), output + "\n")
+    }
+
     // MARK: unusedArguments
 
     // closures