[Contact] Refactor unit test