Merge pull request #37631 from hamishknight/but-wait-its-gone

Author: hamishknight

test/refactoring/ExtractRepeat/Outputs/await/one.swift.expected

@@ -0,0 +1,13 @@
+func myAsync() async -> Int { 0 }
+func myFoo() -> Int { 0 }
+
+func testExtract() async -> Int {
+  let new_name = myFoo()
+let a = new_name
+  let b = new_name
+  let x = await myAsync()
+  let y = await myAsync()
+  return a + b + x + y
+}
+
+// rdar://72199992

test/refactoring/ExtractRepeat/Outputs/await/two.swift.expected

@@ -0,0 +1,13 @@
+func myAsync() async -> Int { 0 }
+func myFoo() -> Int { 0 }
+
+func testExtract() async -> Int {
+  let a = myFoo()
+  let b = myFoo()
+  let new_name = await myAsync()
+let x = new_name
+  let y = new_name
+  return a + b + x + y
+}
+
+// rdar://72199992

test/refactoring/ExtractRepeat/await.swift

@@ -0,0 +1,17 @@
+func myAsync() async -> Int { 0 }
+func myFoo() -> Int { 0 }
+
+func testExtract() async -> Int {
+  let a = myFoo()
+  let b = myFoo()
+  let x = await myAsync()
+  let y = await myAsync()
+  return a + b + x + y
+}
+
+// rdar://72199992
+// RUN: %empty-directory(%t.result)
+// RUN: %refactor -extract-repeat -source-filename %s -pos=5:11 -end-pos=5:18 >> %t.result/one.swift
+// RUN: diff -u %S/Outputs/await/one.swift.expected %t.result/one.swift
+// RUN: %refactor -extract-repeat -source-filename %s -pos=7:11 -end-pos=7:26 >> %t.result/two.swift
+// RUN: diff -u %S/Outputs/await/two.swift.expected %t.result/two.swift