Search in sources :

Example 6 with FakeSnowflakeBasicDataSource

use of org.apache.beam.sdk.io.snowflake.test.FakeSnowflakeBasicDataSource in project beam by apache.

the class QueryDispositionLocationTest method setup.

@Before
public void setup() {
    options.setStagingBucketName(BUCKET_NAME);
    options.setStorageIntegrationName("STORAGE_INTEGRATION");
    options.setServerName("NULL.snowflakecomputing.com");
    dc = SnowflakeIO.DataSourceConfiguration.create(new FakeSnowflakeBasicDataSource()).withServerName(options.getServerName());
}
Also used : FakeSnowflakeBasicDataSource(org.apache.beam.sdk.io.snowflake.test.FakeSnowflakeBasicDataSource) Before(org.junit.Before)

Aggregations

FakeSnowflakeBasicDataSource (org.apache.beam.sdk.io.snowflake.test.FakeSnowflakeBasicDataSource)6 TestSnowflakePipelineOptions (org.apache.beam.sdk.io.snowflake.test.TestSnowflakePipelineOptions)4 BeforeClass (org.junit.BeforeClass)4 FakeSnowflakeBatchServiceImpl (org.apache.beam.sdk.io.snowflake.test.FakeSnowflakeBatchServiceImpl)3 Before (org.junit.Before)2 AvroGeneratedUser (org.apache.beam.sdk.io.AvroGeneratedUser)1 FakeSnowflakeStreamingServiceImpl (org.apache.beam.sdk.io.snowflake.test.FakeSnowflakeStreamingServiceImpl)1