From c267f39e00382f3b92478fff6830266ee44d5a83 Mon Sep 17 00:00:00 2001 From: Ryan Laddusaw Date: Fri, 11 Sep 2020 16:24:05 -0500 Subject: [PATCH] Updated product model tests --- .../edu/tamu/app/model/ProductModelTest.java | 24 +++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/src/test/java/edu/tamu/app/model/ProductModelTest.java b/src/test/java/edu/tamu/app/model/ProductModelTest.java index 70964ac9..432170e1 100644 --- a/src/test/java/edu/tamu/app/model/ProductModelTest.java +++ b/src/test/java/edu/tamu/app/model/ProductModelTest.java @@ -2,6 +2,8 @@ import static org.junit.Assert.assertEquals; +import java.util.ArrayList; +import java.util.Arrays; import java.util.List; import org.junit.Test; @@ -125,4 +127,26 @@ public void testRemoveRemoteProduct() { assertEquals("Product did not correctly add the remote project!", false, remoteProjectInfo.contains(TEST_REMOTE_PROJECT_INFO1)); } + @Test + public void testSetOtherUrls() { + Product product = new Product(TEST_PRODUCT_NAME, TEST_PRODUCT_REMOTE_PROJECT_INFO_LIST1); + product.setOtherUrls(TEST_OTHER_URLS); + assertEquals("Product did not correctly set the other URLs", 2, product.getOtherUrls().size()); + } + + @Test + public void testAddOtherUrl() { + Product product = new Product(TEST_PRODUCT_NAME, TEST_PRODUCT_REMOTE_PROJECT_INFO_LIST1, TEST_PROJECT_SCOPE1, "", "", "", "", new ArrayList(Arrays.asList(TEST_OTHER_URL_1))); + product.addOtherUrl(TEST_OTHER_URL_2); + assertEquals("Product did not correctly add the second URL", 2, product.getOtherUrls().size()); + } + + @Test + public void testRemoveOtherUrl() { + Product product = new Product(TEST_PRODUCT_NAME, TEST_PRODUCT_REMOTE_PROJECT_INFO_LIST1, TEST_PROJECT_SCOPE1, "", "", "", "", TEST_OTHER_URLS); + product.removeOtherUrl(TEST_OTHER_URL_1); + assertEquals("Product did not remove other URL", 1, product.getOtherUrls().size()); + assertEquals("Product did not remove correct other URL", TEST_OTHER_URL_2, product.getOtherUrls().get(0)); + } + }