From 2773ae9b9c80d865b21ed9a474034012acfce299 Mon Sep 17 00:00:00 2001
From: Puneet Behl <behlp@objectcomputing.com>
Date: Fri, 5 May 2023 16:12:38 +0530
Subject: [PATCH] Added cleanup in the test

---
 .../gorm/tests/assignedid/OneToManyCreateSpec.groovy  | 11 ++++++++---
 .../gorm/tests/cypher/OneToManyCreateSpec.groovy      | 11 +++++++++--
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/assignedid/OneToManyCreateSpec.groovy b/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/assignedid/OneToManyCreateSpec.groovy
index 9aea2680..f008b1d9 100644
--- a/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/assignedid/OneToManyCreateSpec.groovy
+++ b/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/assignedid/OneToManyCreateSpec.groovy
@@ -11,7 +11,7 @@ import spock.lang.Specification
 /**
  * Created by graemerocher on 14/03/2017.
  */
-class OneToManyCreateSpec extends Specification{
+class OneToManyCreateSpec extends Specification {
     // tag::setup[]
     @Shared @AutoCleanup Neo4jDatastore datastore = new Neo4jDatastore(getClass().getPackage())
     // end::setup[]
@@ -19,19 +19,24 @@ class OneToManyCreateSpec extends Specification{
     @Rollback
     void "test save one-to-many"() {
         given:
+        List<Owner> deleteOwners = []
         // tag::save[]
-        new Owner(name:"Fred")
+        deleteOwners << new Owner(name:"Fred")
                 .addToPets(name: "Dino")
                 .addToPets(name: "Joe")
                 .save()
-        new Owner(name:"Barney")
+        deleteOwners << new Owner(name:"Barney")
                 .addToPets(name: "Hoppy")
                 .save(flush:true)
         // end::save[]
         Owner.withSession { it.clear() }
+
         expect:
         Owner.count == 2
         Owner.findByName("Fred").pets.size() == 2
         Owner.findByName("Barney").pets.size() == 1
+
+        cleanup:
+        Owner.deleteAll(deleteOwners)
     }
 }
diff --git a/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/cypher/OneToManyCreateSpec.groovy b/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/cypher/OneToManyCreateSpec.groovy
index ded58e28..dbf8ddc6 100644
--- a/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/cypher/OneToManyCreateSpec.groovy
+++ b/grails-datastore-gorm-neo4j/src/test/groovy/grails/gorm/tests/cypher/OneToManyCreateSpec.groovy
@@ -14,7 +14,7 @@ class OneToManyCreateSpec extends Specification {
     void "test save one-to-many"() {
         given:
         // tag::save[]
-        new Owner(name:"Fred")
+        Owner owner = new Owner(name:"David")
             .addToPets(name: "Dino")
             .save(flush:true)
             .discard()
@@ -22,13 +22,16 @@ class OneToManyCreateSpec extends Specification {
         expect:
         Owner.count == 1
         Owner.first().pets.size() == 1
+
+        cleanup:
+        Owner.deleteAll(owner)
     }
 
     @Rollback
     void "test save one-to-many with dynamic attributes"() {
 
         given:
-        def owner = new Owner(name: "Fred")
+        def owner = new Owner(name: "John")
                             .addToPets(name: "Dino")
         owner.age = 40
         owner
@@ -38,6 +41,10 @@ class OneToManyCreateSpec extends Specification {
         expect:
         Owner.count == 1
         Owner.first().pets.size() == 1
+        Owner.first().name == "John"
         Owner.first().age == 40
+
+        cleanup:
+        Owner.deleteAll(owner)
     }
 }