From 619179f3e97bc75cae7fd778a6f27671d9a82c55 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Diana=20Gul=C4=8D=C3=ADkov=C3=A1?= <xgulcik@fi.muni.cz>
Date: Sun, 16 Apr 2023 21:33:38 +0200
Subject: [PATCH] added functionality to tests
---
.../pa165/driver/rest/DriverRepositoryTest.java | 17 +++++++++++------
.../muni/pa165/driver/rest/DriverTestUtil.java | 3 +--
2 files changed, 12 insertions(+), 8 deletions(-)
diff --git a/driver/src/test/java/cz/muni/pa165/driver/rest/DriverRepositoryTest.java b/driver/src/test/java/cz/muni/pa165/driver/rest/DriverRepositoryTest.java
index eaa1bc32..50e813f6 100644
--- a/driver/src/test/java/cz/muni/pa165/driver/rest/DriverRepositoryTest.java
+++ b/driver/src/test/java/cz/muni/pa165/driver/rest/DriverRepositoryTest.java
@@ -32,7 +32,7 @@ class DriverRepositoryTest {
Driver savedDriver = driverRepository.save(driver);
Assertions.assertEquals(savedDriver, driver);
- Assertions.assertEquals(entityManager.find(Driver.class, 1L).getId(), 1);
+ Assertions.assertEquals(entityManager.find(Driver.class, savedDriver.getId()), savedDriver);
}
@Test
@@ -59,6 +59,8 @@ class DriverRepositoryTest {
() -> Assertions.assertNotEquals(original.getSurname(), updatedDriver.getSurname()),
() -> Assertions.assertNotEquals(original.getNationality(), updatedDriver.getNationality())
);
+ Assertions.assertEquals(entityManager.find(Driver.class, savedDriver.getId()), savedDriver);
+
}
@Test
@@ -90,15 +92,18 @@ class DriverRepositoryTest {
.characteristics(Map.of()).build();
Driver driver2 = Driver.builder()
- .name("Name")
- .surname("Surname")
- .nationality("Nationality")
+ .name("Name2")
+ .surname("Surname2")
+ .nationality("Nationality2")
.characteristics(Map.of()).build();
- driverRepository.save(driver);
- driverRepository.save(driver2);
+ var savedDriver1 = driverRepository.save(driver);
+ var savedDriver2 = driverRepository.save(driver2);
var savedDrivers = driverRepository.findAll();
Assertions.assertEquals(2, savedDrivers.size());
+
+ Assertions.assertEquals(entityManager.find(Driver.class, savedDriver1.getId()), savedDriver1);
+ Assertions.assertEquals(entityManager.find(Driver.class, savedDriver2.getId()), savedDriver2);
}
}
diff --git a/driver/src/test/java/cz/muni/pa165/driver/rest/DriverTestUtil.java b/driver/src/test/java/cz/muni/pa165/driver/rest/DriverTestUtil.java
index 9fa615fa..17636a80 100644
--- a/driver/src/test/java/cz/muni/pa165/driver/rest/DriverTestUtil.java
+++ b/driver/src/test/java/cz/muni/pa165/driver/rest/DriverTestUtil.java
@@ -17,9 +17,8 @@ public class DriverTestUtil {
* @return add driver dto object.
*/
public static DriverAddDto getAddDriver() {
- var driver = new DriverAddDto("Name",
+ return new DriverAddDto("Name",
"Surname", "Nationality", new HashMap<>());
- return driver;
}
/**
--
GitLab