diff --git a/app/src/main/java/com/example/simpleparadox/listycity/CityList.java b/app/src/main/java/com/example/simpleparadox/listycity/CityList.java index 6da242f3..5f901957 100644 --- a/app/src/main/java/com/example/simpleparadox/listycity/CityList.java +++ b/app/src/main/java/com/example/simpleparadox/listycity/CityList.java @@ -77,4 +77,13 @@ public void delete(City city) { public int countCities() { return cities.size(); } + + public boolean checkEmpty(){ + if (cities.size() == 0) { + return true; + } + + return false; + } + } diff --git a/app/src/test/java/com/example/simpleparadox/listycity/CityListTest.java b/app/src/test/java/com/example/simpleparadox/listycity/CityListTest.java index 787eab7e..c225be9d 100644 --- a/app/src/test/java/com/example/simpleparadox/listycity/CityListTest.java +++ b/app/src/test/java/com/example/simpleparadox/listycity/CityListTest.java @@ -3,6 +3,9 @@ import org.junit.jupiter.api.Test; +import java.util.ArrayList; +import java.util.List; + import static org.junit.jupiter.api.Assertions.*; class CityListTest { @@ -93,4 +96,15 @@ void testCountCities() { assertEquals(1, cityList.countCities()); } + + @Test + void testCheckEmpty() { + CityList cityList = new CityList(); + City city = new City("Victoria", "British Columbia"); + cityList.add(city); + assertFalse(cityList.checkEmpty()); + cityList.delete(city); + assertTrue(cityList.checkEmpty()); + + } } \ No newline at end of file