zeus
73 lines
2.8 KiB
Swift
73 lines
2.8 KiB
Swift
import XCTest
|
|
import Foundation
|
|
import GRDB
|
|
#if GRDB_COMPARE
|
|
import CoreData
|
|
import RealmSwift
|
|
#endif
|
|
|
|
func generateSQLiteDatabaseIfMissing(at url: URL, insertedRowCount: Int) throws {
|
|
try DatabaseQueue(path: url.path).write { db in
|
|
if try db.tableExists("item") {
|
|
let count = try Int.fetchOne(db, sql: "SELECT COUNT(*) FROM item")!
|
|
if count == insertedRowCount {
|
|
return
|
|
} else {
|
|
try db.execute(sql: "DROP TABLE item")
|
|
}
|
|
}
|
|
try db.execute(sql: "CREATE TABLE item (i0 INT, i1 INT, i2 INT, i3 INT, i4 INT, i5 INT, i6 INT, i7 INT, i8 INT, i9 INT)")
|
|
|
|
let statement = try! db.makeStatement(sql: "INSERT INTO item (i0, i1, i2, i3, i4, i5, i6, i7, i8, i9) VALUES (?,?,?,?,?,?,?,?,?,?)")
|
|
for i in 0..<insertedRowCount {
|
|
try statement.execute(arguments: [i, i, i, i, i, i, i, i, i, i])
|
|
}
|
|
}
|
|
}
|
|
|
|
#if GRDB_COMPARE
|
|
func generateCoreDataDatabaseIfMissing(at url: URL, fromModelAt modelURL: URL, insertedRowCount: Int) throws {
|
|
let mom = NSManagedObjectModel(contentsOf: modelURL)!
|
|
let psc = NSPersistentStoreCoordinator(managedObjectModel: mom)
|
|
try psc.addPersistentStore(ofType: NSSQLiteStoreType, configurationName: nil, at: url, options: nil)
|
|
let moc = NSManagedObjectContext(concurrencyType: .mainQueueConcurrencyType)
|
|
moc.performAndWait {
|
|
moc.persistentStoreCoordinator = psc
|
|
let request = NSFetchRequest<NSFetchRequestResult>(entityName: "Item")
|
|
if try! moc.count(for: request) == insertedRowCount {
|
|
return
|
|
}
|
|
try! moc.execute(NSBatchDeleteRequest(fetchRequest: request))
|
|
|
|
for i in 0..<insertedRowCount {
|
|
let item = NSEntityDescription.insertNewObject(forEntityName: "Item", into: moc)
|
|
item.setValue(NSNumber(value: i), forKey: "i0")
|
|
item.setValue(NSNumber(value: i), forKey: "i1")
|
|
item.setValue(NSNumber(value: i), forKey: "i2")
|
|
item.setValue(NSNumber(value: i), forKey: "i3")
|
|
item.setValue(NSNumber(value: i), forKey: "i4")
|
|
item.setValue(NSNumber(value: i), forKey: "i5")
|
|
item.setValue(NSNumber(value: i), forKey: "i6")
|
|
item.setValue(NSNumber(value: i), forKey: "i7")
|
|
item.setValue(NSNumber(value: i), forKey: "i8")
|
|
item.setValue(NSNumber(value: i), forKey: "i9")
|
|
}
|
|
try! moc.save()
|
|
}
|
|
}
|
|
|
|
func generateRealmDatabaseIfMissing(at url: URL, insertedRowCount: Int) throws {
|
|
let realm = try Realm(fileURL: url)
|
|
try realm.write {
|
|
if realm.objects(RealmItem.self).count == insertedRowCount {
|
|
return
|
|
}
|
|
|
|
realm.deleteAll()
|
|
for i in 0..<insertedRowCount {
|
|
realm.add(RealmItem(i0: i, i1: i, i2: i, i3: i, i4: i, i5: i, i6: i, i7: i, i8: i, i9: i))
|
|
}
|
|
}
|
|
}
|
|
#endif
|