org.spearce.jgit.test/tst/org/spearce/jgit/treewalk/PostOrderTreeWalkTest.java

   1 /*
   2  * Copyright (C) 2008, Google Inc.
   3  *
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or
   7  * without modification, are permitted provided that the following
   8  * conditions are met:
   9  *
  10  * - Redistributions of source code must retain the above copyright
  11  *   notice, this list of conditions and the following disclaimer.
  12  *
  13  * - Redistributions in binary form must reproduce the above
  14  *   copyright notice, this list of conditions and the following
  15  *   disclaimer in the documentation and/or other materials provided
  16  *   with the distribution.
  17  *
  18  * - Neither the name of the Git Development Community nor the
  19  *   names of its contributors may be used to endorse or promote
  20  *   products derived from this software without specific prior
  21  *   written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  24  * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  25  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  26  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  28  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  31  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  32  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  33  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  34  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  35  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  36  */
  37
  38 package org.spearce.jgit.treewalk;
  39
  40 import java.io.ByteArrayInputStream;
  41
  42 import org.spearce.jgit.dircache.DirCache;
  43 import org.spearce.jgit.dircache.DirCacheBuilder;
  44 import org.spearce.jgit.dircache.DirCacheEntry;
  45 import org.spearce.jgit.dircache.DirCacheIterator;
  46 import org.spearce.jgit.lib.Constants;
  47 import org.spearce.jgit.lib.FileMode;
  48 import org.spearce.jgit.lib.ObjectWriter;
  49 import org.spearce.jgit.lib.RepositoryTestCase;
  50
  51 import static org.spearce.jgit.lib.FileMode.REGULAR_FILE;
  52 import static org.spearce.jgit.lib.FileMode.TREE;
  53
  54 public class PostOrderTreeWalkTest extends RepositoryTestCase {
  55         public void testInitialize_NoPostOrder() throws Exception {
  56                 final TreeWalk tw = new TreeWalk(db);
  57                 assertFalse(tw.isPostOrderTraversal());
  58         }
  59
  60         public void testInitialize_TogglePostOrder() throws Exception {
  61                 final TreeWalk tw = new TreeWalk(db);
  62                 assertFalse(tw.isPostOrderTraversal());
  63                 tw.setPostOrderTraversal(true);
  64                 assertTrue(tw.isPostOrderTraversal());
  65                 tw.setPostOrderTraversal(false);
  66                 assertFalse(tw.isPostOrderTraversal());
  67         }
  68
  69         public void testResetDoesNotAffectPostOrder() throws Exception {
  70                 final TreeWalk tw = new TreeWalk(db);
  71                 tw.setPostOrderTraversal(true);
  72                 assertTrue(tw.isPostOrderTraversal());
  73                 tw.reset();
  74                 assertTrue(tw.isPostOrderTraversal());
  75
  76                 tw.setPostOrderTraversal(false);
  77                 assertFalse(tw.isPostOrderTraversal());
  78                 tw.reset();
  79                 assertFalse(tw.isPostOrderTraversal());
  80         }
  81
  82         public void testNoPostOrder() throws Exception {
  83                 final DirCache tree = DirCache.read(db);
  84                 {
  85                         final DirCacheBuilder b = tree.builder();
  86
  87                         b.add(makeFile("a"));
  88                         b.add(makeFile("b/c"));
  89                         b.add(makeFile("b/d"));
  90                         b.add(makeFile("q"));
  91
  92                         b.finish();
  93                         assertEquals(4, tree.getEntryCount());
  94                 }
  95
  96                 final TreeWalk tw = new TreeWalk(db);
  97                 tw.reset();
  98                 tw.setPostOrderTraversal(false);
  99                 tw.addTree(new DirCacheIterator(tree));
 100
 101                 assertModes("a", REGULAR_FILE, tw);
 102                 assertModes("b", TREE, tw);
 103                 assertTrue(tw.isSubtree());
 104                 assertFalse(tw.isPostChildren());
 105                 tw.enterSubtree();
 106                 assertModes("b/c", REGULAR_FILE, tw);
 107                 assertModes("b/d", REGULAR_FILE, tw);
 108                 assertModes("q", REGULAR_FILE, tw);
 109         }
 110
 111         public void testWithPostOrder_EnterSubtree() throws Exception {
 112                 final DirCache tree = DirCache.read(db);
 113                 {
 114                         final DirCacheBuilder b = tree.builder();
 115
 116                         b.add(makeFile("a"));
 117                         b.add(makeFile("b/c"));
 118                         b.add(makeFile("b/d"));
 119                         b.add(makeFile("q"));
 120
 121                         b.finish();
 122                         assertEquals(4, tree.getEntryCount());
 123                 }
 124
 125                 final TreeWalk tw = new TreeWalk(db);
 126                 tw.reset();
 127                 tw.setPostOrderTraversal(true);
 128                 tw.addTree(new DirCacheIterator(tree));
 129
 130                 assertModes("a", REGULAR_FILE, tw);
 131
 132                 assertModes("b", TREE, tw);
 133                 assertTrue(tw.isSubtree());
 134                 assertFalse(tw.isPostChildren());
 135                 tw.enterSubtree();
 136                 assertModes("b/c", REGULAR_FILE, tw);
 137                 assertModes("b/d", REGULAR_FILE, tw);
 138
 139                 assertModes("b", TREE, tw);
 140                 assertTrue(tw.isSubtree());
 141                 assertTrue(tw.isPostChildren());
 142
 143                 assertModes("q", REGULAR_FILE, tw);
 144         }
 145
 146         public void testWithPostOrder_NoEnterSubtree() throws Exception {
 147                 final DirCache tree = DirCache.read(db);
 148                 {
 149                         final DirCacheBuilder b = tree.builder();
 150
 151                         b.add(makeFile("a"));
 152                         b.add(makeFile("b/c"));
 153                         b.add(makeFile("b/d"));
 154                         b.add(makeFile("q"));
 155
 156                         b.finish();
 157                         assertEquals(4, tree.getEntryCount());
 158                 }
 159
 160                 final TreeWalk tw = new TreeWalk(db);
 161                 tw.reset();
 162                 tw.setPostOrderTraversal(true);
 163                 tw.addTree(new DirCacheIterator(tree));
 164
 165                 assertModes("a", REGULAR_FILE, tw);
 166
 167                 assertModes("b", TREE, tw);
 168                 assertTrue(tw.isSubtree());
 169                 assertFalse(tw.isPostChildren());
 170
 171                 assertModes("q", REGULAR_FILE, tw);
 172         }
 173
 174         private DirCacheEntry makeFile(final String path) throws Exception {
 175                 final byte[] pathBytes = Constants.encode(path);
 176                 final DirCacheEntry ent = new DirCacheEntry(path);
 177                 ent.setFileMode(REGULAR_FILE);
 178                 ent.setObjectId(new ObjectWriter(db).computeBlobSha1(pathBytes.length,
 179                                 new ByteArrayInputStream(pathBytes)));
 180                 return ent;
 181         }
 182
 183         private static void assertModes(final String path, final FileMode mode0,
 184                         final TreeWalk tw) throws Exception {
 185                 assertTrue("has " + path, tw.next());
 186                 assertEquals(path, tw.getPathString());
 187                 assertEquals(mode0, tw.getFileMode(0));
 188         }
 189 }