1 /* 2 * Copyright 2016, Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above 11 * copyright notice, this list of conditions and the following disclaimer 12 * in the documentation and/or other materials provided with the 13 * distribution. 14 * 15 * * Neither the name of Google Inc. nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 package com.google.auth.oauth2; 33 34 import com.google.auth.RequestMetadataCallback; 35 import com.google.common.base.Preconditions; 36 import java.util.List; 37 import java.util.Map; 38 import java.util.concurrent.CountDownLatch; 39 40 /** Mock RequestMetadataCallback */ 41 public final class MockRequestMetadataCallback implements RequestMetadataCallback { 42 volatile Map<String, List<String>> metadata; 43 volatile Throwable exception; 44 CountDownLatch latch = new CountDownLatch(1); 45 46 /** Called when metadata is successfully produced. */ 47 @Override onSuccess(Map<String, List<String>> metadata)48 public void onSuccess(Map<String, List<String>> metadata) { 49 checkNotSet(); 50 this.metadata = metadata; 51 latch.countDown(); 52 } 53 54 /** Called when metadata generation failed. */ 55 @Override onFailure(Throwable exception)56 public void onFailure(Throwable exception) { 57 checkNotSet(); 58 this.exception = exception; 59 latch.countDown(); 60 } 61 reset()62 public void reset() { 63 this.metadata = null; 64 this.exception = null; 65 latch = new CountDownLatch(1); 66 } 67 awaitResult()68 public Map<String, List<String>> awaitResult() throws Throwable { 69 latch.await(); 70 if (exception != null) { 71 throw exception; 72 } else { 73 return metadata; 74 } 75 } 76 checkNotSet()77 private void checkNotSet() { 78 Preconditions.checkState(this.metadata == null); 79 Preconditions.checkState(this.exception == null); 80 } 81 } 82