Fix the forgotten scheme-test

Make the test properly handle key migration.

Change-Id: Id1e468b1da807d52f6054ee7d0edb9c0472264fd

diff --git a/misc/encryption_scheme/scheme-test.cpp b/misc/encryption_scheme/scheme-test.cpp
index 70d5e8d..89f8a7b 100644 (file)
--- a/misc/encryption_scheme/scheme-test.cpp
+++ b/misc/encryption_scheme/scheme-test.cpp
@@ -856,6 +856,11 @@ void SchemeTest::EnableDirectDbAccess()
        FileSystem fs(UID);
        auto wrappedDKEK = fs.getDKEK();
        auto keyProvider = KeyProvider(wrappedDKEK, DBPASS);
+       if (!keyProvider.isInitialized()) {
+               keyProvider.migrateDomainKEK(wrappedDKEK, DBPASS);
+               fs.saveDKEK(keyProvider.getWrappedDomainKEK(DBPASS));
+               LogInfo("DKEK migrated");
+       }
 
        auto wrappedDatabaseDEK = fs.getDBDEK();
        RawBuffer key = keyProvider.getPureDEK(wrappedDatabaseDEK);