|
| 1 | +import 'package:flutter_test/flutter_test.dart'; |
| 2 | +import 'package:viam_sdk/src/components/sensor/sensor.dart'; |
| 3 | +import 'package:viam_sdk/src/exceptions.dart'; |
| 4 | +import 'package:viam_sdk/src/resource/manager.dart'; |
| 5 | + |
| 6 | +import '../components/sensor_test.dart'; |
| 7 | + |
| 8 | +void main() { |
| 9 | + group('ResourceManager Tests', () { |
| 10 | + group('getResource', () { |
| 11 | + test('Local', () { |
| 12 | + final localName = 'local-sensor'; |
| 13 | + final localRN = Sensor.getResourceName(localName); |
| 14 | + final localResource = FakeSensor(localName); |
| 15 | + final manager = ResourceManager(); |
| 16 | + manager.register(localRN, localResource); |
| 17 | + |
| 18 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(localName)), localResource); |
| 19 | + }); |
| 20 | + |
| 21 | + test('Remote', () { |
| 22 | + final remoteName = 'remote-sensor'; |
| 23 | + final remotePath = 'remote2:remote1'; |
| 24 | + final fullRemoteName = '$remotePath:$remoteName'; |
| 25 | + final remoteRN = Sensor.subtype.getResourceName(fullRemoteName); |
| 26 | + final remoteResource = FakeSensor(fullRemoteName); |
| 27 | + final manager = ResourceManager(); |
| 28 | + manager.register(remoteRN, remoteResource); |
| 29 | + |
| 30 | + // Works with full name |
| 31 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(fullRemoteName)), remoteResource); |
| 32 | + |
| 33 | + // Works with short name -- no collisions |
| 34 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(remoteName)), remoteResource); |
| 35 | + }); |
| 36 | + |
| 37 | + test('Local and Remote - Same Names', () { |
| 38 | + final manager = ResourceManager(); |
| 39 | + |
| 40 | + final localName = 'my-sensor'; |
| 41 | + final localRN = Sensor.getResourceName(localName); |
| 42 | + final localResource = FakeSensor(localName); |
| 43 | + |
| 44 | + final remoteName = 'my-sensor'; |
| 45 | + final remotePath = 'remote2:remote1'; |
| 46 | + final fullRemoteName = '$remotePath:$remoteName'; |
| 47 | + final remoteRN = Sensor.subtype.getResourceName(fullRemoteName); |
| 48 | + final remoteResource = FakeSensor(fullRemoteName); |
| 49 | + |
| 50 | + manager.register(localRN, localResource); |
| 51 | + manager.register(remoteRN, remoteResource); |
| 52 | + |
| 53 | + // When using fully qualified names, it should return appropriately |
| 54 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(localName)), localResource); |
| 55 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(fullRemoteName)), remoteResource); |
| 56 | + |
| 57 | + // When using just `my-sensor`, it should always return the local |
| 58 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(localName)), localResource); |
| 59 | + expect(manager.getResource<Sensor>(Sensor.getResourceName(remoteName)), localResource); |
| 60 | + }); |
| 61 | + |
| 62 | + test('Multiple Remotes - Same Names', () { |
| 63 | + final remoteName = 'my-sensor'; |
| 64 | + |
| 65 | + final remotePath1 = 'remote2:remote1'; |
| 66 | + final fullRemoteName1 = '$remotePath1:$remoteName'; |
| 67 | + final remoteRN1 = Sensor.subtype.getResourceName(fullRemoteName1); |
| 68 | + final remoteResource1 = FakeSensor(fullRemoteName1); |
| 69 | + |
| 70 | + final remotePath2 = 'remote4:remote3'; |
| 71 | + final fullRemoteName2 = '$remotePath2:$remoteName'; |
| 72 | + final remoteRN2 = Sensor.subtype.getResourceName(fullRemoteName2); |
| 73 | + final remoteResource2 = FakeSensor(fullRemoteName2); |
| 74 | + |
| 75 | + final manager = ResourceManager(); |
| 76 | + manager.register(remoteRN1, remoteResource1); |
| 77 | + manager.register(remoteRN2, remoteResource2); |
| 78 | + |
| 79 | + // Error when using short name only |
| 80 | + expect(() => manager.getResource<Sensor>(Sensor.getResourceName(remoteName)), |
| 81 | + throwsA(isA<MultipleRemoteResourcesSameNameException>())); |
| 82 | + }); |
| 83 | + }); |
| 84 | + }); |
| 85 | +} |
0 commit comments