Migrated all test to XCTests