|
30 | 30 |
|
31 | 31 | static const uint16_t kLocalPort = 5541;
|
32 | 32 | static const uint16_t kTestVendorId = 0xFFF1u;
|
33 |
| -static const uint16_t kTestProductId1 = 0x8000u; |
34 |
| -static const uint16_t kTestProductId2 = 0x8001u; |
35 |
| -static const uint16_t kTestDiscriminator1 = 3840u; |
36 |
| -static const uint16_t kTestDiscriminator2 = 3839u; |
| 33 | +static const __auto_type kTestProductIds = @[ @(0x8000u) ]; |
| 34 | +static const __auto_type kTestDiscriminators = @[ @(3840u) ]; |
37 | 35 | static const uint16_t kDiscoverDeviceTimeoutInSeconds = 10;
|
38 |
| -static const uint16_t kExpectedDiscoveredDevicesCount = 2; |
| 36 | +static const uint16_t kExpectedDiscoveredDevicesCount = 1; |
39 | 37 |
|
40 | 38 | // Singleton controller we use.
|
41 | 39 | static MTRDeviceController * sController = nil;
|
@@ -96,8 +94,8 @@ - (void)controller:(MTRDeviceController *)controller didFindCommissionableDevice
|
96 | 94 |
|
97 | 95 | XCTAssertEqual(instanceName.length, 16); // The instance name is random, so just ensure the len is right.
|
98 | 96 | XCTAssertEqualObjects(vendorId, @(kTestVendorId));
|
99 |
| - XCTAssertTrue([productId isEqual:@(kTestProductId1)] || [productId isEqual:@(kTestProductId2)]); |
100 |
| - XCTAssertTrue([discriminator isEqual:@(kTestDiscriminator1)] || [discriminator isEqual:@(kTestDiscriminator2)]); |
| 97 | + XCTAssertTrue([kTestProductIds containsObject:productId]); |
| 98 | + XCTAssertTrue([kTestDiscriminators containsObject:discriminator]); |
101 | 99 | XCTAssertEqual(commissioningMode, YES);
|
102 | 100 |
|
103 | 101 | NSLog(@"Found Device (%@) with discriminator: %@ (vendor: %@, product: %@)", instanceName, discriminator, vendorId, productId);
|
|
0 commit comments