Updated product model tests

TAMULib · Sep 11, 2020 · c267f39 · c267f39
1 parent 75b1051
commit c267f39
Showing 1 changed file with 24 additions and 0 deletions.
diff --git 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<String>(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));
+    }
+
 }