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

解决的问题:方法内部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代码较多,测试时需要设标志。

 

其它的问题: 等待高人指出,谢谢!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值