测试失败/异常后的assemblyCleanup()

我正在运行一些需要连接到数据库的单元测试。当我的测试项目初始化时,将创建数据库的快照,当测试完成时,数据库将还原回快照。

以下是实现方法:

[TestClass]
public static class AssemblyInitializer
{
    [AssemblyInitialize()]
    public static void AssemblyInit(TestContext context)
    {
        var dbss = new DatabaseSnapshot(...);    
        dbss.CreateSnapshot();
    }

    [AssemblyCleanup()]
    public static void AssemblyCleanup()
    {
        var dbss = new DatabaseSnapshot(...);
        dbss.RevertDatabase();
    }
}

现在这一切都起作用了,但是当我有一个失败的测试或一些异常时,我的问题就出现了。当然不会调用AssemblyCleanup,所以如何解决此问题?无论发生什么,都必须还原快照。这可能吗?