package net.sourceforge.jenesis4java.impl;
   
   /*
    * #%L
    * Jenesis 4 Java Code Generator
    * %%
    * Copyright (C) 2000 - 2016 jenesis4java
    * %%
    * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Lesser General Public License as
   * published by the Free Software Foundation, either version 3 of the
   * License, or (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Lesser Public License for more details.
   * 
   * You should have received a copy of the GNU General Lesser Public
   * License along with this program.  If not, see
   * <http://www.gnu.org/licenses/lgpl-3.0.html>.
   * #L%
   */
  
  import net.sourceforge.jenesis4java.*;
  import net.sourceforge.jenesis4java.util.InstanceReplacingVisitor;
  import net.sourceforge.jenesis4java.util.StatementTestBase;
  import net.sourceforge.jenesis4java.util.WithVmAndMethod;
  import org.junit.Before;
  import org.junit.Test;
  import org.junit.experimental.runners.Enclosed;
  import org.junit.runner.RunWith;
  
  import static java.util.Collections.emptyList;
  import static java.util.Collections.singletonList;
  import static net.sourceforge.jenesis4java.impl.CommentsTest.MultiLineComment.DEFAULT_MULTILINE_COMMENT;
  import static net.sourceforge.jenesis4java.util.Shortcuts.dummyClass;
  import static net.sourceforge.jenesis4java.util.TestHelper.*;
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertSame;
  
  @RunWith(Enclosed.class)
  public class IfTest {
  
      public static class IfStatement extends WithVmAndMethod {
  
          @Test
          public void simpleIf() {
              Expression trueExpr = vm.newTrue();
              If anIf = method.newIf(trueExpr);
              anIf.getElse(); // else is empty, so not printed
              assertEquals("if (true) {" + LS + "}", anIf.toString());
          }
  
          @Test
          public void simpleElseIf() {
              Expression falseExpr = vm.newFalse();
              If anIf = method.newIf(falseExpr);
              Expression trueExpr = vm.newTrue();
              anIf.newElseIf(trueExpr);
              assertEquals("if (false) {" + LS + "} else if (true) {" + LS + "}", anIf.toString());
          }
  
          @Test
          public void simpleElseIfElse() {
              Expression falseExpr = vm.newFalse();
              If anIf = method.newIf(falseExpr);
              Expression trueExpr = vm.newTrue();
              anIf.newElseIf(trueExpr);
              Statement statementInElse = anIf.getElse().newEmpty();
  
              String expected = "if (false) {" + LS //
                      + "} else if (true) {" + LS //
                      + "} else {" + LS //
                      + INDENT + statementInElse + LS //
                      + "}";
              assertEquals(expected, anIf.toString());
          }
      }
  
      public static class GettersAndSetters extends WithVmAndMethod {
  
          @Test
          public void getElseIf() {
              If anIf = method.newIf(vm.newFree("irrelevant"));
              assertEquals(emptyList(), anIf.getElseIfs());
  
              ElseIf elseIf = anIf.newElseIf(vm.newTrue());
              assertEquals(singletonList(elseIf), anIf.getElseIfs());
          }
  
          @Test
          public void predicate() {
              If anIf = method.newIf(vm.newFree("initial_prediacate"));
              Freeform replacingPredicate = vm.newFree("new_predicate");
              ConditionalStatement returnedIf = anIf.setPredicate(replacingPredicate);
              assertSame(anIf, returnedIf);
              assertEquals(replacingPredicate, anIf.getPredicate());
          }
     }
 
     public static class Visitor extends WithVmAndMethod {
 
         @Test
         public void visitIf() {
             Expression originalExpression = vm.newTrue();
             If anIf = method.newIf(originalExpression);
             Expression replacementExpression = vm.newFalse();
             anIf.visit(new InstanceReplacingVisitor(originalExpression, replacementExpression));
             assertEquals(replacementExpression, anIf.getPredicate());
         }
 
         @Test
         public void visitElseIf() {
             If anIf = method.newIf(vm.newFree("irrelevant"));
             Expression originalExpression = vm.newTrue();
             ElseIf elseIf = anIf.newElseIf(originalExpression);
             Expression replacementExpression = vm.newFalse();
 
             elseIf.visit(new InstanceReplacingVisitor(originalExpression, replacementExpression));
             assertEquals(replacementExpression, elseIf.getPredicate());
         }
 
         @Test
         public void removeElseIfFromIf() {
             If anIf = method.newIf(vm.newFree("irrelevant"));
             ElseIf elseIf1 = anIf.newElseIf(vm.newTrue());
             ElseIf elseIf2 = anIf.newElseIf(vm.newTrue());
 
             anIf.visit(new InstanceReplacingVisitor(elseIf1, null));
             assertEquals(singletonList(elseIf2), anIf.getElseIfs());
         }
     }
 
     public static class IfInheritedFunctionality extends StatementTestBase {
 
         @Override
         public Statement getInstance() {
             return dummyClass().newMethod("method").newIf(vm().newTrue());
         }
     }
 
     public static class ElseIfInheritedFunctionality extends StatementTestBase {
 
         @Override
         public Statement getInstance() {
             return dummyClass().newMethod("method").newIf(vm().newTrue()).newElseIf(vm().newFalse());
         }
     }
 
     public static class ElseInheritedFunctionality extends StatementTestBase {
 
         @Override
         public Statement getInstance() {
             If anIf = dummyClass().newMethod("method").newIf(vm().newFalse());
             anIf.getElse().newEmpty();
             return anIf.getElse();
         }
 
     }
 
     /**
      * Focus on comment-functionality not already covered by
      * InheritedFunctionality.
      */
     public static class Comments {
 
         private If ifStmt;
 
         @Before
         public void setUp() {
             this.ifStmt = dummyClass().newMethod("method").newIf(vm().newFalse());
         }
 
         @Test
         public void singleLineOnElseIfInbetweenBlocks() {
             String ifOnly = ifStmt.toString();
 
             ElseIf elseIf = ifStmt.newElseIf(vm().newTrue());
             String elseIfOnly = elseIf.toString();
             elseIf.setComment(Comment.SINGLE_LINE, "comment");
 
             String expectedIfCommentElseIf = ifOnly + LS + "// comment" + LS + elseIfOnly;
             assertEquals(expectedIfCommentElseIf, ifStmt.toString());
         }
 
         @Test
         public void multiLineOnElseIfInbetweenBlocks() {
             String ifOnly = ifStmt.toString();
 
             ElseIf elseIf = ifStmt.newElseIf(vm().newTrue());
             String elseIfOnly = elseIf.toString();
             elseIf.setComment(Comment.MULTI_LINE, "comment");
 
             String expectedIfCommentElseIf = ifOnly + LS + DEFAULT_MULTILINE_COMMENT + elseIfOnly;
             assertEquals(expectedIfCommentElseIf, ifStmt.toString());
         }
 
         @Test
         public void singleLineCommentInbetweenBlocks() {
             String ifOnly = ifStmt.toString();
             Else elsePart = ifStmt.getElse();
             elsePart.newEmpty();
             String elseIfOnly = elsePart.toString();
             elsePart.setComment(Comment.SINGLE_LINE, "comment");
 
             String expectedIfCommentElseIf = ifOnly + LS + "// comment" + LS + elseIfOnly;
             assertEquals(expectedIfCommentElseIf, ifStmt.toString());
         }
 
         @Test
         public void multiLineCommentInbetweenBlocks() {
             String ifOnly = ifStmt.toString();
 
             Else elsePart = ifStmt.getElse();
             elsePart.newEmpty();
             String elseIfOnly = elsePart.toString();
             elsePart.setComment(Comment.MULTI_LINE, "comment");
 
             String expectedIfCommentElseIf = ifOnly + LS + DEFAULT_MULTILINE_COMMENT + elseIfOnly;
             assertEquals(expectedIfCommentElseIf, ifStmt.toString());
         }
 
     }
 }