@@ -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 (Object .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