Migrate to XCTest framework for unit testing