Implemented nested() method

decorators-squad · Feb 17, 2017 · baaf99d · baaf99d
1 parent 50fbb05
commit baaf99d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/main/java/com/amihaiemil/camel/RtYamlLines.java b/src/main/java/com/amihaiemil/camel/RtYamlLines.java
@@ -8,7 +8,7 @@
 /**
  * YamlLines default implementation.
  * @author Mihai Andronache (amihaiemil@gmail.com)
- * @version $Id: aa086c8b637f198e26bc2ab0550fff6763e30016 $
+ * @version $Id$
  * @since 1.0.0
  */
 final class RtYamlLines implements YamlLines {
@@ -54,19 +54,17 @@ public Iterator<YamlLine> iterator() {
     public YamlLines nested(final int after) {
         final List<YamlLine> nestedLines = new ArrayList<YamlLine>();
         YamlLine start = null;
-        int index = 0;
         for(final YamlLine line: this.lines) {
-            if(index == after) {
+            if(line.number() == after) {
                 start = line;
             }
-            if(index > after) {
+            if(line.number() > after) {
                 if(line.indentation() > start.indentation()) {
                     nestedLines.add(line);   
                 } else {
                     break;
                 }
             }
-            index++;
         }
         return new RtYamlLines(nestedLines);
     }

diff --git a/src/test/java/com/amihaiemil/camel/RtYamlLinesTest.java b/src/test/java/com/amihaiemil/camel/RtYamlLinesTest.java
@@ -38,7 +38,7 @@
 /**
  * Unit tests for {@link RtYamlLines}.
  * @author Mihai Andronache (amihaiemil@gmail.com)
- * @version $Id: 9009d046234339ba4ef3974ddf83b83c8134c3ad $
+ * @version $Id$
  * @sinve 1.0.0
  */
 public final class RtYamlLinesTest {