復制內存中的SQLite數據庫以使單元測試更快

Question

在C＃/ nHibernate-projects中，我正在使用SQLite對我的代碼進行單元測試，大致使用此處描述的方法： http : //ayende.com/blog/3983/nhibernate-unit-testing 。

但是，我發現構建和配置內存數據庫通常需要大約150ms。 我有很多單元測試，所以很快就加起來了。

我想一次構建和配置數據庫，將其存儲在靜態變量中，並在每次單元測試需要數據庫時將其復制。

如何備份內存數據庫？

我首先嘗試創建一個命名的內存數據庫。 根據https://www.sqlite.org/inmemorydb.html，這是可能的。 我曾經有過：

    private const string ConnectionString = "Data Source=:memory:;Version=3;";

我嘗試的連接字符串是：

    private const string ConnectionString = "FullUri=file:memorydb.db?mode=memory&cache=shared";
    private const string ConnectionString2 = "FullUri=file:memorydb2.db?mode=memory&cache=shared";

因此，現在我只需要了解如何快速將內容從一個復制到另一個？ 我快到了：我可以創建兩個內存數據庫，然后調用“ BackupDatabase”來復制數據庫。

但是，單元測試的行為就像“實例”數據庫沒有表一樣，即使“原型”數據庫也沒有。

        private static ISessionFactory _prototypeSessionFactory;
        private const string InstanceConnectionString = "FullUri=file:memorydb.db?mode=memory&cache=shared";

        private const string PrototypeConnectionString = "FullUri=file:memorydb2.db?mode=memory&cache=shared";
        private SQLiteConnection _instanceConnection;
        private ISessionFactory _instanceSessionFactory;

        public DatabaseScope(Assembly assembly)
        {
            var prototyeConfiguration = SQLiteConfiguration.Standard.ConnectionString(PrototypeConnectionString);
            var cfg = Fluently
                .Configure()
                .Database(prototyeConfiguration)
                .Mappings(m => m.HbmMappings.AddFromAssembly(assembly));
            cfg.ExposeConfiguration(BuildSchema);
            _prototypeSessionFactory = cfg.BuildSessionFactory();

            var instanceConfiguration = SQLiteConfiguration.Standard.ConnectionString(InstanceConnectionString);
            _instanceSessionFactory = Fluently
                .Configure()
                .Database(instanceConfiguration)
                .BuildSessionFactory();

            CopyDatabase();
        }

        private void CopyDatabase()
        {
            var cnnIn = new SQLiteConnection(PrototypeConnectionString);
            var cnnOut = new SQLiteConnection(InstanceConnectionString);
            cnnIn.Open();
            cnnOut.Open();
            cnnIn.BackupDatabase(cnnOut, "main", "main", -1, null, -1);
            cnnIn.Close();
            cnnOut.Close();
        }

Answer 1

我最終得到了這個工作代碼。 我的單元測試時間從十多分鍾縮短到不到兩分鍾。 （為便於閱讀，代碼略有簡化）

using System;
using System.Data;
using System.Data.SQLite;
using System.IO;
using System.Reflection;
using FluentNHibernate.Cfg;
using FluentNHibernate.Cfg.Db;
using NHibernate;
using NHibernate.Mapping;
using NHibernate.Tool.hbm2ddl;

namespace TestHelper.DbHelper.SqLite
{
    public class DatabaseScope : IDisposable
    {
        private static Assembly _prototypeAssembly;
        private const string PrototypeConnectionString = "FullUri=file:prototype.db?mode=memory&cache=shared";
        private static ISessionFactory _prototypeSessionFactory;
        private static SQLiteConnection _prototypeConnection;

        private const string InstanceConnectionString = "FullUri=file:instance.db?mode=memory&cache=shared";
        private ISessionFactory _instanceSessionFactory;
        private SQLiteConnection _instanceConnection;

        public DatabaseScope(Assembly assembly)
        {
            InitDatabasePrototype(assembly);
            InitDatabaseInstance();
        }

        private void InitDatabasePrototype(Assembly assembly)
        {
            if (_prototypeAssembly == assembly) return;

            if (_prototypeConnection != null)
            {
                _prototypeConnection.Close();
                _prototypeConnection.Dispose();
                _prototypeSessionFactory.Dispose();
            }

            _prototypeAssembly = assembly;

            _prototypeConnection = new SQLiteConnection(PrototypeConnectionString);
            _prototypeConnection.Open();

            _prototypeSessionFactory = Fluently
                .Configure()
                .Database(SQLiteConfiguration.Standard.ConnectionString(PrototypeConnectionString))
                .Mappings(m => m.HbmMappings.AddFromAssembly(assembly))
                .ExposeConfiguration(cfg => new SchemaExport(cfg).Execute(false, true, false, _prototypeConnection, null))
                .BuildSessionFactory();
        }

        private void InitDatabaseInstance()
        {
            _instanceSessionFactory = Fluently
                .Configure()
                .Database(SQLiteConfiguration.Standard.ConnectionString(InstanceConnectionString))
                .Mappings(m => m.HbmMappings.AddFromAssembly(_prototypeAssembly))
                .BuildSessionFactory();

            _instanceConnection = new SQLiteConnection(InstanceConnectionString);
            _instanceConnection.Open();

            _prototypeConnection.BackupDatabase(_instanceConnection, "main", "main", -1, null, -1);
        }

        public ISession OpenSession()
        {
            return _instanceSessionFactory.OpenSession(_instanceConnection);
        }

        public void Dispose()
        {
            _instanceConnection.Close();
            _instanceConnection.Dispose();
            _instanceSessionFactory.Dispose();
        }
    }
}

Answer 2

我在內存數據庫中使用SQLite觀察到的是，一旦關閉連接，數據庫中的所有內容都消失了。 所以做你想做的，

1. 為備份數據庫創建會話工廠，打開會話並建立架構，直到您完成整個測試套件后才關閉該會話

2. 為目標數據庫創建會話工廠，打開會話並使用此會話對象中的連接以及步驟1中創建的會話中的連接來復制數據

3. 使用在步驟2中創建的會話進行測試，並在測試完成后將其關閉

另一個解決方案是使用單個會話執行多個測試（所有測試都在單個測試夾具中），那么您不需要為每個測試創建新的會話工廠，而是為每個測試夾具創建新的會話工廠。