From 58ad03cbe1037964cf7124a7fc188b5dfb2e168e Mon Sep 17 00:00:00 2001
From: Grzegorz Piwowarek <gpiwowarek@gmail.com>
Date: Sun, 23 Sep 2018 07:28:22 +0200
Subject: [PATCH] Refator tests

---
 .../java/com/baeldung/protonpack/ProtonpackUnitTest.java     | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/java-streams/src/test/java/com/baeldung/protonpack/ProtonpackUnitTest.java b/java-streams/src/test/java/com/baeldung/protonpack/ProtonpackUnitTest.java
index bcda1162df..371ac3a9bb 100644
--- a/java-streams/src/test/java/com/baeldung/protonpack/ProtonpackUnitTest.java
+++ b/java-streams/src/test/java/com/baeldung/protonpack/ProtonpackUnitTest.java
@@ -20,7 +20,6 @@ import static java.util.Arrays.stream;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
 
-@SuppressWarnings("unchecked")
 public class ProtonpackUnitTest {
     @Test
     public void whenTakeWhile_thenTakenWhile() {
@@ -183,14 +182,14 @@ public class ProtonpackUnitTest {
     public void givenProjectionFunction_whenMaxedBy_thenMaxedBy() {
         Stream<String> clubs = Stream.of("Juventus", "Barcelona", "PSG");
         Optional<String> longestName = clubs.collect(CollectorUtils.maxBy(String::length));
-        assertThat(longestName.get()).isEqualTo("Barcelona");
+        assertThat(longestName).contains("Barcelona");
     }
 
     @Test
     public void givenStreamOfMultipleElem_whenUniqueCollector_thenValueReturned() {
         Stream<Integer> singleElement = Stream.of(1);
         Optional<Integer> unique = singleElement.collect(CollectorUtils.unique());
-        assertThat(unique.get()).isEqualTo(1);
+        assertThat(unique).contains(1);
     }
 
     @Test