Tools/DynamicProxy/Castle.DynamicProxy.Tests/MethodInvocationTargetTestCase.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 NUnit.Framework;
  19         using Castle.DynamicProxy.Test.ClassInterfaces;
  20
  21         [TestFixture]
  22         public class MethodInvocationTargetTestCase
  23         {
  24                 [Test]
  25                 public void AttributesOnTargetClasses()
  26                 {
  27                         ProxyGenerator _generator = new ProxyGenerator();
  28
  29                         AttributeCheckerInterceptor interceptor = new AttributeCheckerInterceptor();
  30
  31                         object proxy = _generator.CreateClassProxy(typeof (MyInterfaceImpl), interceptor);
  32
  33                         IMyInterface inter = (IMyInterface) proxy;
  34
  35                         Assert.AreEqual(45, inter.Calc(20, 25));
  36                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  37                         Assert.IsTrue(interceptor.LastMethodHasAttribute);
  38
  39                         Assert.AreEqual(48, inter.Calc(20, 25, 1, 2));
  40                         Assert.IsTrue(interceptor.LastMethodHasAttribute);
  41                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  42
  43                         inter.Name = "hammett";
  44                         Assert.IsFalse(interceptor.LastMethodHasAttribute);
  45                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  46                 }
  47
  48                 [Test]
  49                 public void AttributesOnTargetClassesWithInterfaceProxy()
  50                 {
  51                         ProxyGenerator _generator = new ProxyGenerator();
  52
  53                         AttributeCheckerInterceptor interceptor = new AttributeCheckerInterceptor();
  54
  55                         object proxy = _generator.CreateProxy(typeof (IMyInterface), interceptor, new MyInterfaceImpl());
  56
  57                         IMyInterface inter = (IMyInterface) proxy;
  58
  59                         Assert.AreEqual(45, inter.Calc(20, 25));
  60                         Assert.IsTrue(interceptor.LastMethodHasAttribute);
  61                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  62
  63                         Assert.AreEqual(48, inter.Calc(20, 25, 1, 2));
  64                         Assert.IsTrue(interceptor.LastMethodHasAttribute);
  65                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  66
  67                         inter.Name = "hammett";
  68                         Assert.IsFalse(interceptor.LastMethodHasAttribute);
  69                         Assert.IsTrue(interceptor.LastTargethasAttribute);
  70                 }
  71
  72                 public class AttributeCheckerInterceptor : StandardInterceptor
  73                 {
  74                         bool lastMethodHasAttribute;
  75                         bool lastTargethasAttribute;
  76
  77                         protected override void PreProceed(IInvocation invocation, params object[] args)
  78                         {
  79                                 lastMethodHasAttribute = invocation.MethodInvocationTarget.IsDefined(typeof (MyAttribute), false);
  80                                 lastTargethasAttribute = invocation.InvocationTarget.GetType().IsDefined(typeof (MyAttribute), true);
  81                         }
  82
  83                         public bool LastMethodHasAttribute
  84                         {
  85                                 get { return lastMethodHasAttribute; }
  86                         }
  87
  88                         public bool LastTargethasAttribute
  89                         {
  90                                 get { return lastTargethasAttribute; }
  91                         }
  92                 }
  93         }
  94 }