Browse Source
* changes: Add test for protocol v0 parser ObjectIdMatcher: Custom matcher for sets of ObjectIdsstable-5.2
Jonathan Nieder
6 years ago
committed by
Gerrit Code Review @ Eclipse.org
3 changed files with 313 additions and 23 deletions
@ -0,0 +1,100 @@
|
||||
/* |
||||
* Copyright (C) 2018, Google LLC. |
||||
* 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.transport; |
||||
|
||||
import java.util.Collection; |
||||
import java.util.Set; |
||||
import java.util.stream.Collectors; |
||||
|
||||
import org.eclipse.jgit.lib.ObjectId; |
||||
import org.eclipse.jgit.lib.Sets; |
||||
import org.hamcrest.Description; |
||||
import org.hamcrest.Factory; |
||||
import org.hamcrest.Matcher; |
||||
import org.hamcrest.TypeSafeMatcher; |
||||
|
||||
/** |
||||
* Multiple tests check that a collection of ObjectIds contain certain SHA1 |
||||
* (written as strings). This matcher hides the ObjectId to string conversion to |
||||
* make the assertion more readable: |
||||
* |
||||
* assertThat(req.getWantsIds(), hasOnlyObjectIds("123123", "234234")); |
||||
*/ |
||||
class ObjectIdMatcher extends TypeSafeMatcher<Collection<ObjectId>> { |
||||
|
||||
private final Set<ObjectId> expectedOids; |
||||
|
||||
private ObjectIdMatcher(Set<String> oids) { |
||||
this.expectedOids = oids.stream().map(ObjectId::fromString) |
||||
.collect(Collectors.toSet()); |
||||
} |
||||
|
||||
@Override |
||||
public void describeTo(Description desc) { |
||||
desc.appendText("Object ids:"); |
||||
desc.appendValueList("<", ",", ">", expectedOids); |
||||
} |
||||
|
||||
@Override |
||||
protected boolean matchesSafely(Collection<ObjectId> resultOids) { |
||||
return resultOids.containsAll(expectedOids) |
||||
&& expectedOids.containsAll(resultOids); |
||||
} |
||||
|
||||
/** |
||||
* Assert that all and only the received {@link ObjectId object ids} are in |
||||
* the expected set. |
||||
* <p> |
||||
* ObjectIds are compared by SHA1. |
||||
* |
||||
* @param oids |
||||
* Object ids to examine. |
||||
* @return true if examined and specified sets contains exactly the same |
||||
* elements. |
||||
*/ |
||||
@Factory |
||||
static Matcher<Collection<ObjectId>> hasOnlyObjectIds( |
||||
String... oids) { |
||||
return new ObjectIdMatcher(Sets.of(oids)); |
||||
} |
||||
} |
@ -0,0 +1,199 @@
|
||||
/* |
||||
* Copyright (C) 2018, Google LLC. |
||||
* 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.transport; |
||||
|
||||
import static org.eclipse.jgit.transport.ObjectIdMatcher.hasOnlyObjectIds; |
||||
import static org.junit.Assert.assertEquals; |
||||
import static org.junit.Assert.assertThat; |
||||
import static org.junit.Assert.assertTrue; |
||||
|
||||
import java.io.ByteArrayInputStream; |
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.IOException; |
||||
|
||||
import org.eclipse.jgit.errors.PackProtocolException; |
||||
import org.eclipse.jgit.lib.Config; |
||||
import org.junit.Test; |
||||
|
||||
public class ProtocolV0ParserTest { |
||||
/* |
||||
* Convert the input lines to the PacketLine that the parser reads. |
||||
*/ |
||||
private static PacketLineIn formatAsPacketLine(String... inputLines) |
||||
throws IOException { |
||||
ByteArrayOutputStream send = new ByteArrayOutputStream(); |
||||
PacketLineOut pckOut = new PacketLineOut(send); |
||||
for (String line : inputLines) { |
||||
if (line == PacketLineIn.END) { |
||||
pckOut.end(); |
||||
} else if (line == PacketLineIn.DELIM) { |
||||
pckOut.writeDelim(); |
||||
} else { |
||||
pckOut.writeString(line); |
||||
} |
||||
} |
||||
|
||||
return new PacketLineIn(new ByteArrayInputStream(send.toByteArray())); |
||||
} |
||||
|
||||
private static TransferConfig defaultConfig() { |
||||
Config rc = new Config(); |
||||
rc.setBoolean("uploadpack", null, "allowfilter", true); |
||||
return new TransferConfig(rc); |
||||
} |
||||
|
||||
@Test |
||||
public void testRecvWantsWithCapabilities() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
String.join(" ", "want", |
||||
"4624442d68ee402a94364191085b77137618633e", "thin-pack", |
||||
"no-progress", "include-tag", "ofs-delta", "\n"), |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities() |
||||
.contains(GitProtocolConstants.OPTION_THIN_PACK)); |
||||
assertTrue(request.getClientCapabilities() |
||||
.contains(GitProtocolConstants.OPTION_NO_PROGRESS)); |
||||
assertTrue(request.getClientCapabilities() |
||||
.contains(GitProtocolConstants.OPTION_INCLUDE_TAG)); |
||||
assertTrue(request.getClientCapabilities() |
||||
.contains(GitProtocolConstants.CAPABILITY_OFS_DELTA)); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
} |
||||
|
||||
@Test |
||||
public void testRecvWantsWithAgent() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
String.join(" ", "want", |
||||
"4624442d68ee402a94364191085b77137618633e", "thin-pack", |
||||
"agent=JGit.test/0.0.1", "\n"), |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities() |
||||
.contains(GitProtocolConstants.OPTION_THIN_PACK)); |
||||
assertEquals(1, request.getClientCapabilities().size()); |
||||
assertEquals("JGit.test/0.0.1", request.getAgent()); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
} |
||||
|
||||
/* |
||||
* First round of protocol v0 negotiation. Client send wants, no |
||||
* capabilities. |
||||
*/ |
||||
@Test |
||||
public void testRecvWantsWithoutCapabilities() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
"want 4624442d68ee402a94364191085b77137618633e\n", |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities().isEmpty()); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
} |
||||
|
||||
@Test |
||||
public void testRecvWantsDeepen() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
"want 4624442d68ee402a94364191085b77137618633e\n", |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", "deepen 3\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities().isEmpty()); |
||||
assertEquals(3, request.getDepth()); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
} |
||||
|
||||
@Test |
||||
public void testRecvWantsShallow() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
"want 4624442d68ee402a94364191085b77137618633e\n", |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", |
||||
"shallow 4b643d0ef739a1b494e7d6926d8d8ed80d35edf4\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities().isEmpty()); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
assertThat(request.getClientShallowCommits(), |
||||
hasOnlyObjectIds("4b643d0ef739a1b494e7d6926d8d8ed80d35edf4")); |
||||
} |
||||
|
||||
@Test |
||||
public void testRecvWantsFilter() |
||||
throws PackProtocolException, IOException { |
||||
PacketLineIn pckIn = formatAsPacketLine( |
||||
"want 4624442d68ee402a94364191085b77137618633e\n", |
||||
"want f900c8326a43303685c46b279b9f70411bff1a4b\n", |
||||
"filter blob:limit=13000\n", |
||||
PacketLineIn.END); |
||||
ProtocolV0Parser parser = new ProtocolV0Parser(defaultConfig()); |
||||
FetchV0Request request = parser.recvWants(pckIn); |
||||
assertTrue(request.getClientCapabilities().isEmpty()); |
||||
assertThat(request.getWantIds(), |
||||
hasOnlyObjectIds("4624442d68ee402a94364191085b77137618633e", |
||||
"f900c8326a43303685c46b279b9f70411bff1a4b")); |
||||
assertEquals(13000, request.getFilterBlobLimit()); |
||||
} |
||||
|
||||
} |
Loading…
Reference in new issue