org.eclipse.jgit.http.test/src/org/eclipse/jgit/http/test/RefsUnreadableInMemoryRepository.java

/*
 * Copyright (C) 2016, Google Inc. and others
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Distribution License v. 1.0 which is available at
 * https://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */
package org.eclipse.jgit.http.test;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.eclipse.jgit.internal.storage.dfs.DfsRepositoryDescription;
import org.eclipse.jgit.internal.storage.dfs.InMemoryRepository;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.RefDatabase;

/**
 * An {@link InMemoryRepository} whose refs can be made unreadable for testing
 * purposes.
 */
class RefsUnreadableInMemoryRepository extends InMemoryRepository {

	private final RefsUnreadableRefDatabase refs;

	private volatile boolean failing;

	RefsUnreadableInMemoryRepository(DfsRepositoryDescription repoDesc) {
		super(repoDesc);
		refs = new RefsUnreadableRefDatabase();
		failing = false;
	}

	@Override
	public RefDatabase getRefDatabase() {
		return refs;
	}

	/**
	 * Make the ref database unable to scan its refs.
	 * <p>
	 * It may be useful to follow a call to startFailing with a call to
	 * {@link RefDatabase#refresh()}, ensuring the next ref read fails.
	 */
	void startFailing() {
		failing = true;
	}

	private class RefsUnreadableRefDatabase extends MemRefDatabase {

		@Override
		public Ref exactRef(String name) throws IOException {
			if (failing) {
				throw new IOException("disk failed, no refs found");
			}
			return super.exactRef(name);
		}

		@Override
		public Map<String, Ref> getRefs(String prefix) throws IOException {
			if (failing) {
				throw new IOException("disk failed, no refs found");
			}

			return super.getRefs(prefix);
		}

		@Override
		public List<Ref> getRefsByPrefix(String prefix) throws IOException {
			if (failing) {
				throw new IOException("disk failed, no refs found");
			}

			return super.getRefsByPrefix(prefix);
		}

		@Override
		public List<Ref> getRefsByPrefixWithExclusions(String include, Set<String> excludes)
				throws IOException {
			if (failing) {
				throw new IOException("disk failed, no refs found");
			}

			return super.getRefsByPrefixWithExclusions(include, excludes);
		}

		@Override
		public Set<Ref> getTipsWithSha1(ObjectId id) throws IOException {
			if (failing) {
				throw new IOException("disk failed, no refs found");
			}
			return super.getTipsWithSha1(id);
		}
	}
}