also check mods for ExtensionMethod

bishabosha · bishabosha · commit bc9438204f64 · 2020-09-18T23:20:32.000+02:00
diff --git a/compiler/src/dotty/tools/dotc/ast/Trees.scala b/compiler/src/dotty/tools/dotc/ast/Trees.scala
@@ -344,7 +344,7 @@ object Trees {
         else {
           val realName = name.stripModuleClassSuffix.lastPart
           var length = realName.length
-          if symbol.is(ExtensionMethod) && name.isExtensionName then
+          if (mods.is(ExtensionMethod) || symbol.is(ExtensionMethod)) && name.isExtensionName then
             length -= "extension_".length
           Span(point, point + length, point)
         }

Original file line number	Diff line number	Diff line change
`@@ -344,7 +344,7 @@ object Trees {`
`344`	`344`	`else {`
`345`	`345`	`val realName = name.stripModuleClassSuffix.lastPart`
`346`	`346`	`var length = realName.length`
`347`		`- if symbol.is(ExtensionMethod) && name.isExtensionName then`
	`347`	`+ if (mods.is(ExtensionMethod) \|\| symbol.is(ExtensionMethod)) && name.isExtensionName then`
`348`	`348`	`length -= "extension_".length`
`349`	`349`	`Span(point, point + length, point)`
`350`	`350`	`}`