用Sqlite的内存数据库对nhibernate进行单元测试
针对数据访问代码的单元测试处在一个尴尬的位置上,如果操作不是针对真实的数据库执行的,就无法捕获数据库特定的错误,比如 sql 语句语法是否正确,操作是否违反了数据库约束,事务是否正确提交。并且,测试之间应该是隔离的,一个测试不能影响另一个测试的数据,就是说,每个测试运行之前都要重建表结构,重新安装测试数据。在一个真实的数据库上执行这些操作会让测试成为老牛破车。
所幸的是Sqlite提供了内存数据库,避免磁盘IO可以带来性能提升。内存数据库有一个非常重要的特点:即数据库仅在连接打开的时候存在,一旦连接关闭,数据库随即消失。这正是我们想要的,运行测试的步骤如下:
1,在 [TestInitialize] 方法中打开 Session,创建表结构,安装测试数据。默认情况下,Sqlite的Unicode字符串比较是区分大小写的,所以,创建表结构的时候要为 TEXT 列指定 COLLATE NOCASE。
2,运行测试
3,在 [TestCleanup] 方法中关闭 Session,这将导致底层的连接关闭,内存数据库消失。
代码如下:
[TestClass] public class MyTest { [AssemblyInitialize] public static void InitAssembly(TestContext testContext) { NHibernateHelper.Init(); } [AssemblyCleanup] public static void CleanupAssembly() { NHibernateHelper.CloseSessionFactory(); } protected ISession Session { get; private set; } [TestInitialize()] public void MyTestInitialize() { try { // 1 打开 Session Session = NHibernateHelper.OpenSession(); // 2 创建表 NHibernate.Tool.hbm2ddl.SchemaExport export = new NHibernate.Tool.hbm2ddl.SchemaExport(NHibernateHelper.Configuration); export.SetDelimiter(";"); StringWriter sw = new StringWriter(); export.Execute(false, false, false, Session.Connection, sw); using (IDbCommand cmd = Session.Connection.CreateCommand()) { // 替换字段定义语句 cmd.CommandText = Regex.Replace(sw.ToString(), @"\s+TEXT\s+", " TEXT COLLATE NOCASE ", RegexOptions.IgnoreCase | RegexOptions.Compiled); cmd.ExecuteNonQuery(); } // 3 创建测试数据 using (ITransaction tx = Session.BeginTransaction()) { Role role = new Role(); role.Name = "admins"; Session.Save(role); tx.Commit(); } // 4 清除 Session 缓存 Session.Clear(); } catch (Exception ex) { // 如果发生异常,则 TestCleanup 不会执行,因此在这里回收资源 if (Session != null) { Session.Close(); } throw; } } [TestCleanup()] public void MyTestCleanup() { Session.Close(); Session = null; } [TestMethod] public void MyTestMethod() { using (ITransaction tx = Session.BeginTransaction()) { Role role = Session.Query<Role>().FirstOrDefault(x => x.Name == "admins"); Assert.IsNotNull(role); tx.Commit(); } } }
NHibernate 配置,注意 connection.release_mode 属性一定要设为 on_close
<hibernate-configuration xmlns="urn:nhibernate-configuration-2.2"> <bytecode-provider type="null"/> <reflection-optimizer use="false"/> <session-factory> <property name="connection.provider">NHibernate.Connection.DriverConnectionProvider</property> <property name="connection.driver_class">NHibernate.Driver.SQLite20Driver</property> <property name="connection.connection_string"> data source=:memory: </property> <property name="dialect">NHibernate.Dialect.SQLiteDialect</property> <property name="connection.release_mode">on_close</property> <property name="hbm2ddl.keywords">none</property> <property name="current_session_context_class">managed_web</property> <property name="show_sql">false</property> <property name="query.substitutions">true 1, false 0, yes 'Y', no 'N'</property> <property name="proxyfactory.factory_class">NHibernate.ByteCode.Castle.ProxyFactoryFactory, NHibernate.ByteCode.Castle</property> <mapping assembly="WuTong"/> </session-factory> </hibernate-configuration>
参考链接: