|
40 | 40 | import org.apache.kafka.connect.storage.OffsetStorageReader; |
41 | 41 |
|
42 | 42 | import io.aiven.kafka.connect.common.config.SourceCommonConfig; |
43 | | -import io.aiven.kafka.connect.common.source.input.AvroTestDataFixture; |
| 43 | +import io.aiven.kafka.connect.common.format.AvroTestDataFixture; |
44 | 44 | import io.aiven.kafka.connect.common.source.input.InputFormat; |
45 | | -import io.aiven.kafka.connect.common.source.input.JsonTestDataFixture; |
46 | | -import io.aiven.kafka.connect.common.source.input.ParquetTestDataFixture; |
| 45 | +import io.aiven.kafka.connect.common.format.JsonTestDataFixture; |
| 46 | +import io.aiven.kafka.connect.common.format.ParquetTestDataFixture; |
47 | 47 | import io.aiven.kafka.connect.common.source.input.Transformer; |
48 | 48 | import io.aiven.kafka.connect.common.source.input.TransformerFactory; |
49 | 49 | import io.aiven.kafka.connect.common.source.task.DistributionType; |
@@ -220,13 +220,13 @@ static List<Arguments> inputFormatList() throws IOException { |
220 | 220 | bytes = "Hello World".getBytes(StandardCharsets.UTF_8); |
221 | 221 | break; |
222 | 222 | case AVRO : |
223 | | - bytes = AvroTestDataFixture.generateMockAvroData(1); |
| 223 | + bytes = AvroTestDataFixture.generateAvroData(1); |
224 | 224 | break; |
225 | 225 | case JSONL : |
226 | | - bytes = JsonTestDataFixture.getJsonRecs(1).getBytes(StandardCharsets.UTF_8); |
| 226 | + bytes = JsonTestDataFixture.generateJsonRecs(1).getBytes(StandardCharsets.UTF_8); |
227 | 227 | break; |
228 | 228 | case PARQUET : |
229 | | - bytes = ParquetTestDataFixture.generateMockParquetData("name", 1); |
| 229 | + bytes = ParquetTestDataFixture.generateParquetData("name", 1); |
230 | 230 | break; |
231 | 231 | default : |
232 | 232 | throw new IllegalArgumentException("Unsupported format: " + format); |
@@ -280,13 +280,13 @@ static List<Arguments> multiInputFormatList() throws IOException { |
280 | 280 | Arrays.fill(bytes, (byte) 5); |
281 | 281 | break; |
282 | 282 | case AVRO : |
283 | | - bytes = AvroTestDataFixture.generateMockAvroData(2); |
| 283 | + bytes = AvroTestDataFixture.generateAvroData(2); |
284 | 284 | break; |
285 | 285 | case JSONL : |
286 | | - bytes = JsonTestDataFixture.getJsonRecs(2).getBytes(StandardCharsets.UTF_8); |
| 286 | + bytes = JsonTestDataFixture.generateJsonRecs(2).getBytes(StandardCharsets.UTF_8); |
287 | 287 | break; |
288 | 288 | case PARQUET : |
289 | | - bytes = ParquetTestDataFixture.generateMockParquetData("name", 2); |
| 289 | + bytes = ParquetTestDataFixture.generateParquetData("name", 2); |
290 | 290 | break; |
291 | 291 | default : |
292 | 292 | throw new IllegalArgumentException("Unsupported format: " + format); |
|
0 commit comments