Skip to content

Commit dadb755

Browse files
committed
Fix testComplexTypeKey()
1 parent 4469bf6 commit dadb755

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

msgpack-jackson/src/test/java/org/msgpack/jackson/dataformat/MessagePackGeneratorTest.java

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,9 @@ public void testNonStringKey()
663663

664664
ObjectMapper objectMapper = new ObjectMapper(new MessagePackFactory());
665665
if (mapHolder instanceof NonStringKeyMapHolderWithoutAnnotation) {
666-
objectMapper.setSerializerFactory(new MessagePackSerializerFactory());
666+
SimpleModule mod = new SimpleModule("test");
667+
mod.addKeySerializer(TinyPojo.class, new MessagePackKeySerializer());
668+
objectMapper.registerModule(mod);
667669
}
668670

669671
byte[] bytes = objectMapper.writeValueAsBytes(mapHolder);
@@ -709,7 +711,9 @@ public void testComplexTypeKey()
709711
map.put(pojo, 42);
710712

711713
ObjectMapper objectMapper = new ObjectMapper(new MessagePackFactory());
712-
objectMapper.setSerializerFactory(new MessagePackSerializerFactory());
714+
SimpleModule mod = new SimpleModule("test");
715+
mod.addKeySerializer(TinyPojo.class, new MessagePackKeySerializer());
716+
objectMapper.registerModule(mod);
713717
byte[] bytes = objectMapper.writeValueAsBytes(map);
714718

715719
MessageUnpacker unpacker = MessagePack.newDefaultUnpacker(bytes);
@@ -731,7 +735,9 @@ public void testComplexTypeKeyWithV06Format()
731735

732736
ObjectMapper objectMapper = new ObjectMapper(new MessagePackFactory());
733737
objectMapper.setAnnotationIntrospector(new JsonArrayFormat());
734-
objectMapper.setSerializerFactory(new MessagePackSerializerFactory());
738+
SimpleModule mod = new SimpleModule("test");
739+
mod.addKeySerializer(TinyPojo.class, new MessagePackKeySerializer());
740+
objectMapper.registerModule(mod);
735741
byte[] bytes = objectMapper.writeValueAsBytes(map);
736742

737743
MessageUnpacker unpacker = MessagePack.newDefaultUnpacker(bytes);

0 commit comments

Comments
 (0)