From 93dfb77f1f9e8e86631014bf23b3621cd8ebf390 Mon Sep 17 00:00:00 2001
From: dionisPrifti <dionis_sds@hotmail.com>
Date: Fri, 24 May 2019 18:10:52 +0200
Subject: [PATCH] Bael 2810 data jpa delete (#6759)

* BAEL-2810: Finished the examples and unit tests.

* BAEL-2810: Changed assertions to isEqualTo

* BAEL-2810: Changed the test of the derived query to reflect the number of the deleted records
---
 .../datajpadelete/DeleteFromRepositoryUnitTest.java    | 10 +++++-----
 .../datajpadelete/DeleteInRelationshipsUnitTest.java   |  8 ++++----
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteFromRepositoryUnitTest.java b/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteFromRepositoryUnitTest.java
index 9e7e516735..5f4a36bc0e 100644
--- a/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteFromRepositoryUnitTest.java
+++ b/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteFromRepositoryUnitTest.java
@@ -43,22 +43,22 @@ public class DeleteFromRepositoryUnitTest {
     public void whenDeleteByIdFromRepository_thenDeletingShouldBeSuccessful() {
         repository.deleteById(book1.getId());
 
-        assertThat(repository.count() == 1).isTrue();
+        assertThat(repository.count()).isEqualTo(1);
     }
 
     @Test
     public void whenDeleteAllFromRepository_thenRepositoryShouldBeEmpty() {
         repository.deleteAll();
 
-        assertThat(repository.count() == 0).isTrue();
+        assertThat(repository.count()).isEqualTo(0);
     }
 
     @Test
     @Transactional
     public void whenDeleteFromDerivedQuery_thenDeletingShouldBeSuccessful() {
-        repository.deleteByTitle("The Hobbit");
+        long deletedRecords = repository.deleteByTitle("The Hobbit");
 
-        assertThat(repository.count() == 1).isTrue();
+        assertThat(deletedRecords).isEqualTo(1);
     }
 
     @Test
@@ -66,7 +66,7 @@ public class DeleteFromRepositoryUnitTest {
     public void whenDeleteFromCustomQuery_thenDeletingShouldBeSuccessful() {
         repository.deleteBooks("The Hobbit");
 
-        assertThat(repository.count() == 1).isTrue();
+        assertThat(repository.count()).isEqualTo(1);
     }
 
 }
\ No newline at end of file
diff --git a/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteInRelationshipsUnitTest.java b/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteInRelationshipsUnitTest.java
index 56de8749b2..6275ace6e0 100644
--- a/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteInRelationshipsUnitTest.java
+++ b/persistence-modules/spring-data-jpa-2/src/test/java/com/baeldung/datajpadelete/DeleteInRelationshipsUnitTest.java
@@ -46,15 +46,15 @@ public class DeleteInRelationshipsUnitTest {
     public void whenDeletingCategories_thenBooksShouldAlsoBeDeleted() {
         categoryRepository.deleteAll();
 
-        assertThat(bookRepository.count() == 0).isTrue();
-        assertThat(categoryRepository.count() == 0).isTrue();
+        assertThat(bookRepository.count()).isEqualTo(0);
+        assertThat(categoryRepository.count()).isEqualTo(0);
     }
 
     @Test
     public void whenDeletingBooks_thenCategoriesShouldAlsoBeDeleted() {
         bookRepository.deleteAll();
 
-        assertThat(bookRepository.count() == 0).isTrue();
-        assertThat(categoryRepository.count() == 2).isTrue();
+        assertThat(bookRepository.count()).isEqualTo(0);
+        assertThat(categoryRepository.count()).isEqualTo(2);
     }
 }
\ No newline at end of file