/*
    * Copyright (C) 2019, Google LLC. 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.transport;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertTrue;
  
  import java.util.Arrays;
  import java.util.Collection;
  
  import org.eclipse.jgit.lib.ObjectId;
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.junit.runners.JUnit4;
  
  @RunWith(JUnit4.class)
  public class PreUploadHookChainTest {
  
  	@Test
  	public void testDefaultIfEmpty() {
  		PreUploadHook[] noHooks = {};
  		PreUploadHook newChain = PreUploadHookChain
  				.newChain(Arrays.asList(noHooks));
  		assertEquals(newChain, PreUploadHook.NULL);
  	}
  
  	@Test
  	public void testFlattenChainIfOnlyOne() {
  		FakePreUploadHook hook1 = new FakePreUploadHook();
  		PreUploadHook newChain = PreUploadHookChain
  				.newChain(Arrays.asList(PreUploadHook.NULL, hook1));
  		assertEquals(newChain, hook1);
  	}
  
  	@Test
  	public void testMultipleHooks() throws ServiceMayNotContinueException {
  		FakePreUploadHook hook1 = new FakePreUploadHook();
  		FakePreUploadHook hook2 = new FakePreUploadHook();
  
  		PreUploadHook chained = PreUploadHookChain
  				.newChain(Arrays.asList(hook1, hook2));
  		chained.onBeginNegotiateRound(null, null, 0);
  
  		assertTrue(hook1.wasInvoked());
  		assertTrue(hook2.wasInvoked());
  	}
  
  	private static final class FakePreUploadHook implements PreUploadHook {
  		boolean invoked;
  
  		@Override
  		public void onBeginNegotiateRound(UploadPack up,
  				Collection<? extends ObjectId> wants, int cntOffered)
  				throws ServiceMayNotContinueException {
  			invoked = true;
  		}
  
  		@Override
  		public void onEndNegotiateRound(UploadPack up,
  				Collection<? extends ObjectId> wants, int cntCommon,
  				int cntNotFound, boolean ready)
  				throws ServiceMayNotContinueException {
  			throw new UnsupportedOperationException();
  		}
  
  		@Override
  		public void onSendPack(UploadPack up,
  				Collection<? extends ObjectId> wants,
  				Collection<? extends ObjectId> haves)
  				throws ServiceMayNotContinueException {
  			throw new UnsupportedOperationException();
  		}
  
  		public boolean wasInvoked() {
  			return invoked;
  		}
  	}
  }