Skip to content

Commit 4f8f306

Browse files
committed
Create helper to remove redundancy
1 parent 073bebb commit 4f8f306

File tree

1 file changed

+12
-25
lines changed

1 file changed

+12
-25
lines changed

core/src/test/kotlin/com/segment/analytics/kotlin/core/TelemetryTest.kt

Lines changed: 12 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,13 @@ class TelemetryTest {
2929
queueBytesField.isAccessible = true
3030
return queueBytesField.get(Telemetry) as Int
3131
}
32+
fun TelemetryQueuePeek(): RemoteMetric {
33+
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
34+
queueField.isAccessible = true
35+
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
36+
return queueValue.peek() as RemoteMetric
37+
}
38+
3239
var TelemetryStarted: Boolean
3340
get() {
3441
val startedField: Field = Telemetry::class.java.getDeclaredField("started")
@@ -248,11 +255,7 @@ class TelemetryTest {
248255
it["error"] = "foo_192.168.0.1:8080"
249256
}
250257
assertEquals(1, TelemetryQueueSize())
251-
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
252-
queueField.isAccessible = true
253-
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
254-
val metric = queueValue.peek() as RemoteMetric
255-
assertEquals("foo__IP", metric.tags["error"])
258+
assertEquals("foo__IP", TelemetryQueuePeek().tags["error"])
256259
}
257260

258261
@Test
@@ -263,11 +266,7 @@ class TelemetryTest {
263266
it["error"] = "foo_2001:0db8:85a3:0000:0000:8a2e:0370:7334"
264267
}
265268
assertEquals(1, TelemetryQueueSize())
266-
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
267-
queueField.isAccessible = true
268-
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
269-
val metric = queueValue.peek() as RemoteMetric
270-
assertEquals("foo__IP", metric.tags["error"])
269+
assertEquals("foo__IP", TelemetryQueuePeek().tags["error"])
271270
}
272271

273272
@Test
@@ -278,11 +277,7 @@ class TelemetryTest {
278277
it["error"] = "foo_0x1234567890abcdef_bar"
279278
}
280279
assertEquals(1, TelemetryQueueSize())
281-
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
282-
queueField.isAccessible = true
283-
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
284-
val metric = queueValue.peek() as RemoteMetric
285-
assertEquals("foo_0x00_bar", metric.tags["error"])
280+
assertEquals("foo_0x00_bar", TelemetryQueuePeek().tags["error"])
286281
}
287282

288283
@Test
@@ -293,11 +288,7 @@ class TelemetryTest {
293288
it["error"] = "address_deadbeef_face"
294289
}
295290
assertEquals(1, TelemetryQueueSize())
296-
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
297-
queueField.isAccessible = true
298-
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
299-
val metric = queueValue.peek() as RemoteMetric
300-
assertEquals("address_0x00_face", metric.tags["error"])
291+
assertEquals("address_0x00_face", TelemetryQueuePeek().tags["error"])
301292
}
302293

303294
@Test
@@ -308,10 +299,6 @@ class TelemetryTest {
308299
it["error"] = "a5.b:_some_error_etc"
309300
}
310301
assertEquals(1, TelemetryQueueSize())
311-
val queueField: Field = Telemetry::class.java.getDeclaredField("queue")
312-
queueField.isAccessible = true
313-
val queueValue: ConcurrentLinkedQueue<*> = queueField.get(Telemetry) as ConcurrentLinkedQueue<*>
314-
val metric = queueValue.peek() as RemoteMetric
315-
assertEquals("_some_error_etc", metric.tags["error"])
302+
assertEquals("_some_error_etc", TelemetryQueuePeek().tags["error"])
316303
}
317304
}

0 commit comments

Comments
 (0)