Merge pull request #378 from 0xPolygonID/fix/display_type_mapper

Fixed claim display type mapper.
0xPolygonID · Feb 1, 2024 · 01e62f6 · 01e62f6
2 parents 5e6e70a + 9e41fda
commit 01e62f6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 13 deletions.
diff --git a/lib/credential/data/mappers/claim_mapper.dart b/lib/credential/data/mappers/claim_mapper.dart
@@ -29,9 +29,8 @@ class ClaimMapper extends Mapper<ClaimDTO, ClaimEntity> {
       schema: from.schema,
       type: from.type,
       info: _claimInfoMapper.mapFrom(from.info),
-      displayType: from.displayType == null
-          ? _displayTypeMapper.mapFrom(displayType)
-          : null,
+      displayType:
+          displayType != null ? _displayTypeMapper.mapFrom(displayType) : null,
       credentialRawValue: from.credentialRawValue,
     );
   }
@@ -49,7 +48,7 @@ class ClaimMapper extends Mapper<ClaimDTO, ClaimEntity> {
       schema: to.schema,
       info: _claimInfoMapper.mapTo(to.info),
       displayType:
-          to.displayType == null ? null : _displayTypeMapper.mapTo(displayType),
+          displayType != null ? _displayTypeMapper.mapTo(displayType) : null,
       credentialRawValue: to.credentialRawValue,
     );
   }

diff --git a/lib/credential/data/mappers/display_type_mapper.dart b/lib/credential/data/mappers/display_type_mapper.dart
@@ -18,10 +18,3 @@ class DisplayTypeMapper extends Mapper<Map<String, dynamic>?, DisplayType?> {
     return json;
   }
 }
-
-class DisplayTypeMapperParam {
-  final String type;
-  final Map<String, dynamic> json;
-
-  DisplayTypeMapperParam(this.type, this.json);
-}
diff --git a/test/credential/data/mappers/claim_mapper_test.dart b/test/credential/data/mappers/claim_mapper_test.dart
@@ -71,8 +71,7 @@ void main() {
       expect(verify(infoMapper.mapFrom(captureAny)).captured.first,
           fetchClaimDTO.credential);
       expect(verify(stateMapper.mapFrom(captureAny)).captured.first, '');
-      expect(
-          verify(displayTypeMapper.mapFrom(captureAny)).captured.first, null);
+      verifyNever(displayTypeMapper.mapFrom(captureAny));
     });
   });