Browse Source
This set of tests covers primitive storage of an empty file, and each type of supported reference. Change-Id: I3bdff35cae8ae27283051932f20608b3ac353559stable-4.9
Shawn Pearce
8 years ago
2 changed files with 359 additions and 0 deletions
@ -0,0 +1,357 @@
|
||||
/* |
||||
* Copyright (C) 2017, Google Inc. |
||||
* and other copyright owners as documented in the project's IP log. |
||||
* |
||||
* This program and the accompanying materials are made available |
||||
* under the terms of the Eclipse Distribution License v1.0 which |
||||
* accompanies this distribution, is reproduced below, and is |
||||
* available at http://www.eclipse.org/org/documents/edl-v10.php
|
||||
* |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or |
||||
* without modification, are permitted provided that the following |
||||
* conditions are met: |
||||
* |
||||
* - Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* |
||||
* - Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following |
||||
* disclaimer in the documentation and/or other materials provided |
||||
* with the distribution. |
||||
* |
||||
* - Neither the name of the Eclipse Foundation, Inc. nor the |
||||
* names of its contributors may be used to endorse or promote |
||||
* products derived from this software without specific prior |
||||
* written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND |
||||
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, |
||||
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
||||
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
||||
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
||||
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
||||
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, |
||||
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
||||
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF |
||||
* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
|
||||
package org.eclipse.jgit.internal.storage.reftable; |
||||
|
||||
import static org.eclipse.jgit.lib.Constants.HEAD; |
||||
import static org.eclipse.jgit.lib.Constants.OBJECT_ID_LENGTH; |
||||
import static org.eclipse.jgit.lib.Constants.R_HEADS; |
||||
import static org.eclipse.jgit.lib.Ref.Storage.NEW; |
||||
import static org.eclipse.jgit.lib.Ref.Storage.PACKED; |
||||
import static org.junit.Assert.assertEquals; |
||||
import static org.junit.Assert.assertFalse; |
||||
import static org.junit.Assert.assertNotNull; |
||||
import static org.junit.Assert.assertNull; |
||||
import static org.junit.Assert.assertTrue; |
||||
import static org.junit.Assert.fail; |
||||
|
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.IOException; |
||||
import java.util.ArrayList; |
||||
import java.util.Arrays; |
||||
import java.util.Collection; |
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
|
||||
import org.eclipse.jgit.internal.JGitText; |
||||
import org.eclipse.jgit.internal.storage.io.BlockSource; |
||||
import org.eclipse.jgit.internal.storage.reftable.ReftableWriter.Stats; |
||||
import org.eclipse.jgit.lib.ObjectId; |
||||
import org.eclipse.jgit.lib.ObjectIdRef; |
||||
import org.eclipse.jgit.lib.Ref; |
||||
import org.eclipse.jgit.lib.SymbolicRef; |
||||
import org.junit.Test; |
||||
|
||||
public class ReftableTest { |
||||
private static final String MASTER = "refs/heads/master"; |
||||
private static final String V1_0 = "refs/tags/v1.0"; |
||||
|
||||
private Stats stats; |
||||
|
||||
@Test |
||||
public void emptyTable() throws IOException { |
||||
byte[] table = write(); |
||||
assertEquals(92 /* header, footer */, table.length); |
||||
assertEquals('R', table[0]); |
||||
assertEquals('E', table[1]); |
||||
assertEquals('F', table[2]); |
||||
assertEquals('T', table[3]); |
||||
assertEquals(0x01, table[4]); |
||||
assertTrue(ReftableConstants.isFileHeaderMagic(table, 0, 8)); |
||||
assertTrue(ReftableConstants.isFileHeaderMagic(table, 24, 92)); |
||||
|
||||
Reftable t = read(table); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (RefCursor rc = t.seekRef(HEAD)) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (RefCursor rc = t.seekRef(R_HEADS)) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (LogCursor rc = t.allLogs()) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void emptyVirtualTableFromRefs() throws IOException { |
||||
Reftable t = Reftable.from(Collections.emptyList()); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (RefCursor rc = t.seekRef(HEAD)) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (LogCursor rc = t.allLogs()) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void estimateCurrentBytesOneRef() throws IOException { |
||||
Ref exp = ref(MASTER, 1); |
||||
int expBytes = 24 + 4 + 5 + 3 + MASTER.length() + 20 + 68; |
||||
|
||||
byte[] table; |
||||
ReftableConfig cfg = new ReftableConfig(); |
||||
cfg.setIndexObjects(false); |
||||
ReftableWriter writer = new ReftableWriter().setConfig(cfg); |
||||
try (ByteArrayOutputStream buf = new ByteArrayOutputStream()) { |
||||
writer.begin(buf); |
||||
assertEquals(92, writer.estimateTotalBytes()); |
||||
writer.writeRef(exp); |
||||
assertEquals(expBytes, writer.estimateTotalBytes()); |
||||
writer.finish(); |
||||
table = buf.toByteArray(); |
||||
} |
||||
assertEquals(expBytes, table.length); |
||||
} |
||||
|
||||
@SuppressWarnings("boxing") |
||||
@Test |
||||
public void estimateCurrentBytesWithIndex() throws IOException { |
||||
List<Ref> refs = new ArrayList<>(); |
||||
for (int i = 1; i <= 5670; i++) { |
||||
refs.add(ref(String.format("refs/heads/%04d", i), i)); |
||||
} |
||||
|
||||
ReftableConfig cfg = new ReftableConfig(); |
||||
cfg.setIndexObjects(false); |
||||
cfg.setMaxIndexLevels(1); |
||||
|
||||
int expBytes = 139654; |
||||
byte[] table; |
||||
ReftableWriter writer = new ReftableWriter().setConfig(cfg); |
||||
try (ByteArrayOutputStream buf = new ByteArrayOutputStream()) { |
||||
writer.begin(buf); |
||||
writer.sortAndWriteRefs(refs); |
||||
assertEquals(expBytes, writer.estimateTotalBytes()); |
||||
writer.finish(); |
||||
stats = writer.getStats(); |
||||
table = buf.toByteArray(); |
||||
} |
||||
assertEquals(1, stats.refIndexLevels()); |
||||
assertEquals(expBytes, table.length); |
||||
} |
||||
|
||||
@Test |
||||
public void oneIdRef() throws IOException { |
||||
Ref exp = ref(MASTER, 1); |
||||
byte[] table = write(exp); |
||||
assertEquals(24 + 4 + 5 + 3 + MASTER.length() + 20 + 68, table.length); |
||||
|
||||
ReftableReader t = read(table); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertTrue(rc.next()); |
||||
Ref act = rc.getRef(); |
||||
assertNotNull(act); |
||||
assertEquals(PACKED, act.getStorage()); |
||||
assertTrue(act.isPeeled()); |
||||
assertFalse(act.isSymbolic()); |
||||
assertEquals(exp.getName(), act.getName()); |
||||
assertEquals(exp.getObjectId(), act.getObjectId()); |
||||
assertNull(act.getPeeledObjectId()); |
||||
assertFalse(rc.wasDeleted()); |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (RefCursor rc = t.seekRef(MASTER)) { |
||||
assertTrue(rc.next()); |
||||
Ref act = rc.getRef(); |
||||
assertNotNull(act); |
||||
assertEquals(exp.getName(), act.getName()); |
||||
assertFalse(rc.next()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void oneTagRef() throws IOException { |
||||
Ref exp = tag(V1_0, 1, 2); |
||||
byte[] table = write(exp); |
||||
assertEquals(24 + 4 + 5 + 2 + V1_0.length() + 40 + 68, table.length); |
||||
|
||||
ReftableReader t = read(table); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertTrue(rc.next()); |
||||
Ref act = rc.getRef(); |
||||
assertNotNull(act); |
||||
assertEquals(PACKED, act.getStorage()); |
||||
assertTrue(act.isPeeled()); |
||||
assertFalse(act.isSymbolic()); |
||||
assertEquals(exp.getName(), act.getName()); |
||||
assertEquals(exp.getObjectId(), act.getObjectId()); |
||||
assertEquals(exp.getPeeledObjectId(), act.getPeeledObjectId()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void oneSymbolicRef() throws IOException { |
||||
Ref exp = sym(HEAD, MASTER); |
||||
byte[] table = write(exp); |
||||
assertEquals( |
||||
24 + 4 + 5 + 2 + HEAD.length() + 1 + MASTER.length() + 68, |
||||
table.length); |
||||
|
||||
ReftableReader t = read(table); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertTrue(rc.next()); |
||||
Ref act = rc.getRef(); |
||||
assertNotNull(act); |
||||
assertTrue(act.isSymbolic()); |
||||
assertEquals(exp.getName(), act.getName()); |
||||
assertNotNull(act.getLeaf()); |
||||
assertEquals(MASTER, act.getTarget().getName()); |
||||
assertNull(act.getObjectId()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void oneDeletedRef() throws IOException { |
||||
String name = "refs/heads/gone"; |
||||
Ref exp = newRef(name); |
||||
byte[] table = write(exp); |
||||
assertEquals(24 + 4 + 5 + 2 + name.length() + 68, table.length); |
||||
|
||||
ReftableReader t = read(table); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
|
||||
t.setIncludeDeletes(true); |
||||
try (RefCursor rc = t.allRefs()) { |
||||
assertTrue(rc.next()); |
||||
Ref act = rc.getRef(); |
||||
assertNotNull(act); |
||||
assertFalse(act.isSymbolic()); |
||||
assertEquals(name, act.getName()); |
||||
assertEquals(NEW, act.getStorage()); |
||||
assertNull(act.getObjectId()); |
||||
assertTrue(rc.wasDeleted()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void seekNotFound() throws IOException { |
||||
Ref exp = ref(MASTER, 1); |
||||
ReftableReader t = read(write(exp)); |
||||
try (RefCursor rc = t.seekRef("refs/heads/a")) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
try (RefCursor rc = t.seekRef("refs/heads/n")) { |
||||
assertFalse(rc.next()); |
||||
} |
||||
} |
||||
|
||||
public void unpeeledDoesNotWrite() { |
||||
try { |
||||
write(new ObjectIdRef.Unpeeled(PACKED, MASTER, id(1))); |
||||
fail("expected IOException"); |
||||
} catch (IOException e) { |
||||
assertEquals(JGitText.get().peeledRefIsRequired, e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void nameTooLongDoesNotWrite() throws IOException { |
||||
try { |
||||
ReftableConfig cfg = new ReftableConfig(); |
||||
cfg.setRefBlockSize(64); |
||||
|
||||
ByteArrayOutputStream buffer = new ByteArrayOutputStream(); |
||||
ReftableWriter writer = new ReftableWriter(cfg).begin(buffer); |
||||
writer.writeRef(ref("refs/heads/i-am-not-a-teapot", 1)); |
||||
writer.finish(); |
||||
fail("expected BlockSizeTooSmallException"); |
||||
} catch (BlockSizeTooSmallException e) { |
||||
assertEquals(84, e.getMinimumBlockSize()); |
||||
} |
||||
} |
||||
|
||||
@Test |
||||
public void badCrc32() throws IOException { |
||||
byte[] table = write(); |
||||
table[table.length - 1] = 0x42; |
||||
|
||||
try { |
||||
read(table).seekRef(HEAD); |
||||
fail("expected IOException"); |
||||
} catch (IOException e) { |
||||
assertEquals(JGitText.get().invalidReftableCRC, e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
|
||||
private static Ref ref(String name, int id) { |
||||
return new ObjectIdRef.PeeledNonTag(PACKED, name, id(id)); |
||||
} |
||||
|
||||
private static Ref tag(String name, int id1, int id2) { |
||||
return new ObjectIdRef.PeeledTag(PACKED, name, id(id1), id(id2)); |
||||
} |
||||
|
||||
private static Ref sym(String name, String target) { |
||||
return new SymbolicRef(name, newRef(target)); |
||||
} |
||||
|
||||
private static Ref newRef(String name) { |
||||
return new ObjectIdRef.Unpeeled(NEW, name, null); |
||||
} |
||||
|
||||
private static ObjectId id(int i) { |
||||
byte[] buf = new byte[OBJECT_ID_LENGTH]; |
||||
buf[0] = (byte) (i & 0xff); |
||||
buf[1] = (byte) ((i >>> 8) & 0xff); |
||||
buf[2] = (byte) ((i >>> 16) & 0xff); |
||||
buf[3] = (byte) (i >>> 24); |
||||
return ObjectId.fromRaw(buf); |
||||
} |
||||
|
||||
private static ReftableReader read(byte[] table) { |
||||
return new ReftableReader(BlockSource.from(table)); |
||||
} |
||||
|
||||
private byte[] write(Ref... refs) throws IOException { |
||||
return write(Arrays.asList(refs)); |
||||
} |
||||
|
||||
private byte[] write(Collection<Ref> refs) throws IOException { |
||||
ByteArrayOutputStream buffer = new ByteArrayOutputStream(); |
||||
stats = new ReftableWriter() |
||||
.begin(buffer) |
||||
.sortAndWriteRefs(refs) |
||||
.finish() |
||||
.getStats(); |
||||
return buffer.toByteArray(); |
||||
} |
||||
} |
Loading…
Reference in new issue