@@ -175,30 +175,24 @@ class Neo4jCoroutineDriverTest {
175175
176176 // when: querying the database and collecting the Flow of records
177177 val person = driver.coroutineSession().use { session ->
178-
179- val result = session.executeRead { tx ->
180- tx.run (" MATCH (p:Person) RETURN p" )
181- }
182-
183- result should {
184- have(isOpen())
185- have(keys() == listOf (" p" ))
178+ session.executeRead { tx ->
179+ tx.run (
180+ " MATCH (p:Person) RETURN p"
181+ ).records().map { record ->
182+ record[" p" ].let { p ->
183+ Person (
184+ id = p[" id" ].asString(),
185+ name = p[" name" ].asString(),
186+ email = p[" email" ].asString(),
187+ age = p[" age" ].asInt(),
188+ city = p[" city" ].asString(),
189+ skills = p[" skills" ].asList { it.asString() },
190+ active = p[" active" ].asBoolean(),
191+ createdAt = p[" createdAt" ].asInstant()
192+ )
193+ }
194+ }.first()
186195 }
187-
188- result.records().map { record ->
189- record[" p" ].let { p ->
190- Person (
191- id = p[" id" ].asString(),
192- name = p[" name" ].asString(),
193- email = p[" email" ].asString(),
194- age = p[" age" ].asInt(),
195- city = p[" city" ].asString(),
196- skills = p[" skills" ].asList { it.asString() },
197- active = p[" active" ].asBoolean(),
198- createdAt = p[" createdAt" ].asInstant()
199- )
200- }
201- }.first()
202196 }
203197
204198 // then: the person data matches what was stored
@@ -499,28 +493,31 @@ class Neo4jCoroutineDriverTest {
499493 val friendshipYears = mutableSetOf<Int >()
500494
501495 driver.coroutineSession().use { session ->
502- val result = session.executeRead { tx ->
503- tx.run ("""
496+ session.executeRead { tx ->
497+
498+ val result = tx.run ("""
504499 MATCH (p1:Person)-[f:FRIENDS_WITH]->(p2:Person)
505500 RETURN p1.name AS person1, p1.age AS age1,
506501 p2.name AS person2, p2.age AS age2,
507502 f.since AS friendsSince
508503 ORDER BY person1
509504 """ .trimIndent())
510- }
511505
512- result should {
513- have(isOpen())
514- have(keys() == listOf (" person1" , " age1" , " person2" , " age2" , " friendsSince" ))
515- }
506+ result should {
507+ have(! isOpen())
508+ have(keys() == listOf (" person1" , " age1" , " person2" , " age2" , " friendsSince" ))
509+ }
510+
511+ // Stream and process records using Flow
512+ result.records().collect { record ->
513+ recordCount++
514+ totalAge + = record[" age1" ].asInt()
515+ totalAge + = record[" age2" ].asInt()
516+ friendshipYears.add(record[" friendsSince" ].asInt())
517+ }
516518
517- // Stream and process records using Flow
518- result.records().collect { record ->
519- recordCount++
520- totalAge + = record[" age1" ].asInt()
521- totalAge + = record[" age2" ].asInt()
522- friendshipYears.add(record[" friendsSince" ].asInt())
523519 }
520+
524521 }
525522
526523 // then: verify we processed a large number of records
0 commit comments