This test exercises function and method extraction.

-- flags --
-ignore_extra_diags

-- basic.go --
package extract

//@codeaction(A_XLessThanYP, "refactor.extract.method", edit=meth1)
//@codeaction(A_XLessThanYP, "refactor.extract.function", edit=func1)
//@codeaction(A_AddP1, "refactor.extract.method", edit=meth2)
//@codeaction(A_AddP1, "refactor.extract.function", edit=func2)
//@codeaction(A_AddP2, "refactor.extract.method", edit=meth3)
//@codeaction(A_AddP2, "refactor.extract.function", edit=func3)
//@codeaction(A_XLessThanY, "refactor.extract.method", edit=meth4)
//@codeaction(A_XLessThanY, "refactor.extract.function", edit=func4)
//@codeaction(A_Add1, "refactor.extract.method", edit=meth5)
//@codeaction(A_Add1, "refactor.extract.function", edit=func5)
//@codeaction(A_Add2, "refactor.extract.method", edit=meth6)
//@codeaction(A_Add2, "refactor.extract.function", edit=func6)

type A struct {
	x int
	y int
}

func (a *A) XLessThanYP() bool {
	return a.x < a.y //@loc(A_XLessThanYP, re`return.*a\.y`)
}

func (a *A) AddP() int {
	sum := a.x + a.y //@loc(A_AddP1, re`sum.*a\.y`)
	return sum       //@loc(A_AddP2, re`return.*?sum`)
}

func (a A) XLessThanY() bool {
	return a.x < a.y //@loc(A_XLessThanY, re`return.*a\.y`)
}

func (a A) Add() int {
	sum := a.x + a.y //@loc(A_Add1, re`sum.*a\.y`)
	return sum       //@loc(A_Add2, re`return.*?sum`)
}

-- @func1/basic.go --
@@ -22 +22 @@
-	return a.x < a.y //@loc(A_XLessThanYP, re`return.*a\.y`)
+	return newFunction(a) //@loc(A_XLessThanYP, re`return.*a\.y`)
@@ -25 +25,4 @@
+func newFunction(a *A) bool {
+	return a.x < a.y
+}
+
-- @func2/basic.go --
@@ -26 +26 @@
-	sum := a.x + a.y //@loc(A_AddP1, re`sum.*a\.y`)
+	sum := newFunction(a) //@loc(A_AddP1, re`sum.*a\.y`)
@@ -30 +30,5 @@
+func newFunction(a *A) int {
+	sum := a.x + a.y
+	return sum
+}
+
-- @func3/basic.go --
@@ -27 +27 @@
-	return sum       //@loc(A_AddP2, re`return.*?sum`)
+	return newFunction(sum)       //@loc(A_AddP2, re`return.*?sum`)
@@ -30 +30,4 @@
+func newFunction(sum int) int {
+	return sum
+}
+
-- @func4/basic.go --
@@ -31 +31 @@
-	return a.x < a.y //@loc(A_XLessThanY, re`return.*a\.y`)
+	return newFunction(a) //@loc(A_XLessThanY, re`return.*a\.y`)
@@ -34 +34,4 @@
+func newFunction(a A) bool {
+	return a.x < a.y
+}
+
-- @func5/basic.go --
@@ -35 +35 @@
-	sum := a.x + a.y //@loc(A_Add1, re`sum.*a\.y`)
+	sum := newFunction(a) //@loc(A_Add1, re`sum.*a\.y`)
@@ -39 +39,5 @@
+func newFunction(a A) int {
+	sum := a.x + a.y
+	return sum
+}
+
-- @func6/basic.go --
@@ -36 +36 @@
-	return sum       //@loc(A_Add2, re`return.*?sum`)
+	return newFunction(sum)       //@loc(A_Add2, re`return.*?sum`)
@@ -39 +39,4 @@
+func newFunction(sum int) int {
+	return sum
+}
+
-- @meth1/basic.go --
@@ -22 +22 @@
-	return a.x < a.y //@loc(A_XLessThanYP, re`return.*a\.y`)
+	return a.newMethod() //@loc(A_XLessThanYP, re`return.*a\.y`)
@@ -25 +25,4 @@
+func (a *A) newMethod() bool {
+	return a.x < a.y
+}
+
-- @meth2/basic.go --
@@ -26 +26 @@
-	sum := a.x + a.y //@loc(A_AddP1, re`sum.*a\.y`)
+	sum := a.newMethod() //@loc(A_AddP1, re`sum.*a\.y`)
@@ -30 +30,5 @@
+func (a *A) newMethod() int {
+	sum := a.x + a.y
+	return sum
+}
+
-- @meth3/basic.go --
@@ -27 +27 @@
-	return sum       //@loc(A_AddP2, re`return.*?sum`)
+	return a.newMethod(sum)       //@loc(A_AddP2, re`return.*?sum`)
@@ -30 +30,4 @@
+func (*A) newMethod(sum int) int {
+	return sum
+}
+
-- @meth4/basic.go --
@@ -31 +31 @@
-	return a.x < a.y //@loc(A_XLessThanY, re`return.*a\.y`)
+	return a.newMethod() //@loc(A_XLessThanY, re`return.*a\.y`)
@@ -34 +34,4 @@
+func (a A) newMethod() bool {
+	return a.x < a.y
+}
+
-- @meth5/basic.go --
@@ -35 +35 @@
-	sum := a.x + a.y //@loc(A_Add1, re`sum.*a\.y`)
+	sum := a.newMethod() //@loc(A_Add1, re`sum.*a\.y`)
@@ -39 +39,5 @@
+func (a A) newMethod() int {
+	sum := a.x + a.y
+	return sum
+}
+
-- @meth6/basic.go --
@@ -36 +36 @@
-	return sum       //@loc(A_Add2, re`return.*?sum`)
+	return a.newMethod(sum)       //@loc(A_Add2, re`return.*?sum`)
@@ -39 +39,4 @@
+func (A) newMethod(sum int) int {
+	return sum
+}
+
-- context.go --
package extract

import (
	"context"
	"testing"
)

//@codeaction(B_AddP, "refactor.extract.method", edit=contextMeth1)
//@codeaction(B_AddP, "refactor.extract.function", edit=contextFunc1)
//@codeaction(B_LongList, "refactor.extract.method", edit=contextMeth2)
//@codeaction(B_LongList, "refactor.extract.function", edit=contextFunc2)
//@codeaction(B_AddPWithB, "refactor.extract.function", edit=contextFuncB)
//@codeaction(B_LongListWithT, "refactor.extract.function", edit=contextFuncT)

type B struct {
	x int
	y int
}
	
func (b *B) AddP(ctx context.Context) (int, error) {
	sum := b.x + b.y
	return sum, ctx.Err() //@loc(B_AddP, re`return.*ctx\.Err\(\)`)
}

func (b *B) LongList(ctx context.Context) (int, error) {
	p1 := 1
	p2 := 1
	p3 := 1
	return p1 + p2 + p3, ctx.Err() //@loc(B_LongList, re`return.*ctx\.Err\(\)`)
}

func (b *B) AddPWithB(ctx context.Context, tB *testing.B) (int, error) {
	sum := b.x + b.y //@loc(B_AddPWithB, re`(?s:^.*?Err\(\))`)
	tB.Skip()
	return sum, ctx.Err()
}

func (b *B) LongListWithT(ctx context.Context, t *testing.T) (int, error) {
	p1 := 1
	p2 := 1
	p3 := 1
	p4 := p1 + p2 //@loc(B_LongListWithT, re`(?s:^.*?Err\(\))`)
	t.Skip()
	return p4 + p3, ctx.Err()
}
-- @contextMeth1/context.go --
@@ -22 +22 @@
-	return sum, ctx.Err() //@loc(B_AddP, re`return.*ctx\.Err\(\)`)
+	return b.newMethod(ctx, sum) //@loc(B_AddP, re`return.*ctx\.Err\(\)`)
@@ -25 +25,4 @@
+func (*B) newMethod(ctx context.Context, sum int) (int, error) {
+	return sum, ctx.Err()
+}
+
-- @contextMeth2/context.go --
@@ -29 +29 @@
-	return p1 + p2 + p3, ctx.Err() //@loc(B_LongList, re`return.*ctx\.Err\(\)`)
+	return b.newMethod(ctx, p1, p2, p3) //@loc(B_LongList, re`return.*ctx\.Err\(\)`)
@@ -32 +32,4 @@
+func (*B) newMethod(ctx context.Context, p1 int, p2 int, p3 int) (int, error) {
+	return p1 + p2 + p3, ctx.Err()
+}
+
-- @contextFunc2/context.go --
@@ -29 +29 @@
-	return p1 + p2 + p3, ctx.Err() //@loc(B_LongList, re`return.*ctx\.Err\(\)`)
+	return newFunction(ctx, p1, p2, p3) //@loc(B_LongList, re`return.*ctx\.Err\(\)`)
@@ -32 +32,4 @@
+func newFunction(ctx context.Context, p1 int, p2 int, p3 int) (int, error) {
+	return p1 + p2 + p3, ctx.Err()
+}
+
-- @contextFunc1/context.go --
@@ -22 +22 @@
-	return sum, ctx.Err() //@loc(B_AddP, re`return.*ctx\.Err\(\)`)
+	return newFunction(ctx, sum) //@loc(B_AddP, re`return.*ctx\.Err\(\)`)
@@ -25 +25,4 @@
+func newFunction(ctx context.Context, sum int) (int, error) {
+	return sum, ctx.Err()
+}
+
-- @contextFuncB/context.go --
@@ -33 +33,4 @@
+	return newFunction(ctx, tB, b)
+}
+
+func newFunction(ctx context.Context, tB *testing.B, b *B) (int, error) {
-- @contextFuncT/context.go --
@@ -42 +42,4 @@
+	return newFunction(ctx, t, p1, p2, p3)
+}
+
+func newFunction(ctx context.Context, t *testing.T, p1 int, p2 int, p3 int) (int, error) {
