Teststate enhancement

CHANGELOG.md

@@ -2,6 +2,11 @@
 
 #### Breaking
 
+* Teststate enhancement.
+  [@aydinomer00](https://github.com/aydinomer00)  
+  [#5803](https://github.com/realm/SwiftLint/issues/5803)
+
+
 * The command plugin now requires write permissions so that it works with the `--fix` option without an error.  
   [SimplyDanny](https://github.com/SimplyDanny)
 
@@ -7958,3 +7963,6 @@ This release has seen a phenomenal uptake in community contributions!
 ## 0.1.0: Fresh Out Of The Dryer
 
 First Version!
+
+
+

Source/SwiftLintBuiltInRules/Rules/Lint/QuickDiscouragedCallRule.swift

@@ -47,6 +47,7 @@ struct QuickDiscouragedCallRule: OptInRule {
         guard
             kind == .call,
             let name = dictionary.name,
+            name != "@TestState", // Exception: Do not trigger a warning for @TestState
             let kindName = QuickCallKind(rawValue: name),
             QuickCallKind.restrictiveKinds.contains(kindName)
             else { return [] }
@@ -58,7 +59,7 @@ struct QuickDiscouragedCallRule: OptInRule {
                            reason: "Discouraged call inside a '\(name)' block")
         }
     }
-
+    
     private func violationOffsets(in substructure: [SourceKittenDictionary]) -> [ByteCount] {
         substructure.flatMap { dictionary -> [ByteCount] in
             let substructure = dictionary.substructure.flatMap { dict -> [SourceKittenDictionary] in

Source/SwiftLintBuiltInRules/Rules/Lint/QuickDiscouragedCallRuleExamples.swift

@@ -173,6 +173,13 @@ internal struct QuickDiscouragedCallRuleExamples {
            }
         }
         """),
+        Example("""
+            class TotoTests: QuickSpec {
+                override func spec() {
+                    @TestState var foo = Foo()
+                }
+            }
+            """),
     ]
 
     static let triggeringExamples: [Example] = [
@@ -324,5 +331,14 @@ internal struct QuickDiscouragedCallRuleExamples {
            }
         }
         """),
+        Example("""
+            class TotoTests: QuickSpec {
+                override func spec() {
+                    describe("foo") {
+                        @TestState var foo = Foo()
+                    }
+                }
+            }
+            """),
     ]
 }

Tests/SwiftLintFrameworkTests/PrivateOverFilePrivateRuleTests.swift

@@ -29,3 +29,4 @@ final class PrivateOverFilePrivateRuleTests: SwiftLintTestCase {
         verifyRule(description, ruleConfiguration: ["validate_extensions": false])
     }
 }
+