parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/parser/rfc7950/ir/AntlrSupportTest.java

   1 /*
   2  * Copyright (c) 2018 Pantheon Technologies, s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.yang.parser.rfc7950.ir;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.opendaylight.yangtools.yang.parser.rfc7950.ir.AntlrSupport.trimWhitespace;
  12
  13 import org.junit.Test;
  14
  15 public class AntlrSupportTest {
  16     @Test
  17     public void testTrimWhitespace() {
  18         assertEquals("\n", trimWhitespace("\n", 0));
  19         assertEquals("\n", trimWhitespace("\n", 5));
  20         assertEquals("\n\n\n\n", trimWhitespace("\n\n\n\n", 0));
  21         assertEquals("\n\n\n\n", trimWhitespace("\n\n\n\n", 5));
  22         assertEquals("abc\n\n", trimWhitespace("abc \n  \n", 0));
  23         assertEquals("abc\n\n", trimWhitespace("abc \n  \n", 1));
  24         assertEquals("abc\n  ", trimWhitespace("abc\n   ", 0));
  25         assertEquals("abc\n", trimWhitespace("abc\n   ", 2));
  26         assertEquals("abc\n\n", trimWhitespace("abc\n   \n", 2));
  27         assertEquals("abc\n        ", trimWhitespace("abc\n\t ", 0));
  28         assertEquals("abc\n      ", trimWhitespace("abc\n\t ", 2));
  29         assertEquals("abc\n    ", trimWhitespace("abc\n\t ", 4));
  30         assertEquals("abc\n    ", trimWhitespace("abc\n \t", 4));
  31         assertEquals("abc\n   a\n    a\n", trimWhitespace("abc\n\ta\n\t a\n", 4));
  32         assertEquals("abc\n\n    a\n", trimWhitespace("abc\n\t\n\t a\n", 4));
  33         assertEquals("   \ta\n", trimWhitespace("   \ta\n", 3));
  34         assertEquals("   \ta\n", trimWhitespace("   \ta\n  ", 3));
  35         assertEquals("   \ta\n", trimWhitespace("   \ta\n   ", 3));
  36         assertEquals("   \ta\n", trimWhitespace("   \ta\n    ", 3));
  37         assertEquals("   \ta\n ", trimWhitespace("   \ta\n     ", 3));
  38     }
  39 }