get itineraries from repository

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<Itinerary> getItineraries() {
+        return itineraryRepository.findAll();
+    }
 }

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<Itinerary> 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<Itinerary> new_itineraries = person1.getItineraries();
+
+        assertEquals(2, new_itineraries.size(), "Sizes are equal");
+    }
+
 }