groovy typecast surrounder shouldn't move caret to the start of the file

plugins/groovy/src/org/jetbrains/plugins/groovy/lang/surroundWith/surrounders/GroovyManyStatementsSurrounder.java

@@ -79,18 +79,16 @@ public abstract class GroovyManyStatementsSurrounder implements Surrounder {
         if (parentNode != element.getParent().getNode()) return null;
 
         final int endOffset = element.getTextRange().getEndOffset();
-        final PsiElement semicolon = PsiTreeUtil.findElementOfClassAtRange(element.getContainingFile(), endOffset, endOffset + 1, PsiElement.class);
+        final PsiElement semicolon = PsiTreeUtil.findElementOfClassAtOffset(element.getContainingFile(), endOffset, PsiElement.class, false);
         if (semicolon != null && ";".equals(semicolon.getText())) {
           assert parentNode == semicolon.getParent().getNode();
           parentNode.removeChild(semicolon.getNode());
         }
 
-        if (i < elements.length - 1) {
-          final PsiElement newLine = PsiTreeUtil.findElementOfClassAtRange(element.getContainingFile(), endOffset, elements[i + 1].getTextRange().getStartOffset(), PsiElement.class);
-          if (newLine != null && GroovyElementTypes.mNLS.equals(newLine.getNode().getElementType())) {
-            assert parentNode == newLine.getParent().getNode();
-            parentNode.removeChild(newLine.getNode());
-          }
+        final PsiElement newLine = PsiTreeUtil.findElementOfClassAtOffset(element.getContainingFile(), endOffset, PsiElement.class, false);
+        if (newLine != null && GroovyElementTypes.mNLS.equals(newLine.getNode().getElementType())) {
+          assert parentNode == newLine.getParent().getNode();
+          parentNode.removeChild(newLine.getNode());
         }
 
         parentNode.removeChild(element.getNode());

plugins/groovy/src/org/jetbrains/plugins/groovy/lang/surroundWith/surrounders/surroundersImpl/expressions/GroovyWithTypeCastSurrounder.java

@@ -32,7 +32,7 @@ public class GroovyWithTypeCastSurrounder extends GroovyExpressionSurrounder {
     GrTypeCastExpression typeCast = (GrTypeCastExpression) parenthesized.getOperand();
     replaceToOldExpression(typeCast.getOperand(), expression);
     GrTypeElement typeElement = typeCast.getCastTypeElement();
-    int endOffset = typeElement.getTextRange().getStartOffset();
+    int endOffset = typeElement.getTextRange().getStartOffset() + expression.getTextRange().getStartOffset();
     parenthesized = (GrParenthesizedExpression) expression.replaceWithExpression(parenthesized, false);
 
     final GrTypeCastExpression newTypeCast = (GrTypeCastExpression)parenthesized.getOperand();

plugins/groovy/test/org/jetbrains/plugins/groovy/lang/surroundWith/SurroundExpressionTest.java

@@ -18,6 +18,7 @@ public class SurroundExpressionTest extends SurroundTestCase {
   public void testIf1() throws Exception { doTest(new GroovyWithIfExprSurrounder()); }
   public void testIf_else1() throws Exception { doTest(new GroovyWithIfElseExprSurrounder()); }
   public void testType_cast1() throws Exception { doTest(new GroovyWithTypeCastSurrounder()); }
+  public void testType_cast2() throws Exception { doTest(new GroovyWithTypeCastSurrounder()); }
   public void testWhile1() throws Exception { doTest(new GroovyWithWhileExprSurrounder()); }
   public void testWith2() throws Exception { doTest(new GroovyWithWithExprSurrounder()); }
 

plugins/groovy/test/org/jetbrains/plugins/groovy/lang/surroundWith/SurroundTestCase.java

@@ -25,6 +25,6 @@ public abstract class SurroundTestCase extends LightGroovyTestCase {
       }
     });
 
-    assertEquals(data.get(1), myFixture.getFile().getText().trim());
+    myFixture.checkResult(data.get(1));
   }
 }

plugins/groovy/testdata/groovy/surround/expr/type_cast2.test

@@ -0,0 +1,7 @@
+println foo
+
+<selection>expr</selection>
+-----
+println foo
+
+((<caret>) expr)

plugins/groovy/testdata/groovy/surround/statements/shouldFailWithType.test

@@ -1,7 +1,9 @@
 <selection>a - b
 a - b</selection>
+println foo
 -----
 shouldFail() {
     a - b
     a - b
-}
+}
+println foo

plugins/groovy/testdata/groovy/surround/statements/with2.test

@@ -1,7 +1,9 @@
 <selection>a - b
 a - b</selection>
+println foo
 -----
 with() {
     a - b
     a - b
-}
+}
+println foo