1616import java .util .List ;
1717
1818import io .split .android .client .api .EventMetadata ;
19+ import io .split .android .client .api .MetadataKey ;
20+ import io .split .android .client .api .SdkUpdateMetadataKeys ;
1921
2022public class EventMetadataBuilderTest {
2123
@@ -76,7 +78,8 @@ public void putIgnoresValueWhenValidatorReturnsFalse() {
7678 .put ("key" , "value" )
7779 .build ();
7880
79- assertFalse (metadata .containsKey ("key" ));
81+ MetadataKey <String > KEY = new MetadataKey <>("key" );
82+ assertFalse (metadata .containsKey (KEY ));
8083 }
8184
8285 @ Test
@@ -87,14 +90,15 @@ public void putIncludesValueWhenValidatorReturnsTrue() {
8790 .put ("key" , "value" )
8891 .build ();
8992
90- assertEquals ("value" , metadata .get ("key" ));
93+ MetadataKey <String > KEY = new MetadataKey <>("key" );
94+ assertEquals ("value" , metadata .get (KEY ));
9195 }
9296
9397 @ Test
9498 public void buildCreatesEmptyMetadataWhenNothingAdded () {
9599 EventMetadata metadata = new EventMetadataBuilder ().build ();
96100
97- assertTrue (metadata .keys (). isEmpty ());
101+ assertTrue (metadata .isEmpty ());
98102 }
99103
100104 @ Test
@@ -103,7 +107,8 @@ public void putStringAddsValue() {
103107 .put ("key" , "value" )
104108 .build ();
105109
106- assertEquals ("value" , metadata .get ("key" ));
110+ MetadataKey <String > KEY = new MetadataKey <>("key" );
111+ assertEquals ("value" , metadata .get (KEY ));
107112 }
108113
109114 @ Test
@@ -112,7 +117,8 @@ public void putIntegerAddsValue() {
112117 .put ("count" , 42 )
113118 .build ();
114119
115- assertEquals (42 , metadata .get ("count" ));
120+ MetadataKey <Integer > COUNT = new MetadataKey <>("count" );
121+ assertEquals (Integer .valueOf (42 ), metadata .get (COUNT ));
116122 }
117123
118124 @ Test
@@ -121,7 +127,8 @@ public void putLongAddsValue() {
121127 .put ("timestamp" , 1234567890L )
122128 .build ();
123129
124- assertEquals (1234567890L , metadata .get ("timestamp" ));
130+ MetadataKey <Long > TIMESTAMP = new MetadataKey <>("timestamp" );
131+ assertEquals (Long .valueOf (1234567890L ), metadata .get (TIMESTAMP ));
125132 }
126133
127134 @ Test
@@ -130,7 +137,8 @@ public void putDoubleAddsValue() {
130137 .put ("rate" , 3.14 )
131138 .build ();
132139
133- assertEquals (3.14 , metadata .get ("rate" ));
140+ MetadataKey <Double > RATE = new MetadataKey <>("rate" );
141+ assertEquals (Double .valueOf (3.14 ), metadata .get (RATE ));
134142 }
135143
136144 @ Test
@@ -139,7 +147,8 @@ public void putBooleanTrueAddsValue() {
139147 .put ("enabled" , true )
140148 .build ();
141149
142- assertEquals (true , metadata .get ("enabled" ));
150+ MetadataKey <Boolean > ENABLED = new MetadataKey <>("enabled" );
151+ assertEquals (Boolean .TRUE , metadata .get (ENABLED ));
143152 }
144153
145154 @ Test
@@ -148,7 +157,8 @@ public void putBooleanFalseAddsValue() {
148157 .put ("disabled" , false )
149158 .build ();
150159
151- assertEquals (false , metadata .get ("disabled" ));
160+ MetadataKey <Boolean > DISABLED = new MetadataKey <>("disabled" );
161+ assertEquals (Boolean .FALSE , metadata .get (DISABLED ));
152162 }
153163
154164 @ Test
@@ -159,7 +169,7 @@ public void putListOfStringsAddsValue() {
159169 .put ("updatedFlags" , flags )
160170 .build ();
161171
162- assertEquals (flags , metadata .get ("updatedFlags" ));
172+ assertEquals (flags , metadata .get (SdkUpdateMetadataKeys . UPDATED_FLAGS ));
163173 }
164174
165175 @ Test
@@ -171,11 +181,11 @@ public void chainingMultiplePutsWorks() {
171181 .put ("list" , Arrays .asList ("a" , "b" ))
172182 .build ();
173183
174- assertEquals (4 , metadata .keys (). size ());
175- assertEquals ("text" , metadata .get ("string" ));
176- assertEquals (100 , metadata .get ("number" ));
177- assertEquals (true , metadata .get ("flag" ));
178- assertEquals (Arrays .asList ("a" , "b" ), metadata .get ("list" ));
184+ assertEquals (4 , metadata .size ());
185+ assertEquals ("text" , metadata .get (new MetadataKey < String >( "string" ) ));
186+ assertEquals (Integer . valueOf ( 100 ) , metadata .get (new MetadataKey < Integer >( "number" ) ));
187+ assertEquals (Boolean . TRUE , metadata .get (new MetadataKey < Boolean >( "flag" ) ));
188+ assertEquals (Arrays .asList ("a" , "b" ), metadata .get (new MetadataKey < List < String >>( "list" ) ));
179189 }
180190
181191 @ Test
@@ -185,7 +195,8 @@ public void overwritingKeyUsesLastValue() {
185195 .put ("key" , "second" )
186196 .build ();
187197
188- assertEquals ("second" , metadata .get ("key" ));
198+ MetadataKey <String > KEY = new MetadataKey <>("key" );
199+ assertEquals ("second" , metadata .get (KEY ));
189200 }
190201
191202 @ Test
@@ -196,6 +207,7 @@ public void buildReturnsNewInstanceEachTime() {
196207 EventMetadata metadata1 = builder .build ();
197208 EventMetadata metadata2 = builder .build ();
198209
199- assertEquals (metadata1 .get ("key" ), metadata2 .get ("key" ));
210+ MetadataKey <String > KEY = new MetadataKey <>("key" );
211+ assertEquals (metadata1 .get (KEY ), metadata2 .get (KEY ));
200212 }
201213}
0 commit comments