- when(extensionConverterProvider.getConverter(KEY))
- .thenReturn((input, path) -> {
- MockAugmentation mockAugmentation = new MockAugmentation();
- return new ExtensionAugment<>(MockAugmentation.class, mockAugmentation, MockExtensionKey.class);
- });
+ when(extensionConverterProvider.getConverter(KEY_1))
+ .thenReturn((input, path) -> {
+ MockAugmentation mockAugmentation = new MockAugmentation();
+ return new ExtensionAugment<>(MockAugmentation.class, mockAugmentation, MockExtensionKey1.class);
+ });
+ when(extensionConverterProvider.getConverter(KEY_2))
+ .thenReturn((input, path) -> {
+ MockAugmentation mockAugmentation = new MockAugmentation();
+ return new ExtensionAugment<>(MockAugmentation.class, mockAugmentation, MockExtensionKey2.class);
+ });