您的位置：首页 > 其它

基于Entity Framework 4.1实现一个适用于测试的MockDbContext(上)

2011-07-22 14:46 344 查看

声明：我最近在微软加拿大开发中心工作，这是我个人的博客，跟公司没有关系。如果你在我的博客里看到我推荐微软的产品，就权当广告好了。

我们在作一些CRUD相关的单元测试的时候，通常不会真的连接到数据库，而是写一个Mock的Repository，把Entity放到一个集合啊或者Hash table啊什么的里面。
最近用Entity Framework 4.1写点小项目，在写一个Mock的Repository的时候还走了些弯路，费了一些时间，在此把过程写出来，希望能帮大家节省一点时间。
项目中用的是Model First，在Database First模型中应该也适用，先设计完数据库模型，生成数据库和两个.tt文件——MTBDbContext.Context.tt和MTBDbContext.tt。第一个文件里是一个DbContext类，担任数据持久化操作；第二个文件里是实体类。这两个文件原来与数据库模型是在同一个项目中的，做了一些小动作，把它们分开了^_^。

根据这样的结构，写MockDbContext的思路是这样子的：以添加一个实体类为例，我们通常会写这样的代码：

public void AddBatch(Handbook handbook)


[code]{

dbContext.Set<Handbook>().Add(handbook);

dbContext.SaveChanges();

[/code]

.csharpcode, .csharpcode pre
{
font-size: small;
color: black;
font-family: consolas, "Courier New", courier, monospace;
background-color: #ffffff;
/*white-space: pre;*/
}
.csharpcode pre { margin: 0em; }
.csharpcode .rem { color: #008000; }
.csharpcode .kwrd { color: #0000ff; }
.csharpcode .str { color: #006080; }
.csharpcode .op { color: #0000c0; }
.csharpcode .preproc { color: #cc6633; }
.csharpcode .asp { background-color: #ffff00; }
.csharpcode .html { color: #800000; }
.csharpcode .attr { color: #ff0000; }
.csharpcode .alt
{
background-color: #f4f4f4;
width: 100%;
margin: 0em;
}
.csharpcode .lnum { color: #606060; }

如果直接调用EF4.1的DbContext，在调用 DbContext里的SaveChanges()，数据就会被固化到数据库里。但我们相信，只要数据能够在本地保存，通过DbContext，它就一定会存到数据库里，因此，测试时没有必要把数据库写到数据库中去，只要在本地进行验证。

综上，我们需要的MockDbContext只要满足两个条件：第一，SaveChanges()不把数据固化到数据库，而是存在本地；第二，可以在本地作数据验证。

我们首先来满足第一个条件。打开DbContext看一下：

public partial class MTBContainer : DbContext


[code]{

public MTBContainer()

: base("name=MTBContainer")

protected override void OnModelCreating(DbModelBuilder modelBuilder)

throw new UnintentionalCodeFirstException();

public DbSet<Handbook> Handbooks1 { get; set; }

public DbSet<Trip> Trips { get; set; }

// … more DbSet<T>...


public class DbContext : IDisposable, IObjectContextAdapter


[code]{

protected DbContext();

   //… more constructors

  // Other code ...

public virtual int SaveChanges();

public DbSet<TEntity> Set<TEntity>() where TEntity : class;

public DbSet Set(Type entityType);

  //...

[/code]

.csharpcode, .csharpcode pre
{
font-size: small;
color: black;
font-family: consolas, "Courier New", courier, monospace;
background-color: #ffffff;
/*white-space: pre;*/
}
.csharpcode pre { margin: 0em; }
.csharpcode .rem { color: #008000; }
.csharpcode .kwrd { color: #0000ff; }
.csharpcode .str { color: #006080; }
.csharpcode .op { color: #0000c0; }
.csharpcode .preproc { color: #cc6633; }
.csharpcode .asp { background-color: #ffff00; }
.csharpcode .html { color: #800000; }
.csharpcode .attr { color: #ff0000; }
.csharpcode .alt
{
background-color: #f4f4f4;
width: 100%;
margin: 0em;
}
.csharpcode .lnum { color: #606060; }

里面有一个SaveChanges()方法，而且还是虚的。这样，事情就简单了（起初是这样认为的，也是从这里开始走的弯路）：重写一下SaveChanges()方法，让它什么都不做，它不就不会把数据存回数据库了吗？OK，第一个条件基本达成。

第二个条件，本地验证。由于实体类对象都保存在DbSet<T>里，而DbSet<T>里有一个ObservableCollection<T>类型的Local属性保存的正是本地实体类对象（这个理解有问题的）。这样，事情就好办了，我们写测试验证的时候直接验证这里的结果就行了。

按着这个思路，复制粘贴了一份MTBDbContext.Context.tt，改名为MockMTBDbContext.Context.tt并且放到了对应的单元测试项目中。然后，稍微修改了一下模板：添加了一些using的命名空间，改了生存的类名和构造函数名，然后就是重点添加一个什么都不做的SaveChanges()的重写方法——保存，自动生成代码如下：


... 


[code]public partial class MockMTBContainer : DbContext, ITestableDbContext

public MockMTBContainer()

: base("name=MTBContainer")

public override int SaveChanges()

//Do nothing

return 0;

...

public DbSet<Handbook> Handbooks1 { get; set; }

public DbSet<Trip> Trips { get; set; }

…


/// <summary>


[code]///A test for Add

///</summary>

[TestMethod()]

public void AddTest()

var mockDbContext = new MockMTBContainer();

BizHandbook target = new BizHandbook(mockDbContext); // use mockDbContext here.

Handbook handbook = new Handbook();

target.Add(handbook);

Assert.AreEqual<int>(1, mockDbContext.Handbooks1.Local.Count); // see whether the entity object's added

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航