Updated tests for user class

1 files changed, 8 insertions, 6 deletions

diff --git a/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java b/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
index ad44d58..1b4f804 100644
--- a/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
+++ b/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
@@ -3,9 +3,8 @@ package com.ufund.api.ufundapi.model;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import org.junit.jupiter.api.Test;
-import org.springframework.boot.test.context.SpringBootTest;
 
-@SpringBootTest
+
 public class UserTest {
     
     @Test
@@ -31,16 +30,19 @@ public class UserTest {
     }
 
     @Test
-    public void testCopy() {
+    public void addNeedToBasket() {
 
         String expectedName = "Bob";
 
         User user = new User(expectedName);
-        User user2 = new User(user);
-        
+        Need need = new Need("Test", 0, 100, Need.GoalType.MONETARY);
+        Need[] needs = {need};
 
-        assertEquals(user, user2);
+        user.addToBasket(need);
+
+        assertEquals(needs[0], user.getBasketNeeds()[0]);
 
     }
 
+
 }