summaryrefslogtreecommitdiff
path: root/opendc-trace/opendc-trace-parquet/src/test
diff options
context:
space:
mode:
authorDante Niewenhuis <d.niewenhuis@hotmail.com>2024-03-05 13:23:57 +0100
committerGitHub <noreply@github.com>2024-03-05 13:23:57 +0100
commit5864cbcbfe2eb8c36ca05c3a39c7e5916aeecaec (patch)
tree5b2773b8dc21c2e1b526fb70f829c376dd80532a /opendc-trace/opendc-trace-parquet/src/test
parentd28002a3c151d198298574312f32f1cb43f3a660 (diff)
Updated package versions, updated web server tests. (#207)
* Updated all package versions including kotlin. Updated all web-server tests to run. * Changed the java version of the tests. OpenDC now only supports java 19. * small update * test update * new update * updated docker version to 19 * updated docker version to 19
Diffstat (limited to 'opendc-trace/opendc-trace-parquet/src/test')
-rw-r--r--opendc-trace/opendc-trace-parquet/src/test/kotlin/org/opendc/trace/util/parquet/ParquetTest.kt112
1 files changed, 60 insertions, 52 deletions
diff --git a/opendc-trace/opendc-trace-parquet/src/test/kotlin/org/opendc/trace/util/parquet/ParquetTest.kt b/opendc-trace/opendc-trace-parquet/src/test/kotlin/org/opendc/trace/util/parquet/ParquetTest.kt
index b6c5a423..fc90aded 100644
--- a/opendc-trace/opendc-trace-parquet/src/test/kotlin/org/opendc/trace/util/parquet/ParquetTest.kt
+++ b/opendc-trace/opendc-trace-parquet/src/test/kotlin/org/opendc/trace/util/parquet/ParquetTest.kt
@@ -51,49 +51,52 @@ import java.nio.file.Path
internal class ParquetTest {
private lateinit var path: Path
- private val schema = Types.buildMessage()
- .addField(
- Types.primitive(PrimitiveType.PrimitiveTypeName.INT32, Type.Repetition.REQUIRED)
- .named("field")
- )
- .named("test")
- private val writeSupport = object : WriteSupport<Int>() {
- lateinit var recordConsumer: RecordConsumer
-
- override fun init(configuration: Configuration): WriteContext {
- return WriteContext(schema, emptyMap())
- }
+ private val schema =
+ Types.buildMessage()
+ .addField(
+ Types.primitive(PrimitiveType.PrimitiveTypeName.INT32, Type.Repetition.REQUIRED)
+ .named("field"),
+ )
+ .named("test")
+ private val writeSupport =
+ object : WriteSupport<Int>() {
+ lateinit var recordConsumer: RecordConsumer
+
+ override fun init(configuration: Configuration): WriteContext {
+ return WriteContext(schema, emptyMap())
+ }
- override fun prepareForWrite(recordConsumer: RecordConsumer) {
- this.recordConsumer = recordConsumer
- }
+ override fun prepareForWrite(recordConsumer: RecordConsumer) {
+ this.recordConsumer = recordConsumer
+ }
- override fun write(record: Int) {
- val consumer = recordConsumer
+ override fun write(record: Int) {
+ val consumer = recordConsumer
- consumer.startMessage()
- consumer.startField("field", 0)
- consumer.addInteger(record)
- consumer.endField("field", 0)
- consumer.endMessage()
+ consumer.startMessage()
+ consumer.startField("field", 0)
+ consumer.addInteger(record)
+ consumer.endField("field", 0)
+ consumer.endMessage()
+ }
}
- }
- private val readSupport = object : ReadSupport<Int>() {
- @Suppress("OVERRIDE_DEPRECATION")
- override fun init(
- configuration: Configuration,
- keyValueMetaData: Map<String, String>,
- fileSchema: MessageType
- ): ReadContext = ReadContext(fileSchema)
-
- override fun prepareForRead(
- configuration: Configuration,
- keyValueMetaData: Map<String, String>,
- fileSchema: MessageType,
- readContext: ReadContext
- ): RecordMaterializer<Int> = TestRecordMaterializer()
- }
+ private val readSupport =
+ object : ReadSupport<Int>() {
+ @Suppress("OVERRIDE_DEPRECATION")
+ override fun init(
+ configuration: Configuration,
+ keyValueMetaData: Map<String, String>,
+ fileSchema: MessageType,
+ ): ReadContext = ReadContext(fileSchema)
+
+ override fun prepareForRead(
+ configuration: Configuration,
+ keyValueMetaData: Map<String, String>,
+ fileSchema: MessageType,
+ readContext: ReadContext,
+ ): RecordMaterializer<Int> = TestRecordMaterializer()
+ }
/**
* Set up the test
@@ -117,9 +120,10 @@ internal class ParquetTest {
@Test
fun testSmoke() {
val n = 4
- val writer = LocalParquetWriter.builder(path, writeSupport)
- .withWriteMode(ParquetFileWriter.Mode.OVERWRITE)
- .build()
+ val writer =
+ LocalParquetWriter.builder(path, writeSupport)
+ .withWriteMode(ParquetFileWriter.Mode.OVERWRITE)
+ .build()
try {
repeat(n) { i ->
@@ -166,19 +170,23 @@ internal class ParquetTest {
private class TestRecordMaterializer : RecordMaterializer<Int>() {
private var current: Int = 0
- private val fieldConverter = object : PrimitiveConverter() {
- override fun addInt(value: Int) {
- current = value
+ private val fieldConverter =
+ object : PrimitiveConverter() {
+ override fun addInt(value: Int) {
+ current = value
+ }
}
- }
- private val root = object : GroupConverter() {
- override fun getConverter(fieldIndex: Int): Converter {
- require(fieldIndex == 0)
- return fieldConverter
+ private val root =
+ object : GroupConverter() {
+ override fun getConverter(fieldIndex: Int): Converter {
+ require(fieldIndex == 0)
+ return fieldConverter
+ }
+
+ override fun start() {}
+
+ override fun end() {}
}
- override fun start() {}
- override fun end() {}
- }
override fun getCurrentRecord(): Int = current