def test_throw_if_random_type(
self, mock_endpoint_operations: EndpointOperations) -> None:
with pytest.raises(Exception):
mock_endpoint_operations.list(type="test_type")
def test_batch_list(self,
mock_endpoint_operations: EndpointOperations) -> None:
mock_endpoint_operations.list(type=BATCH_ENDPOINT_TYPE)
mock_endpoint_operations._batch_operation.list.assert_called_once()
def test_online_list(self,
mock_endpoint_operations: EndpointOperations) -> None:
mock_endpoint_operations.list(type=ONLINE_ENDPOINT_TYPE)
mock_endpoint_operations._online_operation.list.assert_called_once()