Update tests to use XCTest.