From 163beefc0f1e4e58962f3091e58fcc9095f1be2b Mon Sep 17 00:00:00 2001
From: Kyriakos Tharrouniatis <kyriakos.tharrouniatis@r3.com>
Date: Mon, 17 Feb 2020 10:34:52 +0000
Subject: [PATCH] Splitting heartBeat into heartBeatOnNext and heartBeatOnError

---
 .../net/corda/node/utilities/ObservablesTests.kt   | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/node/src/test/kotlin/net/corda/node/utilities/ObservablesTests.kt b/node/src/test/kotlin/net/corda/node/utilities/ObservablesTests.kt
index 315ae65e56..c4b4fbbca1 100644
--- a/node/src/test/kotlin/net/corda/node/utilities/ObservablesTests.kt
+++ b/node/src/test/kotlin/net/corda/node/utilities/ObservablesTests.kt
@@ -317,19 +317,21 @@ class ObservablesTests {
 
     @Test
     fun `throwing FlowSafeSubscriber as a leaf will call onError`() {
-        var heartBeat = 0
+        var heartBeatOnNext = 0
+        var heartBeatOnError = 0
         val source = FlowSafeSubject(PublishSubject.create<Int>())
         // add a leaf FlowSafeSubscriber
-        source.subscribe(/*onNext*/{
-            heartBeat++
+        source.subscribe({
+            heartBeatOnNext++
             throw IllegalStateException()
-        },/*onError*/{
-            heartBeat++
+        }, {
+            heartBeatOnError++
         })
 
         source.onNext(1)
         source.onNext(1)
-        assertEquals(4, heartBeat)
+        assertEquals(2, heartBeatOnNext)
+        assertEquals(2, heartBeatOnError)
     }
 
     @Test