Skip to content

Commit

Permalink
Bmoric/test bq standard (#15270)
Browse files Browse the repository at this point in the history
* Add generic test to test the per stream state behavior

* Add missing dependency

* Add test for the big query record consumer

* Add license

* Add missing newline
  • Loading branch information
benmoriceau authored Aug 3, 2022
1 parent daafb61 commit a837101
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@ dependencies {
implementation project(':airbyte-integrations:connectors:destination-gcs')
implementation ('com.github.airbytehq:json-avro-converter:1.0.1') { exclude group: 'ch.qos.logback', module: 'logback-classic'}

testImplementation project(':airbyte-integrations:bases:standard-destination-test')

integrationTestJavaImplementation project(':airbyte-integrations:bases:standard-destination-test')
integrationTestJavaImplementation files(project(':airbyte-integrations:bases:base-normalization').airbyteDocker.outputs)
integrationTestJavaImplementation project(':airbyte-integrations:connectors:destination-bigquery')
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
package io.airbyte.integrations.destination.bigquery;

import io.airbyte.integrations.base.AirbyteStreamNameNamespacePair;
import io.airbyte.integrations.base.FailureTrackingAirbyteMessageConsumer;
import io.airbyte.integrations.destination.bigquery.uploader.AbstractBigQueryUploader;
import io.airbyte.integrations.standardtest.destination.PerStreamStateMessageTest;
import io.airbyte.protocol.models.AirbyteMessage;
import java.util.Map;
import java.util.function.Consumer;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

@ExtendWith(MockitoExtension.class)
public class BigQueryRecordConsumerTest extends PerStreamStateMessageTest {

@Mock
private Map<AirbyteStreamNameNamespacePair, AbstractBigQueryUploader<?>> uploaderMap;
@Mock
private Consumer<AirbyteMessage> outputRecordCollector;

@InjectMocks
private BigQueryRecordConsumer bigQueryRecordConsumer;

@Override protected Consumer<AirbyteMessage> getMockedConsumer() {
return outputRecordCollector;
}

@Override protected FailureTrackingAirbyteMessageConsumer getMessageConsumer() {
return bigQueryRecordConsumer;
}
}

0 comments on commit a837101

Please sign in to comment.