Factor out expectedResultImpl(for:)

Author: hamishknight

SourceKitStressTester/Sources/StressTester/ActionGenerators.swift

@@ -476,6 +476,31 @@ private struct ActionToken {
       }
   }
 
+  static private func expectedResultImpl(for token: TokenSyntax, parent: Syntax) -> ExpectedResult {
+    if let parent = parent.as(DeclReferenceExprSyntax.self), parent.baseName == token {
+      if let refArgs = parent.argumentNames {
+        let name = SwiftName(base: token.text, labels: refArgs.arguments.map{ $0.name.text })
+        return ExpectedResult(name: name, kind: .reference)
+      }
+      if let (kind, callArgs) = getParentArgs(of: parent) {
+        let name = SwiftName(base: token.text, labels: callArgs)
+        return ExpectedResult(name: name, kind: kind)
+      }
+      if let parentsParent = parent.parent?.as(MemberAccessExprSyntax.self), parentsParent.declName.baseName == token {
+        if let refArgs = parentsParent.declName.argumentNames {
+          let name = SwiftName(base: token.text, labels: refArgs.arguments.map{ $0.name.text })
+          return ExpectedResult(name: name, kind: .reference)
+        }
+        if let (kind, callArgs) = getParentArgs(of: parentsParent) {
+          let name = SwiftName(base: token.text, labels: callArgs)
+          return ExpectedResult(name: name, kind: kind)
+        }
+      }
+    }
+    let name = SwiftName(base: token.text, labels: [])
+    return ExpectedResult(name: name, kind: .reference)
+  }
+
   /// A nil result means the results shouldn't be checked, e.g. if the provided token corresponds to the
   /// name of a newly declared variable.
   static private func expectedResult(for token: TokenSyntax) -> ExpectedResult? {
@@ -503,28 +528,7 @@ private struct ActionToken {
         return nil
       }
     }
-    if let parent = parent.as(DeclReferenceExprSyntax.self), parent.baseName == token {
-      if let refArgs = parent.argumentNames {
-        let name = SwiftName(base: token.text, labels: refArgs.arguments.map{ $0.name.text })
-        return ExpectedResult(name: name, kind: .reference)
-      }
-      if let (kind, callArgs) = getParentArgs(of: parent) {
-        let name = SwiftName(base: token.text, labels: callArgs)
-        return ExpectedResult(name: name, kind: kind)
-      }
-      if let parentsParent = parent.parent?.as(MemberAccessExprSyntax.self), parentsParent.declName.baseName == token {
-        if let refArgs = parentsParent.declName.argumentNames {
-          let name = SwiftName(base: token.text, labels: refArgs.arguments.map{ $0.name.text })
-          return ExpectedResult(name: name, kind: .reference)
-        }
-        if let (kind, callArgs) = getParentArgs(of: parentsParent) {
-          let name = SwiftName(base: token.text, labels: callArgs)
-          return ExpectedResult(name: name, kind: kind)
-        }
-      }
-    }
-    let name = SwiftName(base: token.text, labels: [])
-    return ExpectedResult(name: name, kind: .reference)
+    return expectedResultImpl(for: token, parent: parent)
   }
 
   static func getParentArgs<T:SyntaxProtocol>(of syntax: T) -> (kind: ExpectedResult.Kind, args: [String])? {