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()