diff --git a/src/main/java/hs/fulda/de/ci/exam/project/Customer.java b/src/main/java/hs/fulda/de/ci/exam/project/Customer.java index a85e014..4d8491f 100644 --- a/src/main/java/hs/fulda/de/ci/exam/project/Customer.java +++ b/src/main/java/hs/fulda/de/ci/exam/project/Customer.java @@ -1,9 +1,14 @@ package hs.fulda.de.ci.exam.project; +import java.util.ArrayList; + public class Customer extends Person{ private String frequentFlyerNumber; + private ItineraryRepository itineraryRepository; + + public Customer(String name, Address address, String email, String phone) { super(name, address, email, phone); @@ -16,4 +21,8 @@ public class Customer extends Person{ public String getFrequentFlyerNumber() { return frequentFlyerNumber; } + + public ArrayList getItineraries() { + return itineraryRepository.findAll(); + } } diff --git a/src/test/java/hs/fulda/de/ci/exam/project/CustomerTest.java b/src/test/java/hs/fulda/de/ci/exam/project/CustomerTest.java index 6c53905..dafeff1 100644 --- a/src/test/java/hs/fulda/de/ci/exam/project/CustomerTest.java +++ b/src/test/java/hs/fulda/de/ci/exam/project/CustomerTest.java @@ -1,17 +1,22 @@ package hs.fulda.de.ci.exam.project; import org.junit.jupiter.api.Test; - +import org.junit.jupiter.api.extension.ExtendWith; +import org.mockito.InjectMocks; +import org.mockito.Mock; +import static org.mockito.Mockito.*; +import org.mockito.junit.jupiter.MockitoExtension; import java.lang.reflect.Field; - -import static org.assertj.core.api.Assertions.assertThat; +import java.util.ArrayList; import static org.junit.jupiter.api.Assertions.assertEquals; +@ExtendWith(MockitoExtension.class) public class CustomerTest { final Address address1 = new Address("Fuldaer str", "Fulda", "Hessen", "36037", "Germany"); - final Customer person1 = new Customer("Max Mustermann", address1, "max.mustermann@gmail.com", "015147890206"); + @InjectMocks + final Customer person1 = new Customer("Max Mustermann", address1, "max.mustermann@gmail.com", "015147890206"); @Test public void test_setFlyingNumber() throws NoSuchFieldException, IllegalAccessException{ @@ -35,4 +40,25 @@ public class CustomerTest { assertEquals(result, "10","get frequent flying number of the customer"); } + @Mock + private ItineraryRepository itineraryRepo; + + @Test + public void test_getItineraries() { + ArrayList itineraries = new ArrayList<>(); + + Airport airport_fr = new Airport("Fraport", address1, "1234"); + Airport airport_be = new Airport("Berlin", address1, "5678"); + Itinerary item1 = new Itinerary(airport_fr, airport_be); + Itinerary item2 = new Itinerary(airport_be, airport_fr); + itineraries.add(item1); + itineraries.add(item2); + + when(itineraryRepo.findAll()).thenReturn(itineraries); + + ArrayList new_itineraries = person1.getItineraries(); + + assertEquals(2, new_itineraries.size(), "Sizes are equal"); + } + }