diff --git a/src/test/java/ntnu/idatt2016/v233/SmartMat/service/group/WasteServiceTest.java b/src/test/java/ntnu/idatt2016/v233/SmartMat/service/group/WasteServiceTest.java
index d9480d974143ca588e3ea6fa33692a6975dceef1..1517a1cb1facca1f21a73193428dd2fae12cdb5b 100644
--- a/src/test/java/ntnu/idatt2016/v233/SmartMat/service/group/WasteServiceTest.java
+++ b/src/test/java/ntnu/idatt2016/v233/SmartMat/service/group/WasteServiceTest.java
@@ -1,78 +1,230 @@
package ntnu.idatt2016.v233.SmartMat.service.group;
+import ntnu.idatt2016.v233.SmartMat.dto.request.WasteRequest;
import ntnu.idatt2016.v233.SmartMat.entity.Waste;
+import ntnu.idatt2016.v233.SmartMat.entity.group.Group;
+import ntnu.idatt2016.v233.SmartMat.entity.group.UserGroupAsso;
+import ntnu.idatt2016.v233.SmartMat.entity.product.Category;
+import ntnu.idatt2016.v233.SmartMat.entity.product.Product;
+import ntnu.idatt2016.v233.SmartMat.entity.user.User;
+import ntnu.idatt2016.v233.SmartMat.repository.group.GroupRepository;
import ntnu.idatt2016.v233.SmartMat.repository.group.WasteRepository;
+import ntnu.idatt2016.v233.SmartMat.repository.product.ProductRepository;
import ntnu.idatt2016.v233.SmartMat.service.group.WasteService;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
+import org.mockito.junit.jupiter.MockitoExtension;
import java.sql.Timestamp;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
import java.util.Optional;
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.*;
+import static org.mockito.ArgumentMatchers.*;
+import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
+@ExtendWith(MockitoExtension.class)
public class WasteServiceTest {
+ @InjectMocks
private WasteService wasteService;
@Mock
private WasteRepository wasteRepository;
- /**
+ @Mock
+ private GroupRepository groupRepository;
+
+ @Mock
+ private ProductRepository productRepository;
+
+ public Group group;
+ public Product product;
+ public Waste waste;
+ public WasteRequest wasteRequest;
+
@BeforeEach
public void setUp() {
- MockitoAnnotations.openMocks(this);
- wasteService = new WasteService(wasteRepository);
+ Category category = Category.builder().categoryName("Test").build();
+ group = Group.builder().groupId(1).build();
+ product = Product.builder().ean(1).amount(2.0).unit("kg").category(category).build();
+ waste = Waste.builder().wasteId(1).groupId(group).ean(product).amount(1).unit("kg").timestamp(new Timestamp(System.currentTimeMillis())).build();
+ wasteRequest = new WasteRequest(1, 1, 1, "kg");
+ }
+
+ @Test
+ void createWaste_Success() {
+ when(groupRepository.findByGroupId(wasteRequest.groupId())).thenReturn(Optional.of(group));
+ when(productRepository.findById(wasteRequest.ean())).thenReturn(Optional.of(product));
+ when(wasteRepository.save(any(Waste.class))).thenReturn(waste);
+
+ Optional<Waste> createdWaste = wasteService.createWaste(wasteRequest);
+
+ assertTrue(createdWaste.isPresent());
+ assertEquals(waste, createdWaste.get());
}
+ @Test
+ void createWaste_Failure() {
+ when(groupRepository.findByGroupId(wasteRequest.groupId())).thenReturn(Optional.empty());
+ when(productRepository.findById(wasteRequest.ean())).thenReturn(Optional.empty());
+
+ Optional<Waste> createdWaste = wasteService.createWaste(wasteRequest);
+ assertFalse(createdWaste.isPresent());
+ }
@Test
- public void testCreateWaste() {
- Waste waste = Waste.builder()
- .groupId(1L)
- .ean(1234567890123L)
- .timestamp(new Timestamp(System.currentTimeMillis()))
- .amount(1.0)
- .unit("kg")
- .build();
+ void getWasteById_Success() {
+ when(wasteRepository.findById(waste.getWasteId())).thenReturn(Optional.of(waste));
- Waste createdWaste = Waste.builder()
- .wasteId(1L)
- .groupId(1L)
- .ean(1234567890123L)
- .timestamp(waste.getTimestamp())
- .amount(1.0)
- .unit("kg")
- .build();
+ Optional<Waste> fetchedWaste = wasteService.getWasteById(waste.getWasteId());
+
+ assertTrue(fetchedWaste.isPresent());
+ assertEquals(waste, fetchedWaste.get());
+ }
- when(wasteRepository.save(waste)).thenReturn(createdWaste);
+ @Test
+ void getWasteById_Failure() {
+ when(wasteRepository.findById(anyLong())).thenReturn(Optional.empty());
- Waste result = wasteService.createWaste(waste);
+ Optional<Waste> fetchedWaste = wasteService.getWasteById(999L);
- assertEquals(createdWaste, result);
+ assertFalse(fetchedWaste.isPresent());
}
@Test
- public void testGetWasteById() {
- Waste waste = Waste.builder()
- .wasteId(1L)
- .groupId(1L)
- .ean(1234567890123L)
- .timestamp(new Timestamp(System.currentTimeMillis()))
- .amount(1.0)
- .unit("kg")
- .build();
+ void getWasteByGroupId_Success() {
+ when(groupRepository.findByGroupId(group.getGroupId())).thenReturn(Optional.of(group));
+ when(wasteRepository.findByGroupId(group)).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<List<Waste>> fetchedWastes = wasteService.getWasteByGroupId(group.getGroupId());
- when(wasteRepository.findById(1L)).thenReturn(Optional.of(waste));
+ assertTrue(fetchedWastes.isPresent());
+ assertEquals(1, fetchedWastes.get().size());
+ assertEquals(waste, fetchedWastes.get().get(0));
+ }
+
+ @Test
+ void getWasteByGroupId_Failure() {
+ when(groupRepository.findByGroupId(anyLong())).thenReturn(Optional.empty());
- Optional<Waste> result = wasteService.getWasteById(1L);
+ Optional<List<Waste>> fetchedWastes = wasteService.getWasteByGroupId(999L);
- assertTrue(result.isPresent());
- assertEquals(waste, result.get());
+ assertFalse(fetchedWastes.isPresent());
}
- */
-}
+
+ @Test
+ void getWasteOfCategoryByGroupId_Success() {
+ when(wasteRepository.findAllWasteOfOneCategoryFromGroup(group.getGroupId(), product.getCategory().getCategoryName())).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<List<Waste>> fetchedWastes = wasteService.getWasteOfCategoryByGroupId(group.getGroupId(), product.getCategory().getCategoryName());
+
+ assertTrue(fetchedWastes.isPresent());
+ assertEquals(1, fetchedWastes.get().size());
+ assertEquals(waste, fetchedWastes.get().get(0));
+ }
+
+ @Test
+ void getWasteOfCategoryByGroupId_Failure() {
+ when(wasteRepository.findAllWasteOfOneCategoryFromGroup(anyLong(), anyString())).thenReturn(Optional.empty());
+
+ Optional<List<Waste>> fetchedWastes = wasteService.getWasteOfCategoryByGroupId(999L, "Nonexistent Category");
+
+ assertFalse(fetchedWastes.isPresent());
+ }
+
+ @Test
+ void getCakeDiagram_Success() {
+ when(groupRepository.findByGroupId(group.getGroupId())).thenReturn(Optional.of(group));
+ when(wasteRepository.findByGroupId(group)).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<double[]> fetchedCakeDiagram = wasteService.getCakeDiagram(group.getGroupId());
+
+ assertTrue(fetchedCakeDiagram.isPresent());
+ }
+
+ @Test
+ void getCakeDiagram_Failure() {
+ when(groupRepository.findByGroupId(anyLong())).thenReturn(Optional.empty());
+
+ Optional<double[]> fetchedCakeDiagram = wasteService.getCakeDiagram(999L);
+
+ assertFalse(fetchedCakeDiagram.isPresent());
+ }
+
+ @Test
+ void getLastMonth_Success() {
+ when(groupRepository.findByGroupId(group.getGroupId())).thenReturn(Optional.of(group));
+ when(wasteRepository.findByGroupId(group)).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<double[]> fetchedLastMonthData = wasteService.getLastMonth(group.getGroupId());
+
+ assertTrue(fetchedLastMonthData.isPresent());
+ }
+
+ @Test
+ void getLastMonth_Failure() {
+ when(groupRepository.findByGroupId(anyLong())).thenReturn(Optional.empty());
+
+ Optional<double[]> fetchedLastMonthData = wasteService.getLastMonth(999L);
+
+ assertFalse(fetchedLastMonthData.isPresent());
+ }
+
+ @Test
+ void getLostMoney_Success() {
+ when(groupRepository.findByGroupId(group.getGroupId())).thenReturn(Optional.of(group));
+ when(wasteRepository.findByGroupId(group)).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<Double> fetchedLostMoney = wasteService.getLostMoney(group.getGroupId());
+
+ assertTrue(fetchedLostMoney.isPresent());
+ }
+
+ @Test
+ void getLostMoney_Failure() {
+ when(groupRepository.findByGroupId(anyLong())).thenReturn(Optional.empty());
+
+ Optional<Double> fetchedLostMoney = wasteService.getLostMoney(999L);
+
+ assertFalse(fetchedLostMoney.isPresent());
+ }
+
+ @Test
+ void getCO2PerPerson_Success() {
+ when(groupRepository.findByGroupId(group.getGroupId())).thenReturn(Optional.of(group));
+ when(groupRepository.countAllUserInGroup(group.getGroupId())).thenReturn(1);
+ when(wasteRepository.findByGroupId(group)).thenReturn(Optional.of(Collections.singletonList(waste)));
+
+ Optional<Double> fetchedCO2PerPerson = wasteService.getCO2PerPerson(group.getGroupId());
+
+ assertTrue(fetchedCO2PerPerson.isPresent());
+ }
+
+ @Test
+ void getCO2PerPerson_Failure() {
+ when(groupRepository.findByGroupId(anyLong())).thenReturn(Optional.empty());
+
+ Optional<Double> fetchedCO2PerPerson = wasteService.getCO2PerPerson(999L);
+
+ assertFalse(fetchedCO2PerPerson.isPresent());
+ }
+
+
+ @Test
+ void isUserAssosiatedWithWaste_Failure() {
+ when(wasteRepository.findById(anyLong())).thenReturn(Optional.empty());
+
+ boolean isUserAssociated = wasteService.isUserAssosiatedWithWaste("TestUsername", 999L);
+
+ assertFalse(isUserAssociated);
+ }
+
+}
\ No newline at end of file