Files
impala/tests/custom_cluster/test_external_planner.py
Steve Carlin bc83d46a9a IMPALA-12424: Allow third party JniFrontend interface.
This patch allows a third party to inject their own frontend
class instead of using the default JniFrontend included in the
project.

The test case includes an interface that runs queries as normal
except for the "select 1" query which gets changed to "select 42".

Change-Id: I89e677da557b39232847644b6ff17510e2b3c3d5
Reviewed-on: http://gerrit.cloudera.org:8080/20459
Reviewed-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
2023-09-08 20:20:56 +00:00

44 lines
1.7 KiB
Python

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
from __future__ import absolute_import, division, print_function
import logging
import pytest
import os
from tests.common.custom_cluster_test_suite import CustomClusterTestSuite
LOG = logging.getLogger(__name__)
class TestExternalPlanner(CustomClusterTestSuite):
@classmethod
def setup_class(cls):
super(TestExternalPlanner, cls).setup_class()
os.environ["CUSTOM_CLASSPATH"] = os.getenv("IMPALA_HOME") + \
"/java/external-frontend/target/external-frontend-1.0.jar"
@pytest.mark.execute_serially
@CustomClusterTestSuite.with_args(
impalad_args="-jni_frontend_class=org/apache/impala/externalfrontend/TestJniFrontend")
def test_external_frontend(self, vector):
setup_client = self.create_impala_client()
assert setup_client.execute("select 2").data == ['2']
# The custom JniFrontend overrides the 'select 1' query with a 'select 42' query.
assert setup_client.execute("select 1").data == ['42']