RSDK-6463: Add updateRobotPart (#174)

* add updateRobotPart
viamrobotics · Jan 26, 2024 · 4665b24 · 4665b24
1 parent f9276a6
commit 4665b24
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 1 deletion.
diff --git a/lib/src/app/app.dart b/lib/src/app/app.dart
@@ -1,5 +1,7 @@
 import 'dart:async';
 
+import 'package:viam_sdk/src/gen/google/protobuf/struct.pb.dart';
+
 import '../gen/app/v1/app.pbgrpc.dart';
 import 'permissions.dart';
 
@@ -69,6 +71,16 @@ class AppClient {
     return response.part;
   }
 
+  /// Update a specific [RobotPart] by ID
+  Future<RobotPart> updateRobotPart(String partId, String name, Struct robotConfig) async {
+    final updateRobotPartRequest = UpdateRobotPartRequest()
+      ..id = partId
+      ..name = name
+      ..robotConfig = robotConfig;
+    final response = await _client.updateRobotPart(updateRobotPartRequest);
+    return response.part;
+  }
+
   /// Get a page of [LogEntry] for a specific [RobotPart]. Logs are sorted by descending time (newest first)
   Future<RobotPartLogPage> getLogs(RobotPart part, {bool errorsOnly = false, String pageToken = ''}) async {
     final request = GetRobotPartLogsRequest()

diff --git a/lib/src/utils.dart b/lib/src/utils.dart
@@ -18,7 +18,7 @@ extension NullableStringUtils on String? {
 extension ValueUtils on Value {
   dynamic toPrimitive() {
     if (hasBoolValue()) return boolValue;
-    if (hasListValue()) return listValue.values.map((e) => e.toPrimitive());
+    if (hasListValue()) return listValue.values.map((e) => e.toPrimitive()).toList();
     if (hasNullValue()) return null;
     if (hasNumberValue()) return numberValue;
     if (hasStringValue()) return stringValue;

diff --git a/test/unit_test/app/app_client_test.dart b/test/unit_test/app/app_client_test.dart
@@ -3,6 +3,7 @@ import 'package:mockito/mockito.dart';
 import 'package:viam_sdk/protos/app/app.dart';
 import 'package:viam_sdk/src/app/app.dart';
 import 'package:viam_sdk/src/gen/app/v1/app.pbgrpc.dart';
+import 'package:viam_sdk/src/gen/google/protobuf/struct.pb.dart';
 import 'package:viam_sdk/src/gen/google/protobuf/timestamp.pb.dart';
 
 import '../mocks/mock_response_future.dart';
@@ -105,6 +106,16 @@ void main() {
       expect(response, equals(expected));
     });
 
+    test('updateRobotPart', () async {
+      final expected = RobotPart()
+        ..id = 'id'
+        ..name = 'name2';
+      when(serviceClient.updateRobotPart(any)).thenAnswer((_) => MockResponseFuture.value(UpdateRobotPartResponse()..part = expected));
+      final rc = Struct();
+      final response = await appClient.updateRobotPart('robot part', 'name2', rc);
+      expect(response, equals(expected));
+    });
+
     test('listOrganizationMembers', () async {
       final expected = [OrganizationMember()..userId = 'userId'];
       when(serviceClient.listOrganizationMembers(any))