Tools/Castle.DynamicProxy2/Castle.DynamicProxy.Tests/GenerationHookTestCase.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.Collections;
  19         using System.Reflection;
  20         using Castle.DynamicProxy.Tests.Classes;
  21         using Castle.DynamicProxy.Tests.Interceptors;
  22         using Castle.DynamicProxy.Tests.InterClasses;
  23         using NUnit.Framework;
  24
  25         [TestFixture]
  26         public class GenerationHookTestCase : BasePEVerifyTestCase
  27         {
  28                 [Test]
  29                 public void HookIsUsedForConcreteClassProxy()
  30                 {
  31                         LogInvocationInterceptor logger = new LogInvocationInterceptor();
  32                         LogHook hook = new LogHook(typeof(ServiceClass), true);
  33
  34                         ProxyGenerationOptions options = new ProxyGenerationOptions(hook);
  35
  36                         ServiceClass proxy = (ServiceClass) generator.CreateClassProxy(
  37                                                                 typeof(ServiceClass), options, logger);
  38
  39                         Assert.IsTrue(hook.Completed);
  40                         Assert.AreEqual(10, hook.AskedMembers.Count);
  41                         Assert.AreEqual(2, hook.NonVirtualMembers.Count);
  42
  43                         proxy.Sum(1, 2);
  44                         Assert.IsFalse(proxy.Valid);
  45
  46                         Assert.AreEqual("get_Valid ", logger.LogContents);
  47                 }
  48
  49                 [Test]
  50                 public void HookIsUsedForInterfaceProxy()
  51                 {
  52                         LogInvocationInterceptor logger = new LogInvocationInterceptor();
  53                         LogHook hook = new LogHook(typeof(IService), false);
  54
  55                         ProxyGenerationOptions options = new ProxyGenerationOptions(hook);
  56
  57                         IService proxy = (IService)
  58                                          generator.CreateInterfaceProxyWithTarget(
  59                                                 typeof(IService), new ServiceImpl(), options, logger);
  60
  61                         Assert.IsTrue(hook.Completed);
  62                         Assert.AreEqual(10, hook.AskedMembers.Count);
  63                         Assert.AreEqual(0, hook.NonVirtualMembers.Count);
  64
  65                         Assert.AreEqual(3, proxy.Sum(1, 2));
  66                         Assert.IsFalse(proxy.Valid);
  67
  68                         Assert.AreEqual("Sum get_Valid ", logger.LogContents);
  69                 }
  70
  71                 public class LogHook : IProxyGenerationHook
  72                 {
  73                         private readonly Type targetTypeToAssert;
  74                         private readonly bool screeningEnabled;
  75                         private IList nonVirtualMembers = new ArrayList();
  76                         private IList askedMembers = new ArrayList();
  77                         private bool completed;
  78
  79                         public LogHook(Type targetTypeToAssert, bool screeningEnabled)
  80                         {
  81                                 this.targetTypeToAssert = targetTypeToAssert;
  82                                 this.screeningEnabled = screeningEnabled;
  83                         }
  84
  85                         public IList NonVirtualMembers
  86                         {
  87                                 get { return nonVirtualMembers; }
  88                         }
  89
  90                         public IList AskedMembers
  91                         {
  92                                 get { return askedMembers; }
  93                         }
  94
  95                         public bool Completed
  96                         {
  97                                 get { return completed; }
  98                         }
  99
 100                         public bool ShouldInterceptMethod(Type type, MethodInfo memberInfo)
 101                         {
 102                                 Assert.AreEqual(targetTypeToAssert, type);
 103
 104                                 askedMembers.Add(memberInfo);
 105
 106                                 if (screeningEnabled && memberInfo.Name.StartsWith("Sum"))
 107                                 {
 108                                         return false;
 109                                 }
 110
 111                                 return true;
 112                         }
 113
 114                         public void NonVirtualMemberNotification(Type type, MemberInfo memberInfo)
 115                         {
 116                                 Assert.AreEqual(targetTypeToAssert, type);
 117
 118                                 nonVirtualMembers.Add(memberInfo);
 119                         }
 120
 121                         public void MethodsInspected()
 122                         {
 123                                 completed = true;
 124                         }
 125                 }
 126         }
 127 }