Migrated test suite to XCTest framework