1111import org .commonmark .testutil .RenderingTestCase ;
1212import org .junit .jupiter .api .Test ;
1313
14- import java .util .HashMap ;
1514import java .util .List ;
1615import java .util .Map ;
1716import java .util .Set ;
@@ -47,10 +46,8 @@ private void assertRenderingCustomTitles(String source, String expectedResult) {
4746
4847 @ Test
4948 public void customType () {
50- var customTypes = new HashMap <>(AlertsExtension .STANDARD_TYPES );
51- customTypes .put ("INFO" , "Information" );
5249 var extension = AlertsExtension .builder ()
53- .setAllowedTypes ( customTypes )
50+ .addCustomType ( "INFO" , "Information" )
5451 .build ();
5552
5653 var parser = Parser .builder ().extensions (Set .of (extension )).build ();
@@ -66,11 +63,9 @@ public void customType() {
6663 @ Test
6764 public void multipleCustomTypes () {
6865 var extension = AlertsExtension .builder ()
69- .setAllowedTypes (Map .ofEntries (
70- Map .entry ("INFO" , "Information" ),
71- Map .entry ("SUCCESS" , "Success!" ),
72- Map .entry ("DANGER" , "Danger!" )
73- ))
66+ .addCustomType ("INFO" , "Information" )
67+ .addCustomType ("SUCCESS" , "Success!" )
68+ .addCustomType ("DANGER" , "Danger!" )
7469 .build ();
7570
7671 var parser = Parser .builder ().extensions (Set .of (extension )).build ();
@@ -93,10 +88,8 @@ public void multipleCustomTypes() {
9388
9489 @ Test
9590 public void standardTypesWithCustomConfigured () {
96- var customTypes = new HashMap <>(AlertsExtension .STANDARD_TYPES );
97- customTypes .put ("INFO" , "Information" );
9891 var extension = AlertsExtension .builder ()
99- .setAllowedTypes ( customTypes )
92+ .addCustomType ( "INFO" , "Information" )
10093 .build ();
10194
10295 var parser = Parser .builder ().extensions (Set .of (extension )).build ();
@@ -111,10 +104,8 @@ public void standardTypesWithCustomConfigured() {
111104
112105 @ Test
113106 public void overrideStandardTypeTitle () {
114- var customTypes = new HashMap <>(AlertsExtension .STANDARD_TYPES );
115- customTypes .put ("NOTE" , "Nota" );
116107 var extension = AlertsExtension .builder ()
117- .setAllowedTypes ( customTypes )
108+ .addCustomType ( "NOTE" , "Nota" )
118109 .build ();
119110
120111 var parser = Parser .builder ().extensions (Set .of (extension )).build ();
@@ -132,19 +123,19 @@ public void overrideStandardTypeTitle() {
132123 @ Test
133124 public void customTypeMustBeUppercase () {
134125 assertThrows (IllegalArgumentException .class , () ->
135- AlertsExtension .builder ().setAllowedTypes ( Map . of ( "info" , "Information" ) ).build ());
126+ AlertsExtension .builder ().addCustomType ( "info" , "Information" ).build ());
136127 }
137128
138129 @ Test
139130 public void customTypeMustNotBeEmpty () {
140131 assertThrows (IllegalArgumentException .class , () ->
141- AlertsExtension .builder ().setAllowedTypes ( Map . of ( "" , "Title" ) ).build ());
132+ AlertsExtension .builder ().addCustomType ( "" , "Title" ).build ());
142133 }
143134
144135 @ Test
145136 public void customTypeTitleMustNotBeEmpty () {
146137 assertThrows (IllegalArgumentException .class , () ->
147- AlertsExtension .builder ().setAllowedTypes ( Map . of ( "INFO" , "" ) ).build ());
138+ AlertsExtension .builder ().addCustomType ( "INFO" , "" ).build ());
148139 }
149140
150141 @ Test
@@ -440,10 +431,8 @@ public void alertParsedAsAlertNode() {
440431
441432 @ Test
442433 public void customTypeParsedAsAlertNode () {
443- var customTypes = new HashMap <>(AlertsExtension .STANDARD_TYPES );
444- customTypes .put ("INFO" , "Information" );
445434 var extension = AlertsExtension .builder ()
446- .setAllowedTypes ( customTypes )
435+ .addCustomType ( "INFO" , "Information" )
447436 .build ();
448437
449438 var parser = Parser .builder ().extensions (Set .of (extension )).build ();
0 commit comments