diff --git a/Signal/test/util/OWSDatabaseConverterTest.m b/Signal/test/util/OWSDatabaseConverterTest.m index 3256b3e98..5fe4f1fc6 100644 --- a/Signal/test/util/OWSDatabaseConverterTest.m +++ b/Signal/test/util/OWSDatabaseConverterTest.m @@ -31,6 +31,15 @@ NS_ASSUME_NONNULL_BEGIN #pragma mark - +@interface YapDatabase (OWSDatabaseConverterTest) + +- (void)flushInternalQueue; +- (void)flushCheckpointQueue; + +@end + +#pragma mark - + @implementation OWSDatabaseConverterTest - (NSData *)randomDatabasePassword @@ -97,6 +106,9 @@ NS_ASSUME_NONNULL_BEGIN databaseBlock(database); + [database flushInternalQueue]; + [database flushCheckpointQueue]; + // Close the database. database = nil; }