async def test_creates_default_companion(self) -> None: companion_manager = CompanionManager(companion_path=None, logger=mock.MagicMock()) add_companion(companion_manager, TEST_COMPANION) async with companion_manager.create_companion_for_target_with_udid( None, None) as yielded_companion: self.assertEqual(yielded_companion, TEST_COMPANION)
async def test_get_existing_companion(self) -> None: companion_manager = CompanionManager(logger=mock.MagicMock()) add_companion(companion_manager, TEST_COMPANION) async with companion_manager.create_companion_for_target_with_udid( TEST_COMPANION.udid, None) as yielded_compainion: self.assertEqual(yielded_compainion, TEST_COMPANION)