|
30 | 30 | import org.junit.jupiter.params.provider.MethodSource;
|
31 | 31 |
|
32 | 32 | import java.time.Duration;
|
| 33 | +import java.util.Arrays; |
| 34 | +import java.util.List; |
33 | 35 |
|
34 | 36 | import static org.junit.jupiter.api.Assertions.assertEquals;
|
35 | 37 | import static org.junit.jupiter.api.Assertions.assertNotNull;
|
| 38 | +import static org.junit.jupiter.api.Assertions.assertTrue; |
36 | 39 | import static org.junit.jupiter.api.Assertions.fail;
|
37 | 40 |
|
38 | 41 | public class PhoneNumbersClientIntegrationTest extends PhoneNumbersIntegrationTestBase {
|
@@ -181,8 +184,11 @@ public void getTollFreeAreaCodes(HttpClient httpClient) {
|
181 | 184 | PagedIterable<PhoneNumberAreaCode> areaCodesResult = this
|
182 | 185 | .getClientWithConnectionString(httpClient, "listAvailableTollFreeAreaCodes")
|
183 | 186 | .listAvailableTollFreeAreaCodes("US");
|
184 |
| - PhoneNumberAreaCode areaCodes = areaCodesResult.iterator().next(); |
185 |
| - assertNotNull(areaCodes); |
| 187 | + List<String> expectedAreaCodes = Arrays.asList("888", "877", "866", "855", "844", "800", "833", "88"); |
| 188 | + for (PhoneNumberAreaCode areaCode : areaCodesResult) { |
| 189 | + assertTrue(expectedAreaCodes.contains(areaCode.getAreaCode())); |
| 190 | + } |
| 191 | + assertNotNull(areaCodesResult); |
186 | 192 | }
|
187 | 193 |
|
188 | 194 | @ParameterizedTest
|
@@ -246,8 +252,11 @@ public void getTollFreeAreaCodesWithAAD(HttpClient httpClient) {
|
246 | 252 | PagedIterable<PhoneNumberAreaCode> areaCodesResult = this
|
247 | 253 | .getClientWithManagedIdentity(httpClient, "listAvailableTollFreeAreaCodes")
|
248 | 254 | .listAvailableTollFreeAreaCodes("US");
|
249 |
| - PhoneNumberAreaCode areaCodes = areaCodesResult.iterator().next(); |
250 |
| - assertNotNull(areaCodes); |
| 255 | + List<String> expectedAreaCodes = Arrays.asList("888", "877", "866", "855", "844", "800", "833", "88"); |
| 256 | + for (PhoneNumberAreaCode areaCode : areaCodesResult) { |
| 257 | + assertTrue(expectedAreaCodes.contains(areaCode.getAreaCode())); |
| 258 | + } |
| 259 | + assertNotNull(areaCodesResult); |
251 | 260 | }
|
252 | 261 |
|
253 | 262 | @ParameterizedTest
|
|
0 commit comments