最近在公司项目中使用了Autofac,本人需要解决的问题比较特别,就是需要在单元测试时候替换实际处理的类,通过模拟数据库操作完成单元测试。
本人是个懒惰的程序员,因为不想修改autofac的配置文件,又要结合fakes进行模拟,在网上又未发现类似处理方法,所以特把自己的做法共享出来,供大家参考和改进。
因为实际代码为公司项目,未能尽列代码请见谅,同时认为应该还有更好的改进方案,请有更好想法的朋友分享。
部分参考代码
------------------------------------------------
DAO管理器:
/// <summary> /// Dao 管理器 /// </summary> public class DaoBag { #region 防实例静态化 /// <summary> /// 获取实例 /// </summary> public static DaoBag Instance { get { return Nested.instance; } } /// <summary> /// 密封的内部嵌套辅助类,辅助创建单例对象 /// </summary> sealed class Nested { // 静态钩子 static Nested() { } /// <summary> /// 只读类型 /// </summary> internal static readonly DaoBag instance = new DaoBag(); } // 防止直接实例化 private DaoBag() { builder.RegisterModule(new ConfigurationSettingsReader("autofac")); } #endregion ContainerBuilder builder = new ContainerBuilder(); IContainer container; public T GetDao<T>() { // 延迟加载,第一次使用时候被调用 if (container == null) container = builder.Build(); return container.Resolve<T>(); } public void SetDao<T,I>(T t) where T : class { builder.RegisterInstance<T>(t).As<I>(); } }
Config配置(无论单元测试还是实际项目中,配置文件不变,这也是我非常需要偷懒的地方):
<configuration> <configSections> <section name="autofac" type="Autofac.Configuration.SectionHandler, Autofac.Configuration"/> </configSections> <autofac defaultAssembly="Model"> <components> <component type="Model.Help.HelpNewsRepository, Model" service="Model.Help.IHelpNewsRepository" /> <component type="Model.Pri.PriUserRepository, Model" service="Model.Pri.IPriUserRepository" /> </components> </autofac> </configuration>
最终完成后实现的效果如下
------------------------------------------------------
业务类中使用方法:
private IHelpNewsRepository hnr = DaoBag.Instance.GetDao<IHelpNewsRepository>();
单元测试中替换方法(注意使用了Fakes,单元测试中的替换将使配置失效,其实不是单元测试和实际项目使用相同配置,应该说是单元测试不用配置):
// 准备测试数据 var news_id = 1; var news_viewNum = 2; // 准备伪造数据处理类 StubIHelpNewsRepository stubhnr = new StubIHelpNewsRepository(); StubIPriUserRepository stubpur = new StubIPriUserRepository(); // 模拟底层数据处理方法 stubhnr.GetEntityInt64 = (x) => { return new HelpNews { Title = "新闻测试", Id = 1, ViewNum = 1 }; }; stubpur.GetEntityInt64 = (x) => { return new PriUser { Id = 1, Username = "admin" }; }; // 为IOC容器设置新的Dao (注意:必须设置完所有相关Dao映射,才可以进行其他步骤) DaoBag.Instance.SetDao<StubIHelpNewsRepository, IHelpNewsRepository>(stubhnr); DaoBag.Instance.SetDao<StubIPriUserRepository, IPriUserRepository>(stubpur); //测试第一个业务类 var newbus = NewsBus.Instance; var result = newbus.GetNews(news_id).AppendData as HelpNews; Assert.AreEqual(news_id, result.Id); Assert.AreEqual("新闻测试", result.Title); Assert.AreEqual(news_viewNum, result.ViewNum); //测试第二个业务类 var userbus = UserBus.Instance; var user = userbus.GetUser(1).AppendData as PriUser; Assert.AreEqual(user.Username, "admin");