async def test_add_companion_with_host_and_port_adds_companion( self) -> None: companion_manager = CompanionManager(logger=mock.MagicMock()) companion_manager.add_companion( CompanionInfo(udid="asdasda", host="foohost", port=123, is_local=False)) assert companion_manager._udid_companion_map["asdasda"]
async def test_get_default_companion(self) -> None: companion_manager = CompanionManager(logger=mock.MagicMock()) self.assertFalse(companion_manager.has_default_companion()) add_companion(companion_manager, TEST_COMPANION) self.assertTrue(companion_manager.has_default_companion()) self.assertEqual(companion_manager.get_default_companion(), TEST_COMPANION) companion_manager.add_companion( CompanionInfo( udid="someOtherUdid", host=TEST_COMPANION.host, port=TEST_COMPANION.port, is_local=TEST_COMPANION.is_local, )) self.assertFalse(companion_manager.has_default_companion())
async def test_add_companion_assigns_to_target(self) -> None: companion_manager = CompanionManager(logger=mock.MagicMock()) companion_manager.update_target( TargetDescription( udid="asdasda", name="iPhone", state="Booted", target_type="simulator", os_version="iOS 11.4", architecture="x86_64", companion_info=None, screen_dimensions=None, )) companion_manager.add_companion( CompanionInfo(udid="asdasda", host="foohost", port=123, is_local=False)) assert companion_manager._udid_target_map["asdasda"].companion_info
def add_companion(companion_manager: CompanionManager, companion: CompanionInfo = TEST_COMPANION) -> None: companion_manager.add_companion(companion)