org.apache.arrow.flight.integration.tests

Class FlightSqlScenario

java.lang.Object

org.apache.arrow.flight.integration.tests.FlightSqlScenario

Direct Known Subclasses:

FlightSqlExtensionScenario

public class FlightSqlScenario
extends Object

Integration test scenario for validating Flight SQL specs across multiple implementations. This should ensure that RPC objects are being built and parsed correctly for multiple languages and that the Arrow schemas are returned as expected.

Field Summary

Fields

Modifier and Type	Field and Description
static byte[]	SAVEPOINT_ID
static String	SAVEPOINT_NAME
static FlightSqlClient.SubstraitPlan	SUBSTRAIT_PLAN
static byte[]	SUBSTRAIT_PLAN_TEXT
static String	SUBSTRAIT_VERSION
static byte[]	TRANSACTION_ID
static long	UPDATE_PREPARED_STATEMENT_EXPECTED_ROWS
static long	UPDATE_PREPARED_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS
static long	UPDATE_STATEMENT_EXPECTED_ROWS
static long	UPDATE_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

Constructor Summary

Constructors

Constructor and Description
FlightSqlScenario()

Method Summary

Modifier and Type	Method and Description
void	buildServer(FlightServer.Builder builder) Set any other server options.
void	client(BufferAllocator allocator, Location location, FlightClient client) Run as the client in the scenario.
FlightProducer	producer(BufferAllocator allocator, Location location) Construct the FlightProducer for a server in this scenario.
protected void	validate(Schema expectedSchema, FlightInfo flightInfo, FlightSqlClient sqlClient)
protected void	validateSchema(Schema expected, SchemaResult actual)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

UPDATE_STATEMENT_EXPECTED_ROWS

public static final long UPDATE_STATEMENT_EXPECTED_ROWS

See Also:

Constant Field Values

UPDATE_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

public static final long UPDATE_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

See Also:

Constant Field Values

UPDATE_PREPARED_STATEMENT_EXPECTED_ROWS

public static final long UPDATE_PREPARED_STATEMENT_EXPECTED_ROWS

See Also:

Constant Field Values

UPDATE_PREPARED_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

public static final long UPDATE_PREPARED_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

See Also:

Constant Field Values

SAVEPOINT_ID

public static final byte[] SAVEPOINT_ID

SAVEPOINT_NAME

public static final String SAVEPOINT_NAME

See Also:

Constant Field Values

SUBSTRAIT_PLAN_TEXT

public static final byte[] SUBSTRAIT_PLAN_TEXT

SUBSTRAIT_VERSION

public static final String SUBSTRAIT_VERSION

See Also:

Constant Field Values

SUBSTRAIT_PLAN

public static final FlightSqlClient.SubstraitPlan SUBSTRAIT_PLAN

TRANSACTION_ID

public static final byte[] TRANSACTION_ID

Constructor Detail

FlightSqlScenario

public FlightSqlScenario()

Method Detail

producer

public FlightProducer producer(BufferAllocator allocator,
                               Location location)
                        throws Exception

Construct the FlightProducer for a server in this scenario.

Throws:

Exception

buildServer

public void buildServer(FlightServer.Builder builder)
                 throws Exception

Set any other server options.

Throws:

Exception

client

public void client(BufferAllocator allocator,
                   Location location,
                   FlightClient client)
            throws Exception

Run as the client in the scenario.

Throws:

Exception

validate

protected void validate(Schema expectedSchema,
                        FlightInfo flightInfo,
                        FlightSqlClient sqlClient)
                 throws Exception

Throws:

Exception

validateSchema

protected void validateSchema(Schema expected,
                              SchemaResult actual)