def test_online_get(self, mock_endpoint_operations: EndpointOperations,
randstr: str, mock_aml_services: Mock) -> None:
mock_aml_services.online_endpoints.get.return_value = OnlineEndpointPropertiesTrackedResource(
name=randstr,
properties=OnlineEndpointProperties(
compute_configuration=AksComputeConfiguration(
compute_name="compute"),
auth_mode="key"),
)
mock_aml_services.online_deployments.list.return_value = None
mock_endpoint_operations.get(type=ONLINE_ENDPOINT_TYPE, name=randstr)
mock_endpoint_operations._online_operation.get.assert_called_once()
mock_endpoint_operations._online_deployment.list.assert_called_once()
def test_batch_get(self, mock_endpoint_operations: EndpointOperations,
randstr: str) -> None:
mock_endpoint_operations.get(type=BATCH_ENDPOINT_TYPE, name=randstr)
mock_endpoint_operations._batch_operation.get.assert_called_once()
