add and get flights tests

src/main/java/hs/fulda/de/ci/exam/project/Airline.java

@@ -1,12 +1,16 @@
 package hs.fulda.de.ci.exam.project;
 
+import java.util.HashSet;
+
 public class Airline {
     private String name;
     private String code;
+    private HashSet<Flight> flights;
 
     public Airline(String name, String code) {
         this.name = name;
         this.code = code;
+        this.flights = new HashSet<>();
     }
 
     public String getName() {
@@ -24,4 +28,12 @@ public class Airline {
     public void setCode(String code) {
         this.code = code;
     }
+
+    public HashSet<Flight> getFlights() {
+        return flights;
+    }
+
+    public void addFlight(Flight flight) {
+        flights.add(flight);
+    }
 }

src/test/java/hs/fulda/de/ci/exam/project/AirlineTest.java

@@ -4,19 +4,50 @@ import org.junit.jupiter.api.Test;
 import org.junit.runner.RunWith;
 import org.mockito.junit.MockitoJUnitRunner;
 
+import java.util.HashSet;
+
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.junit.jupiter.api.Assertions.fail;
+import static org.mockito.Mockito.mock;
 
 @RunWith(MockitoJUnitRunner.class)
 public class AirlineTest {
     @Test
-    void testConstructorSetsNameParameterCorrectly(){
+    void testConstructorSetsNameParameterCorrectly() {
         Airline airline = new Airline("Pegasus", "1234");
-        assertThat(airline.getName()).describedAs("name of airline").isEqualTo("Pegasus");
+        assertThat(airline.getName())
+                .describedAs("name of airline")
+                .isEqualTo("Pegasus");
     }
+
     @Test
-    void testConstructorSetsCodeParameterCorrectly(){
+    void testConstructorSetsCodeParameterCorrectly() {
         Airline airline = new Airline("Pegasus", "1234");
-        assertThat(airline.getCode()).describedAs("code of airline").isEqualTo("1234");
+        assertThat(airline.getCode())
+                .describedAs("code of airline")
+                .isEqualTo("1234");
     }
+
+    @Test
+    void getFlightsReturnsEmptyListOfFlights() {
+        Airline airline = new Airline("Pegasus", "1234");
+        HashSet<Flight> flights = new HashSet<>();
+        assertThat(airline.getFlights())
+                .describedAs("flights of airline")
+                .isEqualTo(flights);
+    }
+
+    @Test
+    void addFlightAddsFlightToListOfFlights() {
+        Airline airline = new Airline("Pegasus", "1234");
+        Flight flight1 = mock(Flight.class);
+        HashSet<Flight> flights = new HashSet<>();
+        flights.add(flight1);
+
+        airline.addFlight(flight1);
+
+        assertThat(airline.getFlights())
+                .describedAs("flights of airline")
+                .isEqualTo(flights);
+    }
+
 }