From 7ec00d344d3d9306d6f769be9ab688128881ee71 Mon Sep 17 00:00:00 2001
From: Clint Purser <clint.purser@viam.com>
Date: Wed, 31 Jan 2024 15:03:53 -0700
Subject: [PATCH] RSDK-5953 add get fragment wrapper (#178)

---
 lib/src/app/app.dart                    | 7 +++++++
 test/unit_test/app/app_client_test.dart | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/lib/src/app/app.dart b/lib/src/app/app.dart
index 2bcc55c28b..0be21191ae 100644
--- 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
index 67b42f586f..a1e7d4d99a 100644
--- 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);