feat: adds printSponsors() tests

Author: Evan Philakhong

lesson_16/objects/objects_app/src/main/java/com/codedifferently/lesson16/evanphilakhong/Formula1Car.java

@@ -71,10 +71,4 @@ public boolean rainStrategy() {
     }
     return false;
   }
-
-  public static void main(String[] args) {
-    Formula1Car formula1Car = new Formula1Car();
-    Team team = Team.MCLAREN;
-    formula1Car.setTeam(team);
-  }
 }

lesson_16/objects/objects_app/src/test/java/com/codedifferently/lesson16/evanphilakhong/Formula1CarTest.java

@@ -2,10 +2,12 @@
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.io.ByteArrayOutputStream;
 import java.io.PrintStream;
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
@@ -14,28 +16,44 @@ class Formula1CarTest {
   private final ByteArrayOutputStream outContent = new ByteArrayOutputStream();
   Track track;
   Formula1Car f1Car1;
+  Formula1Car f1Car2;
 
   @BeforeEach
   void setUp() {
     track = new Track();
+
     String[] f1Car1Sponsors = {"Hp", "Shell", "UniCredit", "IBM", "Puma", "VGW Play"};
     f1Car1 =
         new Formula1Car(Team.FERRARI, f1Car1Sponsors, "Lewis Hamilton", 44, 8, TyreCompound.MEDIUM);
     f1Car1.setTrack(track);
+
+    String[] f1Car2Sponsors = {};
+    f1Car2 =
+        new Formula1Car(Team.MCLAREN, f1Car2Sponsors, "Lando Norris", 4, 2, TyreCompound.MEDIUM);
+    f1Car2.setTrack(track);
+
     System.setOut(new PrintStream(outContent));
   }
 
-  @Test
-  void testIsDrsAvailibe() {
-    assertFalse(f1Car1.isDrsAvailible());
+  @AfterEach
+  void restoreStreams() {
+    System.setOut(System.out);
+  }
 
-    // act
+  @Test
+  void testIsDrsAvailibe_returnsTrue() {
     track.setDrsZone(true);
 
-    // assert
     assertTrue(f1Car1.isDrsAvailible());
   }
 
+  @Test
+  void testIsDrsAvailibe_returnsFalse() {
+    track.setDrsZone(false);
+
+    assertFalse(f1Car1.isDrsAvailible());
+  }
+
   @Test
   void testRainStrategy_returnsTrue() {
     // checks if tyres change depending on weather conditions
@@ -55,7 +73,7 @@ void testRainStrategy_returnsFalse() {
   }
 
   @Test
-  void testPrintSponsors() throws Exception {
+  void testPrintSponsors_withSponsors() throws NoSponsorsException {
     if (f1Car1.getSponsors().length == 0) {
       throw new NoSponsorsException(f1Car1.getTeam() + " has no Sponsors");
     } else {
@@ -80,4 +98,12 @@ void testPrintSponsors() throws Exception {
       assertEquals(expected, outContent.toString());
     }
   }
+
+  @Test
+  void testPrintSponsors_noSponsors() {
+    NoSponsorsException exception =
+        assertThrows(NoSponsorsException.class, () -> f1Car2.printSponsors());
+
+    assertEquals(f1Car2.getTeam() + " has no Sponsors", exception.getMessage());
+  }
 }