Add XCTest test