Fix unit test.

Bingqin Zhou · Bingqin Zhou · commit dfbc32fcd1cf · 2019-12-13T11:23:55.000-08:00
diff --git a/kcbq-confluent/src/test/java/com/wepay/kafka/connect/bigquery/schemaregistry/schemaretriever/SchemaRegistrySchemaRetrieverTest.java b/kcbq-confluent/src/test/java/com/wepay/kafka/connect/bigquery/schemaregistry/schemaretriever/SchemaRegistrySchemaRetrieverTest.java
@@ -7,6 +7,7 @@
 
 import com.google.cloud.bigquery.TableId;
 
+import com.wepay.kafka.connect.bigquery.api.KafkaSchemaRecordType;
 import io.confluent.connect.avro.AvroData;
 
 import io.confluent.kafka.schemaregistry.client.SchemaMetadata;
@@ -42,7 +43,7 @@ public void testRetrieveSchema() throws Exception {
     Schema expectedKafkaConnectSchema =
         SchemaBuilder.struct().field("f1", Schema.STRING_SCHEMA).name("testrecord").build();
 
-    assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, "value"));
-    assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, "key"));
+    assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, KafkaSchemaRecordType.VALUE));
+    assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, KafkaSchemaRecordType.KEY));
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,7 @@`
`7`	`7`
`8`	`8`	`import com.google.cloud.bigquery.TableId;`
`9`	`9`
	`10`	`+import com.wepay.kafka.connect.bigquery.api.KafkaSchemaRecordType;`
`10`	`11`	`import io.confluent.connect.avro.AvroData;`
`11`	`12`
`12`	`13`	`import io.confluent.kafka.schemaregistry.client.SchemaMetadata;`
`@@ -42,7 +43,7 @@ public void testRetrieveSchema() throws Exception {`
`42`	`43`	`Schema expectedKafkaConnectSchema =`
`43`	`44`	`SchemaBuilder.struct().field("f1", Schema.STRING_SCHEMA).name("testrecord").build();`
`44`	`45`
`45`		`- assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, "value"));`
`46`		`- assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, "key"));`
	`46`	`+ assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, KafkaSchemaRecordType.VALUE));`
	`47`	`+ assertEquals(expectedKafkaConnectSchema, testSchemaRetriever.retrieveSchema(table, testTopic, KafkaSchemaRecordType.KEY));`
`47`	`48`	`}`
`48`	`49`	`}`