TDD Tip:方法内部New出来的对象如何Mock

来源：互联网发布：光翼网络班视频编辑：程序博客网时间：2024/05/21 07:55

--内容摘要：解决的问题：方法内部new的对象在测试时希望能够用mock对象去代替。问题：以下方法可以解决，但是很是丑陋，各位大侠是否有更好的方法？描述：如果说把内部的方法放到类的一个公开成员变量，或者放到方法的参数里，我的意见是公开了不应该公开的东西。使用第三方的类库: Mock工具 Rhino.Mocks， IOC: Castle.Windsor1. 现在我又这么一段代码,我想测试Math内部这两个方法&n。。。

转 http://www.pin5i.com/showtopic-21589.html

解决的问题：方法内部new的对象在测试时希望能够用mock对象去代替。

问题：以下方法可以解决，但是很是丑陋，各位大侠是否有更好的方法？

描述：如果说把内部的方法放到类的一个公开成员变量，或者放到方法的参数里，我的意见是公开了不应该公开的东西。

使用第三方的类库: Mock工具 Rhino.Mocks， IOC: Castle.Windsor

1. 现在我又这么一段代码,我想测试Math内部这两个方法

public class Math
{
public long MathAdd(int a, int b)
{
Calc c = new Calc();
return c.Add(a, b);
}
public long MathAdd2Price(int a)
{
Calc c = new Calc();
return c.AddCount(a) + c.count;
}
}
public class Calc
{
public int count = 0;
public long Add(int a, int b)
{
return a + b;
}
public long AddCount(int num)
{
return count + num;
}
}

复制代码

2. 我们看到，由于是在内部new的对象，我们就对Calc类产生了很大的依赖，于是我想到注入一个对像，下面是我的设计

public class ContainerFactory
{
public static IWindsorContainer container;
public bool IsDebug = false;
private static readonly ContainerFactory instance = new ContainerFactory();
public static ContainerFactory Instance
{
get
{
return instance;
}
}
private ContainerFactory()
{
AddAllCompent();
}
private void AddAllCompent()
{
if (container == null)
{
container = new WindsorContainer();
//此处职位演示，未使用接口
container.AddComponentWithLifestyle<Calc>("Calc", LifestyleType.Transient); }
}
}

复制代码

这样我们的Math类可改成这样

public class Math
{
public long MathAdd(int a, int b)
{
// Calc c = new Calc();
Calc c =(Calc) ContainerFactory.container["Calc"];
return c.Add(a, b);
}
public long MathAdd2Price(int a)
{
Calc c = (Calc)ContainerFactory.container["Calc"];
return c.AddCount(a) + c.count;
}
}

复制代码

3. 但是，我们如何在测试中用我们Mock的对象代替真实的对象呢？下面是我想的一个自己也认为不好的方法，但能凑活着用

public class ContainerFactory
{
private IWindsorContainer container;
public bool IsDebug = false;
private static readonly ContainerFactory instance = new ContainerFactory();
public static ContainerFactory Instance
{
get
{
return instance;
}
}
private ContainerFactory()
{
AddAllCompent();
}
private void AddAllCompent()
{
if (container == null)
{
container = new WindsorContainer();
//此处职位演示，未使用接口
container.AddComponentWithLifestyle<Calc>("Calc", LifestyleType.Transient); }
}
#region Calc
private Calc DebugCalc;
public Calc Calc
{
get
{
if (IsDebug && DebugCalc != null)
{
return DebugCalc;
}
else
{
return (Calc)container["Calc"];
}
}
set
{
// just for test, for mock object
if (IsDebug)
{
DebugCalc = value;
}
else
{
throw new Exception("just for test");
}
}
}
#endregion
}

复制代码

修改我们的类

public class Math
{
public long MathAdd(int a, int b)
{
// Calc c = new Calc();
Calc c = ContainerFactory.Instance.Calc;
return c.Add(a, b);
}
public long MathAdd2Price(int a)
{
Calc c = ContainerFactory.Instance.Calc;
return c.AddCount(a) + c.count;
}
}

复制代码

这样我们来看我们通过了测试的代码

[TestClass()]
public class MathTest
{
[TestMethod()]
public void MathAddTest()
{
Math m = new Math();
// 想让真实代码内部，使用的是Mock的对象
// Arrange
MockRepository mocks = new MockRepository();
Calc mockCalc = mocks.Stub<Calc>();
mockCalc.count = 5;
ContainerFactory.Instance.IsDebug = true; //这句很重要
ContainerFactory.Instance.Calc = mockCalc;
mocks.ReplayAll();
// Act
Assert.AreEqual(m.MathAdd(5, 5), 10);
Assert.AreEqual(mockCalc.AddCount(6), 11);
mocks.VerifyAll();
Calc mockCalc2 = mocks.Stub<Calc>();
mocks.ReplayAll();
// 这里有问题，我们希望他是7,但实际是12，因为需要测试中的和实际代码用同一个对象，
// 他保留上次的状态count的值5
Assert.AreEqual(mockCalc.AddCount(7), 12);
mocks.VerifyAll();
}
[TestMethod]
public void MathAddTestActual()
{
//这里测试实际使用代码，没用Mock
Math m = new Math();
Assert.AreEqual(m.MathAdd(5,6), 11);
Assert.AreEqual(m.MathAdd2Price(9), 9);
Calc c = ContainerFactory.Instance.Calc;
Assert.AreEqual(c.AddCount(5), 5);
c.count = 20;
Assert.AreEqual(c.AddCount(5), 25);
Calc d = ContainerFactory.Instance.Calc;
Assert.AreEqual(d.AddCount(30), 30);
Assert.AreEqual(c.count, 20);
Assert.AreEqual(d.count, 0);
}
}

复制代码

总结：这样可以不使用用public的类成员变量，不用通过方法参数注入注入对象

问题: ContainerFactory代码较多，测试时需要设标志。