Browse Source
PostReceiveHooks can make use of this information to, for example, update a cached size of the Git repository. Change-Id: I2bf1200959a50531e2155a7609c96035ba45b10d Signed-off-by: Matthias Sohn <matthias.sohn@sap.com>stable-3.3
Saša Živkov
11 years ago
committed by
Matthias Sohn
6 changed files with 216 additions and 0 deletions
@ -0,0 +1,165 @@ |
|||||||
|
/* |
||||||
|
* Copyright (C) 2014, Matthias Sohn <matthias.sohn@sap.com> |
||||||
|
* 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.http.test; |
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals; |
||||||
|
|
||||||
|
import java.util.Collection; |
||||||
|
import java.util.Collections; |
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
|
||||||
|
import org.eclipse.jetty.servlet.ServletContextHandler; |
||||||
|
import org.eclipse.jetty.servlet.ServletHolder; |
||||||
|
import org.eclipse.jgit.errors.RepositoryNotFoundException; |
||||||
|
import org.eclipse.jgit.http.server.GitServlet; |
||||||
|
import org.eclipse.jgit.http.server.resolver.DefaultReceivePackFactory; |
||||||
|
import org.eclipse.jgit.junit.TestRepository; |
||||||
|
import org.eclipse.jgit.junit.http.HttpTestCase; |
||||||
|
import org.eclipse.jgit.lib.Constants; |
||||||
|
import org.eclipse.jgit.lib.NullProgressMonitor; |
||||||
|
import org.eclipse.jgit.lib.ObjectId; |
||||||
|
import org.eclipse.jgit.lib.Repository; |
||||||
|
import org.eclipse.jgit.lib.StoredConfig; |
||||||
|
import org.eclipse.jgit.revwalk.RevBlob; |
||||||
|
import org.eclipse.jgit.revwalk.RevCommit; |
||||||
|
import org.eclipse.jgit.transport.PostReceiveHook; |
||||||
|
import org.eclipse.jgit.transport.PushResult; |
||||||
|
import org.eclipse.jgit.transport.ReceiveCommand; |
||||||
|
import org.eclipse.jgit.transport.ReceivePack; |
||||||
|
import org.eclipse.jgit.transport.RemoteRefUpdate; |
||||||
|
import org.eclipse.jgit.transport.Transport; |
||||||
|
import org.eclipse.jgit.transport.URIish; |
||||||
|
import org.eclipse.jgit.transport.resolver.RepositoryResolver; |
||||||
|
import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException; |
||||||
|
import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException; |
||||||
|
import org.junit.Before; |
||||||
|
import org.junit.Test; |
||||||
|
|
||||||
|
public class MeasurePackSizeTest extends HttpTestCase { |
||||||
|
private Repository remoteRepository; |
||||||
|
|
||||||
|
private URIish remoteURI; |
||||||
|
|
||||||
|
long packSize = -1; |
||||||
|
|
||||||
|
@Before |
||||||
|
public void setUp() throws Exception { |
||||||
|
super.setUp(); |
||||||
|
|
||||||
|
final TestRepository<Repository> src = createTestRepository(); |
||||||
|
final String srcName = src.getRepository().getDirectory().getName(); |
||||||
|
|
||||||
|
ServletContextHandler app = server.addContext("/git"); |
||||||
|
GitServlet gs = new GitServlet(); |
||||||
|
gs.setRepositoryResolver(new RepositoryResolver<HttpServletRequest>() { |
||||||
|
public Repository open(HttpServletRequest req, String name) |
||||||
|
throws RepositoryNotFoundException, |
||||||
|
ServiceNotEnabledException { |
||||||
|
if (!name.equals(srcName)) |
||||||
|
throw new RepositoryNotFoundException(name); |
||||||
|
|
||||||
|
final Repository db = src.getRepository(); |
||||||
|
db.incrementOpen(); |
||||||
|
return db; |
||||||
|
} |
||||||
|
}); |
||||||
|
gs.setReceivePackFactory(new DefaultReceivePackFactory() { |
||||||
|
public ReceivePack create(HttpServletRequest req, Repository db) |
||||||
|
throws ServiceNotEnabledException, |
||||||
|
ServiceNotAuthorizedException { |
||||||
|
ReceivePack recv = super.create(req, db); |
||||||
|
recv.setPostReceiveHook(new PostReceiveHook() { |
||||||
|
|
||||||
|
public void onPostReceive(ReceivePack rp, |
||||||
|
Collection<ReceiveCommand> commands) { |
||||||
|
packSize = rp.getPackSize(); |
||||||
|
} |
||||||
|
}); |
||||||
|
return recv; |
||||||
|
} |
||||||
|
|
||||||
|
}); |
||||||
|
app.addServlet(new ServletHolder(gs), "/*"); |
||||||
|
|
||||||
|
server.setUp(); |
||||||
|
|
||||||
|
remoteRepository = src.getRepository(); |
||||||
|
remoteURI = toURIish(app, srcName); |
||||||
|
|
||||||
|
StoredConfig cfg = remoteRepository.getConfig(); |
||||||
|
cfg.setBoolean("http", null, "receivepack", true); |
||||||
|
cfg.save(); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testPush_packSize() throws Exception { |
||||||
|
final TestRepository src = createTestRepository(); |
||||||
|
final RevBlob Q_txt = src |
||||||
|
.blob("some blob content to measure pack size"); |
||||||
|
final RevCommit Q = src.commit().add("Q", Q_txt).create(); |
||||||
|
final Repository db = src.getRepository(); |
||||||
|
final String dstName = Constants.R_HEADS + "new.branch"; |
||||||
|
Transport t; |
||||||
|
PushResult result; |
||||||
|
|
||||||
|
t = Transport.open(db, remoteURI); |
||||||
|
try { |
||||||
|
final String srcExpr = Q.name(); |
||||||
|
final boolean forceUpdate = false; |
||||||
|
final String localName = null; |
||||||
|
final ObjectId oldId = null; |
||||||
|
|
||||||
|
RemoteRefUpdate update = new RemoteRefUpdate(src.getRepository(), |
||||||
|
srcExpr, dstName, forceUpdate, localName, oldId); |
||||||
|
result = t.push(NullProgressMonitor.INSTANCE, |
||||||
|
Collections.singleton(update)); |
||||||
|
} finally { |
||||||
|
t.close(); |
||||||
|
} |
||||||
|
assertEquals("expected 1 RemoteUpdate", 1, result.getRemoteUpdates() |
||||||
|
.size()); |
||||||
|
assertEquals("unexpected pack size", 1398, packSize); |
||||||
|
} |
||||||
|
|
||||||
|
} |
Loading…
Reference in new issue