Fixed supply mapper to map alse the user

lucoenergia · Mar 21, 2024 · 48ed742 · 48ed742
1 parent 253fb3f
commit 48ed742
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/main/java/org/lucoenergia/conluz/infrastructure/admin/supply/SupplyEntityMapper.java b/src/main/java/org/lucoenergia/conluz/infrastructure/admin/supply/SupplyEntityMapper.java
@@ -2,11 +2,18 @@
 
 import org.lucoenergia.conluz.domain.admin.supply.Supply;
 import org.lucoenergia.conluz.domain.shared.BaseMapper;
+import org.lucoenergia.conluz.infrastructure.admin.user.UserEntityMapper;
 import org.springframework.stereotype.Component;
 
 @Component
 public class SupplyEntityMapper extends BaseMapper<SupplyEntity, Supply> {
 
+    private final UserEntityMapper userEntityMapper;
+
+    public SupplyEntityMapper(UserEntityMapper userEntityMapper) {
+        this.userEntityMapper = userEntityMapper;
+    }
+
     @Override
     public Supply map(SupplyEntity entity) {
         return new Supply.Builder()
@@ -16,6 +23,7 @@ public Supply map(SupplyEntity entity) {
                 .withPartitionCoefficient(entity.getPartitionCoefficient())
                 .withEnabled(entity.getEnabled())
                 .withName(entity.getName())
+                .withUser(userEntityMapper.map(entity.getUser()))
                 .build();
     }
 }
diff --git a/src/test/java/org/lucoenergia/conluz/infrastructure/admin/supply/SupplyEntityMapperTest.java b/src/test/java/org/lucoenergia/conluz/infrastructure/admin/supply/SupplyEntityMapperTest.java
@@ -3,10 +3,12 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 import org.lucoenergia.conluz.domain.admin.supply.Supply;
+import org.lucoenergia.conluz.infrastructure.admin.user.UserEntityMapper;
 
 class SupplyEntityMapperTest {
 
-    private final SupplyEntityMapper mapper = new SupplyEntityMapper();
+    private final UserEntityMapper userEntityMapper = new UserEntityMapper();
+    private final SupplyEntityMapper mapper = new SupplyEntityMapper(userEntityMapper);
 
     @Test
     void testMap() {
@@ -19,5 +21,6 @@ void testMap() {
         Assertions.assertEquals(entity.getAddress(), result.getAddress());
         Assertions.assertEquals(entity.getPartitionCoefficient(), result.getPartitionCoefficient());
         Assertions.assertEquals(entity.getEnabled(), result.getEnabled());
+        Assertions.assertEquals(entity.getUser().getPersonalId(), result.getUser().getPersonalId());
     }
 }