/*******************************************************************************
 * Copyright (c) 2000, 2009 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.ide.ext.java.client.core.rewrite;

import org.eclipse.che.ide.ext.java.jdt.core.dom.AST;
import org.eclipse.che.ide.ext.java.jdt.core.dom.ASTNode;
import org.eclipse.che.ide.ext.java.jdt.core.dom.Block;
import org.eclipse.che.ide.ext.java.jdt.core.dom.CompilationUnit;
import org.eclipse.che.ide.ext.java.jdt.core.dom.ExpressionStatement;
import org.eclipse.che.ide.ext.java.jdt.core.dom.MethodDeclaration;
import org.eclipse.che.ide.ext.java.jdt.core.dom.MethodInvocation;
import org.eclipse.che.ide.ext.java.jdt.core.dom.ReturnStatement;
import org.eclipse.che.ide.ext.java.jdt.core.dom.Statement;
import org.eclipse.che.ide.ext.java.jdt.core.dom.TypeDeclaration;
import org.eclipse.che.ide.ext.java.jdt.core.dom.rewrite.ASTRewrite;
import org.eclipse.che.ide.ext.java.jdt.internal.compiler.env.ICompilationUnit;

import org.junit.Test;

public class ASTRewritingGroupNodeTest extends ASTRewritingTest {

    @Test
    public void testCollapsedTargetNodes() throws Exception {
        StringBuffer buf = new StringBuffer();
        buf.append("package test1;\n");
        buf.append("public class E {\n");
        buf.append("    public void foo(Object o) {\n");
        buf.append("        return;\n");
        buf.append("    }\n");
        buf.append("}\n");
        ICompilationUnit cu =
                new org.eclipse.che.ide.ext.java.jdt.compiler.batch.CompilationUnit(buf.toString().toCharArray(), "E.java", "");

        CompilationUnit astRoot = createAST(cu);
        ASTRewrite rewrite = ASTRewrite.create(astRoot.getAST());

        AST ast = astRoot.getAST();

        TypeDeclaration type = findTypeDeclaration(astRoot, "E");
        MethodDeclaration methodDecl = findMethodDeclaration(type, "foo");

        ReturnStatement returnStatement = (ReturnStatement)methodDecl.getBody().statements().get(0);

        MethodInvocation newMethodInv1 = ast.newMethodInvocation();
        newMethodInv1.setName(ast.newSimpleName("foo1"));
        ExpressionStatement st1 = ast.newExpressionStatement(newMethodInv1);

        MethodInvocation newMethodInv2 = ast.newMethodInvocation();
        newMethodInv2.setName(ast.newSimpleName("foo2"));
        ExpressionStatement st2 = ast.newExpressionStatement(newMethodInv2);

        ASTNode placeholder = rewrite.createGroupNode(new Statement[]{st1, st2});
        rewrite.replace(returnStatement, placeholder, null);

        String preview = evaluateRewrite(cu, rewrite);

        buf = new StringBuffer();
        buf.append("package test1;\n");
        buf.append("public class E {\n");
        buf.append("    public void foo(Object o) {\n");
        buf.append("        foo1();\n");
        buf.append("        foo2();\n");
        buf.append("    }\n");
        buf.append("}\n");
        String expected = buf.toString();

        assertEqualString(preview, expected);
    }

    @Test
    public void testCollapsedTargetNodes2() throws Exception {
        StringBuffer buf = new StringBuffer();
        buf.append("package test1;\n");
        buf.append("public class E {\n");
        buf.append("    public void foo(Object o) {\n");
        buf.append("        {\n");
        buf.append("            return;\n");
        buf.append("        }\n");
        buf.append("    }\n");
        buf.append("}\n");
        ICompilationUnit cu =
                new org.eclipse.che.ide.ext.java.jdt.compiler.batch.CompilationUnit(buf.toString().toCharArray(), "E.java", "");

        CompilationUnit astRoot = createAST(cu);
        ASTRewrite rewrite = ASTRewrite.create(astRoot.getAST());

        AST ast = astRoot.getAST();

        TypeDeclaration type = findTypeDeclaration(astRoot, "E");
        MethodDeclaration methodDecl = findMethodDeclaration(type, "foo");

        Statement statement = (Statement)methodDecl.getBody().statements().get(0);

        MethodInvocation newMethodInv1 = ast.newMethodInvocation();
        newMethodInv1.setName(ast.newSimpleName("foo1"));
        ExpressionStatement st1 = ast.newExpressionStatement(newMethodInv1);

        MethodInvocation newMethodInv2 = ast.newMethodInvocation();
        newMethodInv2.setName(ast.newSimpleName("foo2"));
        ExpressionStatement st2 = ast.newExpressionStatement(newMethodInv2);

        ReturnStatement st3 = (ReturnStatement)rewrite.createCopyTarget((ASTNode)((Block)statement).statements().get(0));

        ASTNode placeholder = rewrite.createGroupNode(new Statement[]{st1, st2, st3});
        rewrite.replace(statement, placeholder, null);

        String preview = evaluateRewrite(cu, rewrite);

        buf = new StringBuffer();
        buf.append("package test1;\n");
        buf.append("public class E {\n");
        buf.append("    public void foo(Object o) {\n");
        buf.append("        foo1();\n");
        buf.append("        foo2();\n");
        buf.append("        return;\n");
        buf.append("    }\n");
        buf.append("}\n");
        String expected = buf.toString();

        assertEqualString(preview, expected);
    }

}