123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350 |
- /*
- * 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.file;
-
- import static java.nio.charset.StandardCharsets.UTF_8;
- import static java.util.concurrent.TimeUnit.NANOSECONDS;
- import static java.util.concurrent.TimeUnit.SECONDS;
- import static org.eclipse.jgit.internal.storage.file.BatchRefUpdateTest.Result.LOCK_FAILURE;
- import static org.eclipse.jgit.internal.storage.file.BatchRefUpdateTest.Result.OK;
- import static org.eclipse.jgit.internal.storage.file.BatchRefUpdateTest.Result.REJECTED_MISSING_OBJECT;
- import static org.eclipse.jgit.internal.storage.file.BatchRefUpdateTest.Result.REJECTED_NONFASTFORWARD;
- import static org.eclipse.jgit.internal.storage.file.BatchRefUpdateTest.Result.TRANSACTION_ABORTED;
- import static org.eclipse.jgit.lib.ObjectId.zeroId;
- import static org.eclipse.jgit.transport.ReceiveCommand.Type.CREATE;
- import static org.eclipse.jgit.transport.ReceiveCommand.Type.DELETE;
- import static org.eclipse.jgit.transport.ReceiveCommand.Type.UPDATE;
- import static org.eclipse.jgit.transport.ReceiveCommand.Type.UPDATE_NONFASTFORWARD;
- 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.Assume.assumeTrue;
-
- import java.io.File;
- import java.io.IOException;
- import java.nio.file.Files;
- import java.util.Arrays;
- import java.util.Collection;
- import java.util.Collections;
- import java.util.LinkedHashMap;
- import java.util.List;
- import java.util.Map;
- import java.util.concurrent.locks.ReentrantLock;
- import java.util.function.Predicate;
-
- import org.eclipse.jgit.events.ListenerHandle;
- import org.eclipse.jgit.events.RefsChangedListener;
- import org.eclipse.jgit.junit.LocalDiskRepositoryTestCase;
- import org.eclipse.jgit.junit.StrictWorkMonitor;
- import org.eclipse.jgit.junit.TestRepository;
- import org.eclipse.jgit.lib.AnyObjectId;
- import org.eclipse.jgit.lib.BatchRefUpdate;
- import org.eclipse.jgit.lib.CheckoutEntry;
- import org.eclipse.jgit.lib.ConfigConstants;
- import org.eclipse.jgit.lib.Constants;
- import org.eclipse.jgit.lib.NullProgressMonitor;
- import org.eclipse.jgit.lib.ObjectId;
- import org.eclipse.jgit.lib.PersonIdent;
- import org.eclipse.jgit.lib.Ref;
- import org.eclipse.jgit.lib.RefDatabase;
- import org.eclipse.jgit.lib.RefUpdate;
- import org.eclipse.jgit.lib.ReflogEntry;
- import org.eclipse.jgit.lib.ReflogReader;
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.lib.StoredConfig;
- import org.eclipse.jgit.revwalk.RevCommit;
- import org.eclipse.jgit.revwalk.RevWalk;
- import org.eclipse.jgit.transport.ReceiveCommand;
- import org.junit.After;
- import org.junit.Before;
- import org.junit.Test;
- import org.junit.runner.RunWith;
- import org.junit.runners.Parameterized;
- import org.junit.runners.Parameterized.Parameter;
- import org.junit.runners.Parameterized.Parameters;
-
- @SuppressWarnings("boxing")
- @RunWith(Parameterized.class)
- public class BatchRefUpdateTest extends LocalDiskRepositoryTestCase {
- @Parameter
- public boolean atomic;
-
- @Parameters(name = "atomic={0}")
- public static Collection<Object[]> data() {
- return Arrays.asList(new Object[][]{ {Boolean.FALSE}, {Boolean.TRUE} });
- }
-
- private Repository diskRepo;
- private TestRepository<Repository> repo;
- private RefDirectory refdir;
- private RevCommit A;
- private RevCommit B;
-
- /**
- * When asserting the number of RefsChangedEvents you must account for one
- * additional event due to the initial ref setup via a number of calls to
- * {@link #writeLooseRef(String, AnyObjectId)} (will be fired in execute()
- * when it is detected that the on-disk loose refs have changed), or for one
- * additional event per {@link #writeRef(String, AnyObjectId)}.
- */
- private int refsChangedEvents;
-
- private ListenerHandle handle;
-
- private RefsChangedListener refsChangedListener = event -> {
- refsChangedEvents++;
- };
-
- @Override
- @Before
- public void setUp() throws Exception {
- super.setUp();
-
- diskRepo = createBareRepository();
- setLogAllRefUpdates(true);
-
- refdir = (RefDirectory) diskRepo.getRefDatabase();
- refdir.setRetrySleepMs(Arrays.asList(0, 0));
-
- repo = new TestRepository<>(diskRepo);
- A = repo.commit().create();
- B = repo.commit(repo.getRevWalk().parseCommit(A));
- refsChangedEvents = 0;
- handle = diskRepo.getListenerList()
- .addRefsChangedListener(refsChangedListener);
- }
-
- @After
- public void removeListener() {
- handle.remove();
- refsChangedEvents = 0;
- }
-
- @Test
- public void packedRefsFileIsSorted() throws IOException {
- assumeTrue(atomic);
-
- for (int i = 0; i < 2; i++) {
- BatchRefUpdate bu = diskRepo.getRefDatabase().newBatchUpdate();
- String b1 = String.format("refs/heads/a%d",i);
- String b2 = String.format("refs/heads/b%d",i);
- bu.setAtomic(atomic);
- ReceiveCommand c1 = new ReceiveCommand(ObjectId.zeroId(), A, b1);
- ReceiveCommand c2 = new ReceiveCommand(ObjectId.zeroId(), B, b2);
- bu.addCommand(c1, c2);
- try (RevWalk rw = new RevWalk(diskRepo)) {
- bu.execute(rw, NullProgressMonitor.INSTANCE);
- }
- assertEquals(c1.getResult(), ReceiveCommand.Result.OK);
- assertEquals(c2.getResult(), ReceiveCommand.Result.OK);
- }
-
- File packed = new File(diskRepo.getDirectory(), "packed-refs");
- String packedStr = new String(Files.readAllBytes(packed.toPath()), UTF_8);
-
- int a2 = packedStr.indexOf("refs/heads/a1");
- int b1 = packedStr.indexOf("refs/heads/b0");
- assertTrue(a2 < b1);
- }
-
- @Test
- public void simpleNoForce() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(B, A, "refs/heads/masters", UPDATE_NONFASTFORWARD));
- execute(newBatchUpdate(cmds));
-
- if (atomic) {
- assertResults(cmds, TRANSACTION_ABORTED, REJECTED_NONFASTFORWARD);
- assertRefs(
- "refs/heads/master", A,
- "refs/heads/masters", B);
- } else {
- assertResults(cmds, OK, REJECTED_NONFASTFORWARD);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/masters", B);
- }
- }
-
- @Test
- public void simpleNoForceRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
- refdir.exactRef("refs/heads/master");
- refdir.exactRef("refs/heads/masters");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(B, A, "refs/heads/masters",
- UPDATE_NONFASTFORWARD));
- execute(newBatchUpdate(cmds));
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void simpleForce() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(B, A, "refs/heads/masters", UPDATE_NONFASTFORWARD));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertResults(cmds, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/masters", A);
- }
-
- @Test
- public void simpleForceRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
- refdir.exactRef("refs/heads/master");
- refdir.exactRef("refs/heads/masters");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(B, A, "refs/heads/masters",
- UPDATE_NONFASTFORWARD));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- }
-
- @Test
- public void nonFastForwardDoesNotDoExpensiveMergeCheck() throws IOException {
- writeLooseRef("refs/heads/master", B);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(B, A, "refs/heads/master", UPDATE_NONFASTFORWARD));
- try (RevWalk rw = new RevWalk(diskRepo) {
- @Override
- public boolean isMergedInto(RevCommit base, RevCommit tip) {
- throw new AssertionError("isMergedInto() should not be called");
- }
- }) {
- newBatchUpdate(cmds)
- .setAllowNonFastForwards(true)
- .execute(rw, new StrictWorkMonitor());
- }
-
- assertResults(cmds, OK);
- assertRefs("refs/heads/master", A);
- }
-
- @Test
- public void nonFastForwardDoesNotDoExpensiveMergeCheckRefsChangedEvents()
- throws IOException {
- writeLooseRef("refs/heads/master", B);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(new ReceiveCommand(B, A,
- "refs/heads/master", UPDATE_NONFASTFORWARD));
- try (RevWalk rw = new RevWalk(diskRepo) {
- @Override
- public boolean isMergedInto(RevCommit base, RevCommit tip) {
- throw new AssertionError("isMergedInto() should not be called");
- }
- }) {
- newBatchUpdate(cmds).setAllowNonFastForwards(true).execute(rw,
- new StrictWorkMonitor());
- }
-
- assertEquals(initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void fileDirectoryConflict() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), A, "refs/heads/master/x", CREATE),
- new ReceiveCommand(zeroId(), A, "refs/heads", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- if (atomic) {
- // Atomic update sees that master and master/x are conflicting, then marks
- // the first one in the list as LOCK_FAILURE and aborts the rest.
- assertResults(cmds,
- LOCK_FAILURE, TRANSACTION_ABORTED, TRANSACTION_ABORTED);
- assertRefs(
- "refs/heads/master", A,
- "refs/heads/masters", B);
- } else {
- // Non-atomic updates are applied in order: master succeeds, then master/x
- // fails due to conflict.
- assertResults(cmds, OK, LOCK_FAILURE, LOCK_FAILURE);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/masters", B);
- }
- }
-
- @Test
- public void fileDirectoryConflictRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
- refdir.exactRef("refs/heads/master");
- refdir.exactRef("refs/heads/masters");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), A, "refs/heads/master/x", CREATE),
- new ReceiveCommand(zeroId(), A, "refs/heads", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void conflictThanksToDelete() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), A, "refs/heads/masters/x", CREATE),
- new ReceiveCommand(B, zeroId(), "refs/heads/masters", DELETE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertResults(cmds, OK, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/masters/x", A);
- }
-
- @Test
- public void conflictThanksToDeleteRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- writeLooseRef("refs/heads/masters", B);
- refdir.exactRef("refs/heads/master");
- refdir.exactRef("refs/heads/masters");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), A, "refs/heads/masters/x", CREATE),
- new ReceiveCommand(B, zeroId(), "refs/heads/masters", DELETE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 3, refsChangedEvents);
- }
-
- @Test
- public void updateToMissingObject() throws IOException {
- writeLooseRef("refs/heads/master", A);
-
- ObjectId bad =
- ObjectId.fromString("deadbeefdeadbeefdeadbeefdeadbeefdeadbeef");
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, bad, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- if (atomic) {
- assertResults(cmds, REJECTED_MISSING_OBJECT, TRANSACTION_ABORTED);
- assertRefs("refs/heads/master", A);
- } else {
- assertResults(cmds, REJECTED_MISSING_OBJECT, OK);
- assertRefs(
- "refs/heads/master", A,
- "refs/heads/foo2", B);
- }
- }
-
- @Test
- public void updateToMissingObjectRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- ObjectId bad = ObjectId
- .fromString("deadbeefdeadbeefdeadbeefdeadbeefdeadbeef");
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, bad, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void addMissingObject() throws IOException {
- writeLooseRef("refs/heads/master", A);
-
- ObjectId bad =
- ObjectId.fromString("deadbeefdeadbeefdeadbeefdeadbeefdeadbeef");
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), bad, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- if (atomic) {
- assertResults(cmds, TRANSACTION_ABORTED, REJECTED_MISSING_OBJECT);
- assertRefs("refs/heads/master", A);
- } else {
- assertResults(cmds, OK, REJECTED_MISSING_OBJECT);
- assertRefs("refs/heads/master", B);
- }
- }
-
- @Test
- public void addMissingObjectRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- ObjectId bad = ObjectId
- .fromString("deadbeefdeadbeefdeadbeefdeadbeefdeadbeef");
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), bad, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true), false);
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void oneNonExistentRef() throws IOException {
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/foo1", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- if (atomic) {
- assertResults(cmds, LOCK_FAILURE, TRANSACTION_ABORTED);
- assertRefs();
- assertEquals(0, refsChangedEvents);
- } else {
- assertResults(cmds, LOCK_FAILURE, OK);
- assertRefs("refs/heads/foo2", B);
- assertEquals(1, refsChangedEvents);
- }
- }
-
- @Test
- public void oneRefWrongOldValue() throws IOException {
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(B, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- if (atomic) {
- assertResults(cmds, LOCK_FAILURE, TRANSACTION_ABORTED);
- assertRefs("refs/heads/master", A);
- } else {
- assertResults(cmds, LOCK_FAILURE, OK);
- assertRefs(
- "refs/heads/master", A,
- "refs/heads/foo2", B);
- }
- }
-
- @Test
- public void oneRefWrongOldValueRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(B, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/foo2", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void nonExistentRef() throws IOException {
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(A, zeroId(), "refs/heads/foo2", DELETE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- if (atomic) {
- assertResults(cmds, TRANSACTION_ABORTED, LOCK_FAILURE);
- assertRefs("refs/heads/master", A);
- } else {
- assertResults(cmds, OK, LOCK_FAILURE);
- assertRefs("refs/heads/master", B);
- }
- }
-
- @Test
- public void nonExistentRefRefsChangedEvents() throws IOException {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(A, zeroId(), "refs/heads/foo2", DELETE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- @Test
- public void noRefLog() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- Map<String, ReflogEntry> oldLogs =
- getLastReflogs("refs/heads/master", "refs/heads/branch");
- assertEquals(Collections.singleton("refs/heads/master"), oldLogs.keySet());
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertResults(cmds, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch", B);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertReflogUnchanged(oldLogs, "refs/heads/master");
- assertReflogUnchanged(oldLogs, "refs/heads/branch");
- }
-
- @Test
- public void reflogDefaultIdent() throws IOException {
- writeRef("refs/heads/master", A);
- writeRef("refs/heads/branch2", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- Map<String, ReflogEntry> oldLogs = getLastReflogs(
- "refs/heads/master", "refs/heads/branch1", "refs/heads/branch2");
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch1", CREATE));
- execute(
- newBatchUpdate(cmds)
- .setAllowNonFastForwards(true)
- .setRefLogMessage("a reflog", false));
-
- assertResults(cmds, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch1", B,
- "refs/heads/branch2", A);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(zeroId(), B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/branch1"));
- assertReflogUnchanged(oldLogs, "refs/heads/branch2");
- }
-
- @Test
- public void reflogAppendStatusNoMessage() throws IOException {
- writeRef("refs/heads/master", A);
- writeRef("refs/heads/branch1", B);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(B, A, "refs/heads/branch1", UPDATE_NONFASTFORWARD),
- new ReceiveCommand(zeroId(), A, "refs/heads/branch2", CREATE));
- execute(
- newBatchUpdate(cmds)
- .setAllowNonFastForwards(true)
- .setRefLogMessage(null, true));
-
- assertResults(cmds, OK, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch1", A,
- "refs/heads/branch2", A);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 3,
- refsChangedEvents);
- assertReflogEquals(
- // Always forced; setAllowNonFastForwards(true) bypasses the check.
- reflog(A, B, new PersonIdent(diskRepo), "forced-update"),
- getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(B, A, new PersonIdent(diskRepo), "forced-update"),
- getLastReflog("refs/heads/branch1"));
- assertReflogEquals(
- reflog(zeroId(), A, new PersonIdent(diskRepo), "created"),
- getLastReflog("refs/heads/branch2"));
- }
-
- @Test
- public void reflogAppendStatusFastForward() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE));
- execute(newBatchUpdate(cmds).setRefLogMessage(null, true));
-
- assertResults(cmds, OK);
- assertRefs("refs/heads/master", B);
- assertEquals(initialRefsChangedEvents + 1, refsChangedEvents);
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "fast-forward"),
- getLastReflog("refs/heads/master"));
- }
-
- @Test
- public void reflogAppendStatusWithMessage() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), A, "refs/heads/branch", CREATE));
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", true));
-
- assertResults(cmds, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch", A);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "a reflog: fast-forward"),
- getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(zeroId(), A, new PersonIdent(diskRepo), "a reflog: created"),
- getLastReflog("refs/heads/branch"));
- }
-
- @Test
- public void reflogCustomIdent() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- PersonIdent ident = new PersonIdent("A Reflog User", "reflog@example.com");
- execute(
- newBatchUpdate(cmds)
- .setRefLogMessage("a reflog", false)
- .setRefLogIdent(ident));
-
- assertResults(cmds, OK, OK);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch", B);
- assertReflogEquals(
- reflog(A, B, ident, "a reflog"),
- getLastReflog("refs/heads/master"),
- true);
- assertReflogEquals(
- reflog(zeroId(), B, ident, "a reflog"),
- getLastReflog("refs/heads/branch"),
- true);
- }
-
- @Test
- public void reflogDelete() throws IOException {
- writeRef("refs/heads/master", A);
- writeRef("refs/heads/branch", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- assertEquals(
- 2, getLastReflogs("refs/heads/master", "refs/heads/branch").size());
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, zeroId(), "refs/heads/master", DELETE),
- new ReceiveCommand(A, B, "refs/heads/branch", UPDATE));
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", false));
-
- assertResults(cmds, OK, OK);
- assertRefs("refs/heads/branch", B);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertNull(getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/branch"));
- }
-
- @Test
- public void reflogFileDirectoryConflict() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, zeroId(), "refs/heads/master", DELETE),
- new ReceiveCommand(zeroId(), A, "refs/heads/master/x", CREATE));
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", false));
-
- assertResults(cmds, OK, OK);
- assertRefs("refs/heads/master/x", A);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertNull(getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(zeroId(), A, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/master/x"));
- }
-
- @Test
- public void reflogOnLockFailure() throws IOException {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- Map<String, ReflogEntry> oldLogs =
- getLastReflogs("refs/heads/master", "refs/heads/branch");
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(A, B, "refs/heads/branch", UPDATE));
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", false));
-
- if (atomic) {
- assertResults(cmds, TRANSACTION_ABORTED, LOCK_FAILURE);
- assertEquals(initialRefsChangedEvents, refsChangedEvents);
- assertReflogUnchanged(oldLogs, "refs/heads/master");
- assertReflogUnchanged(oldLogs, "refs/heads/branch");
- } else {
- assertResults(cmds, OK, LOCK_FAILURE);
- assertEquals(initialRefsChangedEvents + 1, refsChangedEvents);
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/master"));
- assertReflogUnchanged(oldLogs, "refs/heads/branch");
- }
- }
-
- @Test
- public void overrideRefLogMessage() throws Exception {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- cmds.get(0).setRefLogMessage("custom log", false);
- PersonIdent ident = new PersonIdent(diskRepo);
- execute(
- newBatchUpdate(cmds)
- .setRefLogIdent(ident)
- .setRefLogMessage("a reflog", true));
-
- assertResults(cmds, OK, OK);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertReflogEquals(
- reflog(A, B, ident, "custom log"),
- getLastReflog("refs/heads/master"),
- true);
- assertReflogEquals(
- reflog(zeroId(), B, ident, "a reflog: created"),
- getLastReflog("refs/heads/branch"),
- true);
- }
-
- @Test
- public void overrideDisableRefLog() throws Exception {
- writeRef("refs/heads/master", A);
- int initialRefsChangedEvents = refsChangedEvents;
-
- Map<String, ReflogEntry> oldLogs =
- getLastReflogs("refs/heads/master", "refs/heads/branch");
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- cmds.get(0).disableRefLog();
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", true));
-
- assertResults(cmds, OK, OK);
- assertEquals(atomic ? initialRefsChangedEvents + 1
- : initialRefsChangedEvents + 2, refsChangedEvents);
- assertReflogUnchanged(oldLogs, "refs/heads/master");
- assertReflogEquals(
- reflog(zeroId(), B, new PersonIdent(diskRepo), "a reflog: created"),
- getLastReflog("refs/heads/branch"));
- }
-
- @Test
- public void refLogNotWrittenWithoutConfigOption() throws Exception {
- setLogAllRefUpdates(false);
- writeRef("refs/heads/master", A);
-
- Map<String, ReflogEntry> oldLogs =
- getLastReflogs("refs/heads/master", "refs/heads/branch");
- assertTrue(oldLogs.isEmpty());
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", false));
-
- assertResults(cmds, OK, OK);
- assertReflogUnchanged(oldLogs, "refs/heads/master");
- assertReflogUnchanged(oldLogs, "refs/heads/branch");
- }
-
- @Test
- public void forceRefLogInUpdate() throws Exception {
- setLogAllRefUpdates(false);
- writeRef("refs/heads/master", A);
- assertTrue(
- getLastReflogs("refs/heads/master", "refs/heads/branch").isEmpty());
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- execute(
- newBatchUpdate(cmds)
- .setRefLogMessage("a reflog", false)
- .setForceRefLog(true));
-
- assertResults(cmds, OK, OK);
- assertReflogEquals(
- reflog(A, B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/master"));
- assertReflogEquals(
- reflog(zeroId(), B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/branch"));
- }
-
- @Test
- public void forceRefLogInCommand() throws Exception {
- setLogAllRefUpdates(false);
- writeRef("refs/heads/master", A);
-
- Map<String, ReflogEntry> oldLogs =
- getLastReflogs("refs/heads/master", "refs/heads/branch");
- assertTrue(oldLogs.isEmpty());
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
- cmds.get(1).setForceRefLog(true);
- execute(newBatchUpdate(cmds).setRefLogMessage("a reflog", false));
-
- assertResults(cmds, OK, OK);
- assertReflogUnchanged(oldLogs, "refs/heads/master");
- assertReflogEquals(
- reflog(zeroId(), B, new PersonIdent(diskRepo), "a reflog"),
- getLastReflog("refs/heads/branch"));
- }
-
- @Test
- public void packedRefsLockFailure() throws Exception {
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
-
- LockFile myLock = refdir.lockPackedRefs();
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertFalse(getLockFile("refs/heads/master").exists());
- assertFalse(getLockFile("refs/heads/branch").exists());
-
- if (atomic) {
- assertResults(cmds, LOCK_FAILURE, TRANSACTION_ABORTED);
- assertRefs("refs/heads/master", A);
- } else {
- // Only operates on loose refs, doesn't care that packed-refs is locked.
- assertResults(cmds, OK, OK);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch", B);
- }
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void packedRefsLockFailureRefsChangedEvents() throws Exception {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
-
- LockFile myLock = refdir.lockPackedRefs();
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(
- atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 2,
- refsChangedEvents);
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void oneRefLockFailure() throws Exception {
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE),
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE));
-
- LockFile myLock = new LockFile(refdir.fileFor("refs/heads/master"));
- assertTrue(myLock.lock());
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertFalse(LockFile.getLockFile(refdir.packedRefsFile).exists());
- assertFalse(getLockFile("refs/heads/branch").exists());
-
- if (atomic) {
- assertResults(cmds, TRANSACTION_ABORTED, LOCK_FAILURE);
- assertRefs("refs/heads/master", A);
- } else {
- assertResults(cmds, OK, LOCK_FAILURE);
- assertRefs(
- "refs/heads/branch", B,
- "refs/heads/master", A);
- }
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void oneRefLockFailureRefsChangedEvents() throws Exception {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE),
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE));
-
- LockFile myLock = new LockFile(refdir.fileFor("refs/heads/master"));
- assertTrue(myLock.lock());
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(
- atomic ? initialRefsChangedEvents
- : initialRefsChangedEvents + 1,
- refsChangedEvents);
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void singleRefUpdateDoesNotRequirePackedRefsLock() throws Exception {
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE));
-
- LockFile myLock = refdir.lockPackedRefs();
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertFalse(getLockFile("refs/heads/master").exists());
- assertResults(cmds, OK);
- assertRefs("refs/heads/master", B);
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void singleRefUpdateDoesNotRequirePackedRefsLockRefsChangedEvents()
- throws Exception {
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays
- .asList(new ReceiveCommand(A, B, "refs/heads/master", UPDATE));
-
- LockFile myLock = refdir.lockPackedRefs();
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
-
- assertEquals(initialRefsChangedEvents + 1, refsChangedEvents);
- } finally {
- myLock.unlock();
- }
- }
-
- @Test
- public void atomicUpdateRespectsInProcessLock() throws Exception {
- assumeTrue(atomic);
-
- writeLooseRef("refs/heads/master", A);
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
-
- Thread t = new Thread(() -> {
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
- } catch (Exception e) {
- throw new RuntimeException(e);
- }
- });
-
- ReentrantLock l = refdir.inProcessPackedRefsLock;
- l.lock();
- try {
- t.start();
- long timeoutSecs = 10;
- long startNanos = System.nanoTime();
-
- // Hold onto the lock until we observe the worker thread has attempted to
- // acquire it.
- while (l.getQueueLength() == 0) {
- long elapsedNanos = System.nanoTime() - startNanos;
- assertTrue(
- "timed out waiting for work thread to attempt to acquire lock",
- NANOSECONDS.toSeconds(elapsedNanos) < timeoutSecs);
- Thread.sleep(3);
- }
-
- // Once we unlock, the worker thread should finish the update promptly.
- l.unlock();
- t.join(SECONDS.toMillis(timeoutSecs));
- assertFalse(t.isAlive());
- } finally {
- if (l.isHeldByCurrentThread()) {
- l.unlock();
- }
- }
-
- assertResults(cmds, OK, OK);
- assertEquals(2, refsChangedEvents);
- assertRefs(
- "refs/heads/master", B,
- "refs/heads/branch", B);
- }
-
- @Test
- public void atomicUpdateRespectsInProcessLockRefsChangedEvents()
- throws Exception {
- assumeTrue(atomic);
-
- writeLooseRef("refs/heads/master", A);
- refdir.exactRef("refs/heads/master");
- int initialRefsChangedEvents = refsChangedEvents;
-
- List<ReceiveCommand> cmds = Arrays.asList(
- new ReceiveCommand(A, B, "refs/heads/master", UPDATE),
- new ReceiveCommand(zeroId(), B, "refs/heads/branch", CREATE));
-
- Thread t = new Thread(() -> {
- try {
- execute(newBatchUpdate(cmds).setAllowNonFastForwards(true));
- } catch (Exception e) {
- throw new RuntimeException(e);
- }
- });
-
- ReentrantLock l = refdir.inProcessPackedRefsLock;
- l.lock();
- try {
- t.start();
- long timeoutSecs = 10;
-
- // Hold onto the lock until we observe the worker thread has
- // attempted to
- // acquire it.
- while (l.getQueueLength() == 0) {
- Thread.sleep(3);
- }
-
- // Once we unlock, the worker thread should finish the update
- // promptly.
- l.unlock();
- t.join(SECONDS.toMillis(timeoutSecs));
- } finally {
- if (l.isHeldByCurrentThread()) {
- l.unlock();
- }
- }
-
- assertEquals(initialRefsChangedEvents + 1, refsChangedEvents);
- }
-
- private void setLogAllRefUpdates(boolean enable) throws Exception {
- StoredConfig cfg = diskRepo.getConfig();
- cfg.load();
- cfg.setBoolean(ConfigConstants.CONFIG_CORE_SECTION, null,
- ConfigConstants.CONFIG_KEY_LOGALLREFUPDATES, enable);
- cfg.save();
- }
-
- private void writeLooseRef(String name, AnyObjectId id) throws IOException {
- write(new File(diskRepo.getDirectory(), name), id.name() + "\n");
- }
-
- private void writeRef(String name, AnyObjectId id) throws IOException {
- RefUpdate u = diskRepo.updateRef(name);
- u.setRefLogMessage(getClass().getSimpleName(), false);
- u.setForceUpdate(true);
- u.setNewObjectId(id);
- RefUpdate.Result r = u.update();
- switch (r) {
- case NEW:
- case FORCED:
- return;
- default:
- throw new IOException("Got " + r + " while updating " + name);
- }
- }
-
- private BatchRefUpdate newBatchUpdate(List<ReceiveCommand> cmds) {
- BatchRefUpdate u = refdir.newBatchUpdate();
- if (atomic) {
- assertTrue(u.isAtomic());
- } else {
- u.setAtomic(false);
- }
- u.addCommand(cmds);
- return u;
- }
-
- private void execute(BatchRefUpdate u) throws IOException {
- execute(u, false);
- }
-
- private void execute(BatchRefUpdate u, boolean strictWork) throws IOException {
- try (RevWalk rw = new RevWalk(diskRepo)) {
- u.execute(rw,
- strictWork ? new StrictWorkMonitor() : NullProgressMonitor.INSTANCE);
- }
- }
-
- private void assertRefs(Object... args) throws IOException {
- if (args.length % 2 != 0) {
- throw new IllegalArgumentException(
- "expected even number of args: " + Arrays.toString(args));
- }
-
- Map<String, AnyObjectId> expected = new LinkedHashMap<>();
- for (int i = 0; i < args.length; i += 2) {
- expected.put((String) args[i], (AnyObjectId) args[i + 1]);
- }
-
- Map<String, Ref> refs = refdir.getRefs(RefDatabase.ALL);
- Ref actualHead = refs.remove(Constants.HEAD);
- if (actualHead != null) {
- String actualLeafName = actualHead.getLeaf().getName();
- assertEquals(
- "expected HEAD to point to refs/heads/master, got: " + actualLeafName,
- "refs/heads/master", actualLeafName);
- AnyObjectId expectedMaster = expected.get("refs/heads/master");
- assertNotNull("expected master ref since HEAD exists", expectedMaster);
- assertEquals(expectedMaster, actualHead.getObjectId());
- }
-
- Map<String, AnyObjectId> actual = new LinkedHashMap<>();
- refs.forEach((n, r) -> actual.put(n, r.getObjectId()));
-
- assertEquals(expected.keySet(), actual.keySet());
- actual.forEach((n, a) -> assertEquals(n, expected.get(n), a));
- }
-
- enum Result {
- OK(ReceiveCommand.Result.OK),
- LOCK_FAILURE(ReceiveCommand.Result.LOCK_FAILURE),
- REJECTED_NONFASTFORWARD(ReceiveCommand.Result.REJECTED_NONFASTFORWARD),
- REJECTED_MISSING_OBJECT(ReceiveCommand.Result.REJECTED_MISSING_OBJECT),
- TRANSACTION_ABORTED(ReceiveCommand::isTransactionAborted);
-
- @SuppressWarnings("ImmutableEnumChecker")
- final Predicate<? super ReceiveCommand> p;
-
- private Result(Predicate<? super ReceiveCommand> p) {
- this.p = p;
- }
-
- private Result(ReceiveCommand.Result result) {
- this(c -> c.getResult() == result);
- }
- }
-
- private void assertResults(
- List<ReceiveCommand> cmds, Result... expected) {
- if (expected.length != cmds.size()) {
- throw new IllegalArgumentException(
- "expected " + cmds.size() + " result args");
- }
- for (int i = 0; i < cmds.size(); i++) {
- ReceiveCommand c = cmds.get(i);
- Result r = expected[i];
- assertTrue(
- String.format(
- "result of command (%d) should be %s: %s %s%s",
- Integer.valueOf(i), r, c,
- c.getResult(),
- c.getMessage() != null ? " (" + c.getMessage() + ")" : ""),
- r.p.test(c));
- }
- }
-
- private Map<String, ReflogEntry> getLastReflogs(String... names)
- throws IOException {
- Map<String, ReflogEntry> result = new LinkedHashMap<>();
- for (String name : names) {
- ReflogEntry e = getLastReflog(name);
- if (e != null) {
- result.put(name, e);
- }
- }
- return result;
- }
-
- private ReflogEntry getLastReflog(String name) throws IOException {
- ReflogReader r = diskRepo.getReflogReader(name);
- if (r == null) {
- return null;
- }
- return r.getLastEntry();
- }
-
- private File getLockFile(String refName) {
- return LockFile.getLockFile(refdir.fileFor(refName));
- }
-
- private void assertReflogUnchanged(
- Map<String, ReflogEntry> old, String name) throws IOException {
- assertReflogEquals(old.get(name), getLastReflog(name), true);
- }
-
- private static void assertReflogEquals(
- ReflogEntry expected, ReflogEntry actual) {
- assertReflogEquals(expected, actual, false);
- }
-
- private static void assertReflogEquals(
- ReflogEntry expected, ReflogEntry actual, boolean strictTime) {
- if (expected == null) {
- assertNull(actual);
- return;
- }
- assertNotNull(actual);
- assertEquals(expected.getOldId(), actual.getOldId());
- assertEquals(expected.getNewId(), actual.getNewId());
- if (strictTime) {
- assertEquals(expected.getWho(), actual.getWho());
- } else {
- assertEquals(expected.getWho().getName(), actual.getWho().getName());
- assertEquals(
- expected.getWho().getEmailAddress(),
- actual.getWho().getEmailAddress());
- }
- assertEquals(expected.getComment(), actual.getComment());
- }
-
- private static ReflogEntry reflog(ObjectId oldId, ObjectId newId,
- PersonIdent who, String comment) {
- return new ReflogEntry() {
- @Override
- public ObjectId getOldId() {
- return oldId;
- }
-
- @Override
- public ObjectId getNewId() {
- return newId;
- }
-
- @Override
- public PersonIdent getWho() {
- return who;
- }
-
- @Override
- public String getComment() {
- return comment;
- }
-
- @Override
- public CheckoutEntry parseCheckout() {
- throw new UnsupportedOperationException();
- }
- };
- }
- }
|