summaryrefslogtreecommitdiff
path: root/opendc-trace/opendc-trace-azure/src/main
diff options
context:
space:
mode:
authorFabian Mastenbroek <mail.fabianm@gmail.com>2022-06-08 15:06:14 +0200
committerFabian Mastenbroek <mail.fabianm@gmail.com>2022-06-08 15:06:14 +0200
commit9d759c9bc987965fae8b0c16c000772c546bf3a2 (patch)
treebf20f51b434d56e60ad013568ac1a32b912a3b5e /opendc-trace/opendc-trace-azure/src/main
parentf7a5b200e4767213617774a87113d0b56d8d8a59 (diff)
test(trace): Add conformance suite for OpenDC trace API
This change adds a re-usable test suite for the interface of the OpenDC trace API, so implementors can verify whether they match the specification of the interfaces.
Diffstat (limited to 'opendc-trace/opendc-trace-azure/src/main')
-rw-r--r--opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceStateTableReader.kt19
-rw-r--r--opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceTableReader.kt21
2 files changed, 40 insertions, 0 deletions
diff --git a/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceStateTableReader.kt b/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceStateTableReader.kt
index e9017b35..c0c67329 100644
--- a/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceStateTableReader.kt
+++ b/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceStateTableReader.kt
@@ -37,11 +37,20 @@ import java.util.*
* A [TableReader] for the Azure v1 VM resource state table.
*/
internal class AzureResourceStateTableReader(private val parser: CsvParser) : TableReader {
+ /**
+ * A flag to indicate whether a single row has been read already.
+ */
+ private var isStarted = false
+
init {
parser.schema = schema
}
override fun nextRow(): Boolean {
+ if (!isStarted) {
+ isStarted = true
+ }
+
reset()
if (!nextStart()) {
@@ -100,6 +109,7 @@ internal class AzureResourceStateTableReader(private val parser: CsvParser) : Ta
}
override fun getDouble(index: Int): Double {
+ checkActive()
return when (index) {
COL_CPU_USAGE_PCT -> cpuUsagePct
else -> throw IllegalArgumentException("Invalid column")
@@ -107,6 +117,7 @@ internal class AzureResourceStateTableReader(private val parser: CsvParser) : Ta
}
override fun getString(index: Int): String? {
+ checkActive()
return when (index) {
COL_ID -> id
else -> throw IllegalArgumentException("Invalid column")
@@ -118,6 +129,7 @@ internal class AzureResourceStateTableReader(private val parser: CsvParser) : Ta
}
override fun getInstant(index: Int): Instant? {
+ checkActive()
return when (index) {
COL_TIMESTAMP -> timestamp
else -> throw IllegalArgumentException("Invalid column")
@@ -145,6 +157,13 @@ internal class AzureResourceStateTableReader(private val parser: CsvParser) : Ta
}
/**
+ * Helper method to check if the reader is active.
+ */
+ private fun checkActive() {
+ check(isStarted && !parser.isClosed) { "No active row. Did you call nextRow()?" }
+ }
+
+ /**
* Advance the parser until the next object start.
*/
private fun nextStart(): Boolean {
diff --git a/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceTableReader.kt b/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceTableReader.kt
index 456a2536..a8451301 100644
--- a/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceTableReader.kt
+++ b/opendc-trace/opendc-trace-azure/src/main/kotlin/org/opendc/trace/azure/AzureResourceTableReader.kt
@@ -35,11 +35,20 @@ import java.util.*
* A [TableReader] for the Azure v1 VM resources table.
*/
internal class AzureResourceTableReader(private val parser: CsvParser) : TableReader {
+ /**
+ * A flag to indicate whether a single row has been read already.
+ */
+ private var isStarted = false
+
init {
parser.schema = schema
}
override fun nextRow(): Boolean {
+ if (!isStarted) {
+ isStarted = true
+ }
+
reset()
if (!nextStart()) {
@@ -92,6 +101,7 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
override fun getInt(index: Int): Int {
+ checkActive()
return when (index) {
COL_CPU_COUNT -> cpuCores
else -> throw IllegalArgumentException("Invalid column")
@@ -99,6 +109,7 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
override fun getLong(index: Int): Long {
+ checkActive()
return when (index) {
COL_CPU_COUNT -> cpuCores.toLong()
else -> throw IllegalArgumentException("Invalid column")
@@ -110,6 +121,7 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
override fun getDouble(index: Int): Double {
+ checkActive()
return when (index) {
COL_MEM_CAPACITY -> memCapacity
else -> throw IllegalArgumentException("Invalid column")
@@ -117,6 +129,7 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
override fun getString(index: Int): String? {
+ checkActive()
return when (index) {
COL_ID -> id
else -> throw IllegalArgumentException("Invalid column")
@@ -128,6 +141,7 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
override fun getInstant(index: Int): Instant? {
+ checkActive()
return when (index) {
COL_START_TIME -> startTime
COL_STOP_TIME -> stopTime
@@ -156,6 +170,13 @@ internal class AzureResourceTableReader(private val parser: CsvParser) : TableRe
}
/**
+ * Helper method to check if the reader is active.
+ */
+ private fun checkActive() {
+ check(isStarted && !parser.isClosed) { "No active row. Did you call nextRow()?" }
+ }
+
+ /**
* Advance the parser until the next object start.
*/
private fun nextStart(): Boolean {