source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.repository.reporting;
   2
   3 /*
   4  * Copyright 2005-2006 The Apache Software Foundation.
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 import java.lang.reflect.InvocationHandler;
  20 import java.lang.reflect.Method;
  21 import java.util.HashMap;
  22 import java.util.List;
  23 import java.util.Map;
  24
  25 /**
  26  * @author Edwin Punzalan
  27  */
  28 public class GenericMockObject
  29     implements InvocationHandler
  30 {
  31     private Map invocations = new HashMap();
  32
  33     public GenericMockObject()
  34     {
  35         //default constructor
  36     }
  37
  38     public GenericMockObject( Map returnMap )
  39     {
  40         invocations = new HashMap( returnMap );
  41     }
  42
  43     public void setExpectedReturns( Method method, List returnList )
  44     {
  45         invocations.put( method, returnList );
  46     }
  47
  48     public Object invoke( Object proxy, Method method, Object[] args )
  49     {
  50         if ( !invocations.containsKey( method ) )
  51         {
  52             throw new UnsupportedOperationException( "No expected return values defined." );
  53         }
  54
  55         List returnList = (List) invocations.get( method );
  56         if ( returnList.size() < 1 )
  57         {
  58             throw new UnsupportedOperationException( "Too few expected return values defined." );
  59         }
  60         return returnList.remove( 0 );
  61     }
  62 }