RSDK-5953 add get fragment wrapper (viamrobotics#178)

clintpurser · Jan 31, 2024 · 7ec00d3 · 7ec00d3
1 parent d926137
commit 7ec00d3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lib/src/app/app.dart b/lib/src/app/app.dart
@@ -173,4 +173,11 @@ class AppClient {
     final response = await _client.newRobot(request);
     return response.id;
   }
+
+  /// Get a specific [Fragment] by id.
+  Future<Fragment> getFragment(String id) async {
+    final request = GetFragmentRequest()..id = id;
+    final response = await _client.getFragment(request);
+    return response.fragment;
+  }
 }
diff --git a/test/unit_test/app/app_client_test.dart b/test/unit_test/app/app_client_test.dart
@@ -167,6 +167,13 @@ void main() {
       verify(serviceClient.newRobot(any)).called(1);
     });
 
+    test('getFragment', () async {
+      final expected = GetFragmentResponse();
+      when(serviceClient.getFragment(any)).thenAnswer((_) => MockResponseFuture.value(expected));
+      await appClient.getFragment('fake-id');
+      verify(serviceClient.getFragment(any)).called(1);
+    });
+
     test('tailLogs', () async {
       final expected = LogEntry()..message = 'My log entry';
       final response = TailRobotPartLogsResponse()..logs.add(expected);