.net单元测试——常用测试方式（异常模拟、返回值测试、参数测试、数据库访问代码测试）

最近在看.net单元测试艺术，我也喜欢单元测试，今天介绍一下如何测试异常、如何测试返回值、如何测试模拟对象的参数传递、如何测试数据库访问代码。单元测试框架使用的是NUnit，模拟框架使用的是：Rhino.Mocks。

１．测试异常，可以直接对方法进行异常测试，也可以对模拟对象进行异常测试，但是，对模拟对象进行异常测试，很少用，所以，这里就介绍对方法的异常测试。请看如下代码，当用户名为空的时候，抛出异常。

　　　  public bool Valid(string userName, string passWord)
        {
            if (string.IsNullOrEmpty(userName)) throw new ArgumentNullException("userName is null");

            var isValid = userName == "admin" && passWord == "123456";

            Log.Write(userName);

            return isValid;
        }

测试代码如下：

　　  　　［Test]
       [ExpectedException(typeof(ArgumentNullException))]
       public void Vaild_Throw_Test()
        {
            MyLogin l = new MyLogin();

            l.Valid("", "123456");
        }

这里需要特性“ExpectedException”，意思是期待抛出一个异常。这里不需要再进行断言了，抛出异常后，后面的代码也就不会执行了。

２．测试返回值，这里主要讲测试模拟对象的返回值。这里用到了一个LastCall的一个类，比较常用，一些辅助测试的功能，都在该类中。测试代码如下：

　       [Test]
        public void Valid_Return()
        {
            MockRepository mock = new MockRepository();

            var log = mock.DynamicMock<ILog>();

            using (mock.Record())
            {
                log.WriteLog("admin");
                LastCall.Return(0);
            }
　　　　　　　 var returnValue = log.WriteLog("admin");

　　　　　 　　Assert.AreEqual(returnValue, 0);
        }　　

３．测试模拟对象的参数传递，当调用模拟对象时，可能需要传递参数，如果参数的值不一至时，会导致测试失败，比如：我需要传递一个字符串，其中包含了一个GUID+UserName，这里GUID可能我们无法模拟，所以，测试参数时，只需要测试UserName就可以。

方法代码如下：

　　  public bool Valid_Paramter(string userName, string passWord)
        {
            Log.Write(Guid.NewGuid() + userName);

            return userName == "admin" && passWord == "123456";
        }

测试代码如下：

　　　    [Test]
        public void Valid_Paramter()
        {
            MockRepository mock = new MockRepository();

            var log = mock.DynamicMock<ILog>();

            using (mock.Record())
            {
                log.Write("admin");

                LastCall.Constraints(Rhino.Mocks.Constraints.Text.Contains("admin"));
            }
　　　　　　  //这里代码如果不明白，可以看上一节，里面有说明。
            MyLogin login = new MyLogin();

            login.Log = log;

            var valid = login.Valid_Paramter("admin", "123456");

            Assert.AreEqual(valid, true);

            mock.VerifyAll();
        }

　　这里只是测试模拟对象的参数传递。

４．测试数据库访问代码，这里提供的方法是，执行完测试代码后，对数据库进行回滚。有两种实现方式，一种是使用Rollback特性，一种是使用TransactionScope类。其实，Rollback也使用TransactionScope类。

　　1)   Rollback特性

　　开始我以为Rollback是NUnit框架自带的一个特性，结果不是，需要自己编写代码，这里需要继承ITestAction接口，该接口包含BeforeTest方法和AfterTest方法。Rollback特性的代码如下：

  public class RollbackAttribute : Attribute, ITestAction
    {
        private TransactionScope transaction;

        public void BeforeTest(TestDetails testDetails)
        {
            transaction = new TransactionScope();
        }

        public void AfterTest(TestDetails testDetails)
        {
            transaction.Dispose();
        }

        public ActionTargets Targets
        {
            get { return ActionTargets.Test; }
        }
    }

　　下面是被测试的代码，这里我用了EF框架，代码非常简单，向TitleInfo表中，写入数据，该表中只有一个字段，就是title字段。如果写入成功，则返回值大于0.

　　 public int Insert(string title)
        {
            testEntities db = new testEntities();

            var titleInfo = new TitleInfo()
            {
                Title = title
            };

            db.TitleInfo.Add(titleInfo);

            return db.SaveChanges();
        }

　　使用Rollback特性进行测试。

        [Test]
        [Rollback]
        public void Test_Insert()
        {
            DBLibrary db = new DBLibrary();

            var count = db.Insert("admin");

            Assert.True(count > 0);
        }

　　 非常简单，加入Rollback特性即可，这里不会向数据库写入记录，但是测试会成功。

　　2)   使用TransactionScope类测试数据访问层，该类可以实现数据库的事务操作，执行测试代码后，会对数据库的事务进行回滚，达到测试的目的，而不需要修改数据库的数据。测试代码如下：

    [TestFixture]
    class DBLibraryTest
    {
        TransactionScope socp;

        [SetUp]
        public void Init()
        {
            socp = new TransactionScope();
        }

        [TearDown]
        public void Close()
        {
            socp.Dispose();
        }

        [Test]
        public void Test_Insert()
        {
            DBLibrary db = new DBLibrary();

            var count = db.Insert("admin");

            Assert.True(count > 0);
        }
    }

　　这里加入了TransactionScope类，在Test_Insert方法执行前，会先执行Init方法，执行完Test_Insert方法后，执行Close方法，对事务进行回滚。其实Rollback只是对TransactionScope进行了封装，通过NUnit框架，很方便的实现了数据库事务的回滚。

总结：这里只是总结了常用的测试方法，如果大家还有其它的常见方法，可以分享一下，大家共同学习。