package net.sourceforge.jenesis4java.util;
   
   /*
    * #%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.impl.MCodeWriter;
  import org.junit.Test;
  
  import java.io.PrintWriter;
  import java.io.StringWriter;
  import java.util.ArrayList;
  import java.util.List;
  
  import static net.sourceforge.jenesis4java.impl.CommentsTest.MultiLineComment.DEFAULT_MULTILINE_COMMENT;
  import static net.sourceforge.jenesis4java.util.CodeGenerationAssertions.assertThatTrailingCommentIsOnFirstLine;
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertSame;
  
  public abstract class StatementTestBase {
  
      public abstract Statement getInstance();
  
      @Test
      public void testSingleLineComment() {
          Statement statement = getInstance();
  
          String withoutComment = statement.toString();
          statement.setComment(Comment.SINGLE_LINE, "comment");
          String withComment = statement.toString();
  
          String message = "Applying single-line comment to " + statement.getClass() //
                  + " but got unexpected result.";
          assertEquals(message, "// comment\n" + withoutComment, withComment);
      }
  
      @Test
      public void testMultiLineComment() {
          Statement statement = getInstance();
  
          String withoutComment = statement.toString();
          statement.setComment(Comment.MULTI_LINE, "comment");
          String withComment = statement.toString();
  
          String message = "Applying multi-line comment to " + statement.getClass() //
                  + " but got unexpected result.";
          assertEquals(message, DEFAULT_MULTILINE_COMMENT + withoutComment, withComment);
      }
  
      @Test
      public void testTrailingCommentIsOnFirstLine() {
          assertThatTrailingCommentIsOnFirstLine(getInstance());
      }
  
      @Test
      public void testDefaultIsNoComment() {
          Statement statement = getInstance();
          assertEquals("Expecting no comment.", null, statement.getComment());
      }
  
      @Test
      public void testCommentIsTrailingComment() {
          Statement instance = getInstance();
          String withoutComment = instance.toString();
  
          Comment comment = instance.comment("comment");
          assertEquals(Comment.SINGLE_LINE, comment.getType());
          assertEquals("// comment\n" + withoutComment, instance.toString());
      }
  
      @Test
      public void testSetAndGetComment() {
          Statement statement = getInstance();
          statement.setComment(Comment.MULTI_LINE, "comment");
  
          Comment comment = statement.getComment();
          assertEquals(Comment.MULTI_LINE, comment.getType());
          assertEquals("comment", comment.getText());
      }
  
     @Test
     public void testToCodeReturnsGivenCodeWriter() {
         Statement statement = getInstance();
 
         CodeWriter cout = new MCodeWriter(new PrintWriter(new StringWriter()));
         assertSame(cout, statement.toCode(cout));
     }
 
     @Test
     public void testVisitComment() {
         final String commentMsg = "specific comment";
         Statement statement = getInstance();
         statement.setComment(Comment.SINGLE_LINE, commentMsg);
 
         final List<Comment> comments = new ArrayList<>();
 
         statement.visit(new ConditionalVisitor() {
 
             @Override
             public boolean visit(Codeable current, Codeable parent) {
                 if (current instanceof Comment) {
                     comments.add((Comment) current);
                 }
                 return true;
             }
         });
 
         String message = "Expected to visit 1 comment in \n" + statement + "\n, instead got: " + comments.size();
         assertEquals(message,  1, comments.size());
     }
 
     @Test
     public void testSetLabelIsChainable() {
         Statement statement = getInstance();
         Statement returnedStatement = statement.setLabel("label");
         assertSame(statement, returnedStatement);
     }
 }