Package 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	Description
static final byte[]	SAVEPOINT_ID
static final String	SAVEPOINT_NAME
static final FlightSqlClient.SubstraitPlan	SUBSTRAIT_PLAN
static final byte[]	SUBSTRAIT_PLAN_TEXT
static final String	SUBSTRAIT_VERSION
static final byte[]	TRANSACTION_ID
static final long	UPDATE_PREPARED_STATEMENT_EXPECTED_ROWS
static final long	UPDATE_PREPARED_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS
static final long	UPDATE_STATEMENT_EXPECTED_ROWS
static final long	UPDATE_STATEMENT_WITH_TRANSACTION_EXPECTED_ROWS

Constructor Summary

Constructors

Constructor	Description
FlightSqlScenario()

Method Summary

Modifier and Type	Method	Description
void	buildServer(FlightServer.Builder builder)
void	client(BufferAllocator allocator, Location location, FlightClient client)
FlightProducer	producer(BufferAllocator allocator, Location location)
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 Details

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 Details

FlightSqlScenario

public FlightSqlScenario()

Method Details

producer

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

Throws:

Exception

buildServer

public void buildServer(FlightServer.Builder builder)
                 throws Exception

Throws:

Exception

client

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

Throws:

Exception

validate

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

Throws:

Exception

validateSchema

protected void validateSchema(Schema expected,
 SchemaResult actual)