Update tests to XCTest