Moq和参数属性继承

时间:2010-03-05 15:24:57

标签: c# inheritance attributes moq parameters

当我尝试使用Moq模拟一个类时,method属性会继承到mock类,但不会继承参数属性。

基本上行“让a =(ArgumentsAttribute)p.GetCustomAttributes(typeof(ArgumentsAttribute),true).SingleOrDefault()”不返回属性..运行代码以查看失败的位置。

如何让这个测试通过?

[TestFixture]
public class MyTests {
    [Test]
    public void shouldFindAndCallMethodWithAttributes() {
        var myInterface = new Mock<MyClass>();
        myInterface.Setup(x => x.MyMarkedMethod(1));
        myInterface.Setup(x => x.MyMarkedMethod(5));
        myInterface.Setup(x => x.MyMarkedMethod(9));

        var executor = new MarkedMethodExecutor();
        executor.FindAndCallMethodWithAttributes(myInterface.Object);

        myInterface.VerifyAll();
    }
}

public class MarkedMethodExecutor {
    public void FindAndCallMethodWithAttributes(object anObject) {
        var methods = from m in anObject.GetType().GetMethods()
                      where m.GetCustomAttributes(typeof (ExecuteMeAttribute), true).SingleOrDefault() != null
                      select m;
        foreach (var method in methods) {
            var callInfos = from p in method.GetParameters()
                            let a = (ArgumentsAttribute) p.GetCustomAttributes(typeof (ArgumentsAttribute), true).SingleOrDefault()
                            where a != null
                            select new {Parameter = p, Attribute = a};
            // assume its one argument here for simplicity..
            var attribute = callInfos.Single().Attribute;
            foreach (var argument in attribute.Arguments) {
                method.Invoke(anObject, new[] {argument});
            }
        }
    }
}

public class MyClass {
    [ExecuteMe]
    public virtual void MyMarkedMethod([Arguments(1, 5, 9)] int arg) {}
}

[AttributeUsage(AttributeTargets.Method, AllowMultiple = false, Inherited = true)]
public class ExecuteMeAttribute : Attribute {}

[AttributeUsage(AttributeTargets.Parameter, AllowMultiple = false, Inherited = true)]
public class ArgumentsAttribute : Attribute {
    public readonly object[] Arguments;

    public ArgumentsAttribute(params object[] arguments) {
        Arguments = arguments;
    }
}

2 个答案:

答案 0 :(得分:0)

我不明白你的问题:你想设置属性吗?

您可以使用It.is(匹配)功能。也许它有帮助: http://api.moq.me/html/5976987C.htm

答案 1 :(得分:0)

使用TypeDescriptor.AddAttributes方法进行预测会有效吗?我很想知道:

var myInterface = new Mock<IRepository>();
var attribute = new ExecuteMeAttribute();
someInstanceTypeDescriptor.AddAttributes(myInterface.Object, attribute);