Tools/Castle.DynamicProxy2/Castle.DynamicProxy.Tests/MethodsWithAttributesOnParametersTestCase.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.Tests
  16 {
  17         using System;
  18         using System.Reflection;
  19         using Castle.Core.Interceptor;
  20         using NUnit.Framework;
  21
  22         [TestFixture]
  23         public class MethodsWithAttributesOnParametersTestCase : BasePEVerifyTestCase
  24         {
  25                 [Test]
  26                 [ExpectedException(typeof(ArgumentException), "No default value for argument")]
  27                 public void ParametersAreCopiedToProxiedObject()
  28                 {
  29                         ClassWithAttributesOnMethodParameters requiredObj = (ClassWithAttributesOnMethodParameters)
  30                                                                             generator.CreateClassProxy(
  31                                                                                 typeof(ClassWithAttributesOnMethodParameters),
  32                                                                                 new RequiredParamInterceptor());
  33
  34                         requiredObj.MethodOne(-1);
  35                 }
  36
  37                 [Test]
  38                 public void CanGetParameterAttributeFromProxiedObject()
  39                 {
  40                         ClassWithAttributesOnMethodParameters requiredObj = (ClassWithAttributesOnMethodParameters)
  41                                                                             generator.CreateClassProxy(
  42                                                                                 typeof(ClassWithAttributesOnMethodParameters),
  43                                                                                 new RequiredParamInterceptor());
  44
  45                         requiredObj.MethodTwo(null);
  46                 }
  47         }
  48
  49         public class RequiredParamInterceptor : IInterceptor
  50         {
  51                 public void Intercept(IInvocation invocation)
  52                 {
  53                         ParameterInfo[] parameters = invocation.Method.GetParameters();
  54
  55                         object[] args = invocation.Arguments;
  56
  57                         for(int i = 0; i < parameters.Length; i++)
  58                         {
  59                                 if (parameters[i].IsDefined(typeof(RequiredAttribute), false))
  60                                 {
  61                                         RequiredAttribute required =
  62                                                 parameters[i].GetCustomAttributes(typeof(RequiredAttribute), false)[0] as RequiredAttribute;
  63
  64                                         if ((required.BadValue == null && args[i] == null) ||
  65                                             (required.BadValue != null && required.BadValue.Equals(args[i])))
  66                                         {
  67                                                 args[i] = required.DefaultValue;
  68                                         }
  69                                 }
  70                         }
  71
  72                         invocation.Proceed();
  73                 }
  74         }
  75
  76         public class ClassWithAttributesOnMethodParameters
  77         {
  78                 public virtual void MethodOne([Required(BadValue = -1)] int val)
  79                 {
  80                         Assert.IsFalse(val == -1);
  81                 }
  82
  83                 public virtual void MethodTwo([Required("")] string name)
  84                 {
  85                         Assert.IsNotNull(name);
  86                 }
  87         }
  88
  89         public class RequiredAttribute : Attribute
  90         {
  91                 private object defaultValue;
  92                 private bool hasDefault = false;
  93                 public object BadValue;
  94
  95                 public RequiredAttribute()
  96                 {
  97                 }
  98
  99                 public RequiredAttribute(object defaultValue)
 100                 {
 101                         hasDefault = true;
 102                         this.defaultValue = defaultValue;
 103                 }
 104
 105                 public object DefaultValue
 106                 {
 107                         get
 108                         {
 109                                 if (!hasDefault)
 110                                 {
 111                                         throw new ArgumentException("No default value for argument");
 112                                 }
 113                                 return defaultValue;
 114                         }
 115                 }
 116         }
 117 }