Tools/DynamicProxy/Castle.DynamicProxy.Tests/MethodsWithAttributesOnParameters.cs

   1 // Copyright 2004-2007 Castle Project - http://www.castleproject.org/
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 namespace Castle.DynamicProxy.Test
  16 {
  17         using System;
  18         using System.Reflection;
  19         using NUnit.Framework;
  20
  21         [TestFixture]
  22         public class MethodsWithAttributesOnParameters
  23         {
  24                 [Test]
  25                 [ExpectedException(typeof(ArgumentException),"No default value for argument")]
  26                 public void ParametersAreCopiedToProxiedObject()
  27                 {
  28                         ProxyGenerator pg = new ProxyGenerator();
  29                         ClassWithAttributesOnMethodParameters requiredObj = (ClassWithAttributesOnMethodParameters)pg.CreateClassProxy(typeof(ClassWithAttributesOnMethodParameters), new RequiredParamInterceptor());
  30                         requiredObj.MethodOne(-1);
  31                 }
  32
  33                 [Test]
  34                 public void CanGetParameterAttributeFromProxiedObject()
  35                 {
  36                         ProxyGenerator pg = new ProxyGenerator();
  37                         ClassWithAttributesOnMethodParameters requiredObj = (ClassWithAttributesOnMethodParameters)pg.CreateClassProxy(typeof(ClassWithAttributesOnMethodParameters), new RequiredParamInterceptor());
  38                         requiredObj.MethodTwo(null);
  39                 }
  40         }
  41
  42         public class RequiredParamInterceptor :IInterceptor
  43         {
  44                 public object Intercept(IInvocation invocation, params object[] args)
  45                 {
  46                         ParameterInfo[] parameters = invocation.Method.GetParameters();
  47                         for (int i = 0; i < parameters.Length; i++)
  48                         {
  49                                 if(parameters[i].IsDefined(typeof(RequiredAttribute),false))
  50                                 {
  51                                         RequiredAttribute required = parameters[i].GetCustomAttributes(typeof(RequiredAttribute),false)[0] as RequiredAttribute;
  52                                         if( (required.BadValue == null && args[i] == null) ||
  53                                                 (required.BadValue != null && required.BadValue.Equals(args[i])
  54                                                 ))
  55                                         {
  56                                                 args[i] = required.DefaultValue;
  57                                         }
  58                                 }
  59                         }
  60                         return invocation.Proceed(args);
  61                 }
  62         }
  63
  64         public class ClassWithAttributesOnMethodParameters
  65         {
  66                 public virtual void MethodOne([Required(BadValue=-1)]int val)
  67                 {
  68                         Assert.IsFalse(val==-1);
  69                 }
  70
  71                 public virtual void MethodTwo([Required("")]string name)
  72                 {
  73                         Assert.IsNotNull(name);
  74                 }
  75         }
  76
  77         public class RequiredAttribute : Attribute
  78         {
  79                 object defaultValue;
  80                 public object BadValue;
  81                 private bool hasDefault=false;
  82
  83                 public RequiredAttribute(){}
  84
  85                 public RequiredAttribute(object defaultValue)
  86                 {
  87                         hasDefault = true;
  88                         this.defaultValue = defaultValue;
  89                 }
  90
  91
  92                 public object DefaultValue
  93                 {
  94                         get
  95                         {
  96                                 if(!hasDefault)
  97                                         throw new ArgumentException("No default value for argument");
  98                                 return defaultValue;
  99                         }
 100                 }
 101
 102         }
 103 }