app_client_test.dart

import 'package:flutter_test/flutter_test.dart';
import 'package:mockito/mockito.dart';
import 'package:viam_sdk/protos/app/app.dart';
import 'package:viam_sdk/protos/app/packages.dart';
import 'package:viam_sdk/protos/common/common.dart';
import 'package:viam_sdk/src/app/app.dart';
import 'package:viam_sdk/src/gen/app/v1/app.pb.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 'package:viam_sdk/viam_sdk.dart';

import '../mocks/mock_response_future.dart';
import '../mocks/service_clients_mocks.mocks.dart';

void main() {
  late MockAppServiceClient serviceClient;
  late AppClient appClient;

  setUp(() {
    serviceClient = MockAppServiceClient();
    appClient = AppClient(serviceClient);
  });

  group('App RPC Client Tests', () {
    test('getUserIDByEmail', () async {
      when(serviceClient.getUserIDByEmail(any)).thenAnswer((_) => MockResponseFuture.value(GetUserIDByEmailResponse()..userId = 'userId'));
      final response = await appClient.getUserIdByEmail('email');
      expect(response, equals('userId'));
    });

    test('createOrganization', () async {
      final expected = Organization()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.createOrganization(any))
          .thenAnswer((_) => MockResponseFuture.value(CreateOrganizationResponse()..organization = expected));
      final response = await appClient.createOrganization('name');
      expect(response, equals(expected));
    });

    test('listOrganizations', () async {
      final expected = [
        Organization()
          ..id = 'id'
          ..name = 'name'
          ..createdOn = Timestamp.create()
      ];
      when(serviceClient.listOrganizations(any))
          .thenAnswer((_) => MockResponseFuture.value(ListOrganizationsResponse()..organizations.addAll(expected)));
      final response = await appClient.listOrganizations();
      expect(response, equals(expected));
    });

    test('getOrganizationsWithAccessToLocation', () async {
      final expected = [
        OrganizationIdentity()
          ..id = 'id'
          ..name = 'name'
      ];
      when(serviceClient.getOrganizationsWithAccessToLocation(any)).thenAnswer(
          (_) => MockResponseFuture.value(GetOrganizationsWithAccessToLocationResponse()..organizationIdentities.addAll(expected)));
      final response = await appClient.getOrganizationsWithAccessToLocation('locationId');
      expect(response, equals(expected));
    });

    test('listOrganizationsByUser', () async {
      final expected = [
        OrgDetails()
          ..orgId = 'id'
          ..orgName = 'name'
      ];
      when(serviceClient.listOrganizationsByUser(any))
          .thenAnswer((_) => MockResponseFuture.value(ListOrganizationsByUserResponse()..orgs.addAll(expected)));
      final response = await appClient.listOrganizationsByUser('userId');
      expect(response, equals(expected));
    });

    test('getOrganization', () async {
      final expected = Organization()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.getOrganization(any))
          .thenAnswer((_) => MockResponseFuture.value(GetOrganizationResponse()..organization = expected));
      final response = await appClient.getOrganization('id');
      expect(response, equals(expected));
    });

    test('getOrganizationNamespaceAvailability', () async {
      when(serviceClient.getOrganizationNamespaceAvailability(any))
          .thenAnswer((_) => MockResponseFuture.value(GetOrganizationNamespaceAvailabilityResponse()..available = true));
      final response = await appClient.getOrganizationNamespaceAvailability('publicNamespace');
      expect(response, equals(true));
    });

    test('updateOrganization', () async {
      final expected = Organization()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.updateOrganization(any))
          .thenAnswer((_) => MockResponseFuture.value(UpdateOrganizationResponse()..organization = expected));
      final response = await appClient.updateOrganization('organizationId');
      expect(response, equals(expected));
    });

    test('deleteOrganization', () async {
      when(serviceClient.deleteOrganization(any)).thenAnswer((_) => MockResponseFuture.value(DeleteOrganizationResponse()));
      await appClient.deleteOrganization('organizationId');
      verify(serviceClient.deleteOrganization(any)).called(1);
    });

    test('listOrganizationMembers', () async {
      final expected = [OrganizationMember()..userId = 'userId'];
      when(serviceClient.listOrganizationMembers(any))
          .thenAnswer((_) => MockResponseFuture.value(ListOrganizationMembersResponse(members: expected)));
      final response = await appClient.listOrganizationMembers('organizationId');
      expect(response.members, equals(expected));
    });

    test('createOrganizationInvite', () async {
      final expected = OrganizationInvite()
        ..organizationId = 'organizationId'
        ..email = 'email'
        ..createdOn = Timestamp.create();
      when(serviceClient.createOrganizationInvite(any))
          .thenAnswer((_) => MockResponseFuture.value(CreateOrganizationInviteResponse()..invite = expected));
      final response = await appClient.createOrganizationInvite('organizationId', 'email', []);
      expect(response, equals(expected));
    });

    test('updateOrganizationInviteAuthorizations', () async {
      final expected = OrganizationInvite()
        ..organizationId = 'organizationId'
        ..email = 'email'
        ..createdOn = Timestamp.create();
      when(serviceClient.updateOrganizationInviteAuthorizations(any))
          .thenAnswer((_) => MockResponseFuture.value(UpdateOrganizationInviteAuthorizationsResponse()..invite = expected));
      final response = await appClient.updateOrganizationInviteAuthorizations('organizationId', 'email', [], []);
      expect(response, equals(expected));
    });

    test('deleteOrganizationMember', () async {
      final expected = DeleteOrganizationMemberResponse();
      when(serviceClient.deleteOrganizationMember(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteOrganizationMember('organizationId', 'user id');
      verify(serviceClient.deleteOrganizationMember(any)).called(1);
    });

    test('deleteOrganizationInvite', () async {
      final expected = DeleteOrganizationInviteResponse();
      when(serviceClient.deleteOrganizationInvite(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteOrganizationInvite('organizationId', 'email');
      verify(serviceClient.deleteOrganizationInvite(any)).called(1);
    });

    test('resendOrganizationInvite', () async {
      final expected = OrganizationInvite()
        ..organizationId = 'organizationId'
        ..email = 'email'
        ..createdOn = Timestamp.create();
      when(serviceClient.resendOrganizationInvite(any))
          .thenAnswer((_) => MockResponseFuture.value(ResendOrganizationInviteResponse()..invite = expected));
      final response = await appClient.resendOrganizationInvite('organizationId', 'email');
      expect(response, equals(expected));
    });

    test('createLocation', () async {
      final expected = Location()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.createLocation(any)).thenAnswer((_) => MockResponseFuture.value(CreateLocationResponse()..location = expected));
      final response = await appClient.createLocation('organizationId', 'name');
      expect(response, equals(expected));
    });

    test('getLocation', () async {
      final expected = Location()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.getLocation(any)).thenAnswer((_) => MockResponseFuture.value(GetLocationResponse()..location = expected));
      final response = await appClient.getLocation('id');
      expect(response, equals(expected));
    });

    test('updateLocation', () async {
      final expected = Location()
        ..id = 'id'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      when(serviceClient.updateLocation(any)).thenAnswer((_) => MockResponseFuture.value(UpdateLocationResponse()..location = expected));
      final response = await appClient.updateLocation('locationId');
      expect(response, equals(expected));
    });

    test('deleteLocation', () async {
      final expected = DeleteLocationResponse();
      when(serviceClient.deleteLocation(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteLocation('locationId');
      verify(serviceClient.deleteLocation(any)).called(1);
    });

    test('listLocations', () async {
      final expected = [
        Location()
          ..id = 'id'
          ..name = 'name'
          ..createdOn = Timestamp.create()
      ];
      when(serviceClient.listLocations(any))
          .thenAnswer((_) => MockResponseFuture.value(ListLocationsResponse()..locations.addAll(expected)));
      final response = await appClient.listLocations('organizationId');
      expect(response, equals(expected));
    });

    test('shareLocation', () async {
      final expected = ShareLocationResponse();
      when(serviceClient.shareLocation(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.shareLocation('locationId', 'organizationId');
      verify(serviceClient.shareLocation(any)).called(1);
    });

    test('unshareLocation', () async {
      final expected = UnshareLocationResponse();
      when(serviceClient.unshareLocation(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.unshareLocation('locationId', 'organizationId');
      verify(serviceClient.unshareLocation(any)).called(1);
    });

    test('locationAuth', () async {
      final secret = SharedSecret()
        ..id = 'id'
        ..secret = 'secret'
        ..createdOn = Timestamp.create()
        ..state = SharedSecret_State.STATE_UNSPECIFIED;
      final expected = LocationAuth(secrets: [secret])
        ..secret = 'secret'
        ..locationId = 'locationId';
      when(serviceClient.locationAuth(any)).thenAnswer((_) => MockResponseFuture.value(LocationAuthResponse()..auth = expected));
      final response = await appClient.locationAuth('locationId');
      expect(response, equals(expected));
    });

    test('createLocationSecret', () async {
      final secret = SharedSecret()
        ..id = 'id'
        ..secret = 'secret'
        ..createdOn = Timestamp.create()
        ..state = SharedSecret_State.STATE_UNSPECIFIED;
      final expected = LocationAuth(secrets: [secret])
        ..secret = 'secret'
        ..locationId = 'locationId';
      when(serviceClient.createLocationSecret(any))
          .thenAnswer((_) => MockResponseFuture.value(CreateLocationSecretResponse()..auth = expected));
      final response = await appClient.createLocationSecret('locationId');
      expect(response, equals(expected));
    });

    test('deleteLocationSecret', () async {
      final expected = DeleteLocationSecretResponse();
      when(serviceClient.deleteLocationSecret(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteLocationSecret('locationId', 'secretId');
      verify(serviceClient.deleteLocationSecret(any)).called(1);
    });

    test('getRobot', () async {
      final expected = Robot()
        ..id = 'id'
        ..name = 'name';
      when(serviceClient.getRobot(any)).thenAnswer((_) => MockResponseFuture.value(GetRobotResponse()..robot = expected));
      final response = await appClient.getRobot('robot');
      expect(response, equals(expected));
    });

    test('getRoverRentalRobots', () async {
      final expected = [
        RoverRentalRobot()
          ..robotId = 'robotId'
          ..locationId = 'locationId'
          ..robotName = 'robotName'
          ..robotMainPartId = 'robotMainPartId'
      ];
      when(serviceClient.getRoverRentalRobots(any))
          .thenAnswer((_) => MockResponseFuture.value(GetRoverRentalRobotsResponse()..robots.addAll(expected)));
      final response = await appClient.getRoverRentalRobots('orgId');
      expect(response, equals(expected));
    });

    test('listRobotParts', () async {
      final expected = [
        RobotPart()
          ..id = 'id'
          ..name = 'name'
      ];
      when(serviceClient.getRobotParts(any)).thenAnswer((_) => MockResponseFuture.value(GetRobotPartsResponse()..parts.addAll(expected)));
      final response = await appClient.listRobotParts('robotId');
      expect(response, equals(expected));
    });

    test('getRobotPart', () async {
      final expected = RobotPart()
        ..id = 'id'
        ..name = 'name';
      when(serviceClient.getRobotPart(any)).thenAnswer((_) => MockResponseFuture.value(GetRobotPartResponse()..part = expected));
      final response = await appClient.getRobotPart('robot part');
      expect(response, equals(expected));
    });

    test('getLogs', () async {
      final log = LogEntry()..message = 'My log entry';
      final expected = RobotPartLogPage()
        ..logs.add(log)
        ..nextPageToken = 'nextPageToken';
      when(serviceClient.getRobotPartLogs(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.getLogs('robotPart');
      expect(response, equals(expected));
    });

    test('tailLogs', () async {
      final expected = LogEntry()..message = 'My log entry';
      final response = TailRobotPartLogsResponse()..logs.add(expected);
      when(serviceClient.tailRobotPartLogs(any)).thenAnswer((_) => MockResponseStream.list([response]));
      final stream = appClient.tailLogs('robotPart');
      expect(
          stream,
          emitsInOrder([
            emits([expected]),
            emitsDone
          ]));
    });

    test('getRobotPartHistory', () async {
      final expected = [
        RobotPartHistoryEntry()
          ..part = 'part'
          ..robot = 'robot'
          ..when = Timestamp.create()
          ..old = RobotPart()
      ];
      when(serviceClient.getRobotPartHistory(any))
          .thenAnswer((_) => MockResponseFuture.value(GetRobotPartHistoryResponse(history: expected)));
      final response = await appClient.getRobotPartHistory('id');
      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 = {'robot': 'config'};
      final response = await appClient.updateRobotPart('robot part', 'name2', rc);
      expect(response, equals(expected));
    });

    test('newRobotPart', () async {
      const expected = 'partId';
      when(serviceClient.newRobotPart(any)).thenAnswer((_) => MockResponseFuture.value(NewRobotPartResponse()..partId = expected));
      final response = await appClient.newRobotPart('robotId', 'partName');
      expect(response, equals(expected));
    });

    test('deleteRobotPart', () async {
      final expected = DeleteRobotPartResponse();
      when(serviceClient.deleteRobotPart(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteRobotPart('partId');
      verify(serviceClient.deleteRobotPart(any)).called(1);
    });

    test('getRobotAPIKeys', () async {
      final apiKey = APIKey()
        ..id = 'id'
        ..key = 'key'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      final expected = [APIKeyWithAuthorizations(authorizations: [])..apiKey = apiKey];
      when(serviceClient.getRobotAPIKeys(any)).thenAnswer((_) => MockResponseFuture.value(GetRobotAPIKeysResponse(apiKeys: expected)));
      final response = await appClient.getRobotApiKeys('robotId');
      expect(response, equals(expected));
    });

    test('markPartAsMain', () async {
      final expected = MarkPartAsMainResponse();
      when(serviceClient.markPartAsMain(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.markPartAsMain('partId');
      verify(serviceClient.markPartAsMain(any)).called(1);
    });

    test('markPartForRestart', () async {
      final expected = MarkPartForRestartResponse();
      when(serviceClient.markPartForRestart(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.markPartForRestart('partId');
      verify(serviceClient.markPartForRestart(any)).called(1);
    });

    test('createRobotPartSecret', () async {
      final expected = RobotPart()
        ..id = 'id'
        ..name = 'name';
      when(serviceClient.createRobotPartSecret(any))
          .thenAnswer((_) => MockResponseFuture.value(CreateRobotPartSecretResponse()..part = expected));
      final response = await appClient.createRobotPartSecret('partId');
      expect(response, equals(expected));
    });

    test('deleteRobotPartSecret', () async {
      final expected = DeleteRobotPartSecretResponse();
      when(serviceClient.deleteRobotPartSecret(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteRobotPartSecret('partId', 'secretId');
      verify(serviceClient.deleteRobotPartSecret(any)).called(1);
    });

    test('listRobots', () async {
      final expected = [
        Robot()
          ..id = 'id'
          ..name = 'name'
      ];
      when(serviceClient.listRobots(any)).thenAnswer((_) => MockResponseFuture.value(ListRobotsResponse()..robots.addAll(expected)));
      final response = await appClient.listRobots('locationId');
      expect(response, equals(expected));
    });

    test('newMachine', () async {
      final expected = NewRobotResponse();
      when(serviceClient.newRobot(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.newMachine('test', 'fake-id');
      verify(serviceClient.newRobot(any)).called(1);
    });

    test('updateRobot', () async {
      final expected = Robot()
        ..id = 'id'
        ..name = 'name2';
      when(serviceClient.updateRobot(any)).thenAnswer((_) => MockResponseFuture.value(UpdateRobotResponse()..robot = expected));
      final response = await appClient.updateRobot('id', 'name2', 'location');
      expect(response, equals(expected));
    });

    test('deleteRobot', () async {
      final expected = DeleteRobotResponse();
      when(serviceClient.deleteRobot(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteRobot('id');
      verify(serviceClient.deleteRobot(any)).called(1);
    });

    test('listFragments', () async {
      final expected = [
        Fragment()
          ..id = 'id'
          ..name = 'name'
          ..fragment = Struct()
          ..organizationOwner = 'organizationOwner'
          ..public = true
          ..createdOn = Timestamp.create()
          ..organizationName = 'organizationName'
          ..robotPartCount = 3
          ..organizationCount = 2
          ..onlyUsedByOwner = false
      ];
      when(serviceClient.listFragments(any))
          .thenAnswer((_) => MockResponseFuture.value(ListFragmentsResponse()..fragments.addAll(expected)));
      final response = await appClient.listFragments('organizationId', true);
      expect(response, equals(expected));
    });

    test('getFragment', () async {
      final expected = Fragment()
        ..id = 'id'
        ..name = 'name'
        ..fragment = Struct()
        ..organizationOwner = 'organizationOwner'
        ..public = true
        ..createdOn = Timestamp.create()
        ..organizationName = 'organizationName'
        ..robotPartCount = 3
        ..organizationCount = 2
        ..onlyUsedByOwner = false;
      when(serviceClient.getFragment(any)).thenAnswer((_) => MockResponseFuture.value(GetFragmentResponse()..fragment = expected));
      final response = await appClient.getFragment('id');
      expect(response, equals(expected));
    });

    test('createFragment', () async {
      final expected = Fragment()
        ..id = 'id'
        ..name = 'name'
        ..fragment = Struct()
        ..organizationOwner = 'organizationOwner'
        ..public = true
        ..createdOn = Timestamp.create()
        ..organizationName = 'organizationName'
        ..robotPartCount = 3
        ..organizationCount = 2
        ..onlyUsedByOwner = false;
      when(serviceClient.createFragment(any)).thenAnswer((_) => MockResponseFuture.value(CreateFragmentResponse()..fragment = expected));
      final response = await appClient.createFragment('name', {'config': 1}, 'organizationId');
      expect(response, equals(expected));
    });

    test('updateFragment', () async {
      final expected = Fragment()
        ..id = 'id'
        ..name = 'name'
        ..fragment = Struct()
        ..organizationOwner = 'organizationOwner'
        ..public = true
        ..createdOn = Timestamp.create()
        ..organizationName = 'organizationName'
        ..robotPartCount = 3
        ..organizationCount = 2
        ..onlyUsedByOwner = false;
      when(serviceClient.updateFragment(any)).thenAnswer((_) => MockResponseFuture.value(UpdateFragmentResponse()..fragment = expected));
      final response = await appClient.updateFragment('id', 'name', {'config': 2}, public: false);
      expect(response, equals(expected));
    });

    test('deleteFragment', () async {
      final expected = DeleteFragmentResponse();
      when(serviceClient.deleteFragment(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteFragment('id');
      verify(serviceClient.deleteFragment(any)).called(1);
    });

    test('addRole', () async {
      final expected = AddRoleResponse();
      when(serviceClient.addRole(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final authorization = ViamAuthorization(
          authorizationId: AuthorizationId.robotOwner,
          resourceType: ResourceType.robot,
          resourceId: 'resourceId',
          organizationId: 'organizationId',
          identityType: IdentityType.user)
        ..authorizationType = 'authorizationType'
        ..identityId = 'identityId';
      await appClient.addRole(authorization);
      verify(serviceClient.addRole(any)).called(1);
    });

    test('removeRole', () async {
      final authorization = ViamAuthorization(
          authorizationId: AuthorizationId.robotOwner,
          resourceType: ResourceType.robot,
          resourceId: 'resourceId',
          organizationId: 'organizationId',
          identityType: IdentityType.user)
        ..authorizationType = 'authorizationType'
        ..identityId = 'identityId';
      final expected = RemoveRoleResponse();
      when(serviceClient.removeRole(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.removeRole(authorization);
      verify(serviceClient.removeRole(any)).called(1);
    });

    test('changeRole', () async {
      final expected = ChangeRoleResponse();
      when(serviceClient.changeRole(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final oldAuthorization = ViamAuthorization(
          authorizationId: AuthorizationId.robotOwner,
          resourceType: ResourceType.robot,
          resourceId: 'resourceId',
          organizationId: 'organizationId',
          identityType: IdentityType.user)
        ..authorizationType = 'authorizationType'
        ..identityId = 'identityId';
      final newAuthorization = ViamAuthorization(
          authorizationId: AuthorizationId.robotOwner,
          resourceType: ResourceType.robot,
          resourceId: 'resourceId2',
          organizationId: 'organizationId2',
          identityType: IdentityType.user)
        ..authorizationType = 'authorizationType2'
        ..identityId = 'identityId2';
      await appClient.changeRole(oldAuthorization, newAuthorization);
      verify(serviceClient.changeRole(any)).called(1);
    });

    test('listAuthorizations', () async {
      final expected = [
        Authorization()
          ..authorizationType = 'authorizationType'
          ..authorizationId = 'robotOwner'
          ..resourceType = 'robot'
          ..resourceId = 'resourceId'
          ..identityId = 'identityId'
          ..organizationId = 'organizationId'
          ..identityType = 'user'
      ];
      when(serviceClient.listAuthorizations(any))
          .thenAnswer((_) => MockResponseFuture.value(ListAuthorizationsResponse()..authorizations.addAll(expected)));
      final response = await appClient.listAuthorizations('organizationId');
      expect(response, equals(expected));
    });

    test('checkPermissions', () async {
      final expected = [Permission.controlRobot];
      final authorizedPermissions = [
        AuthorizedPermissions()
          ..resourceType = 'robot'
          ..resourceId = 'resourceId'
          ..permissions.addAll(['control_robot'])
      ];
      when(serviceClient.checkPermissions(any))
          .thenAnswer((_) => MockResponseFuture.value(CheckPermissionsResponse()..authorizedPermissions.addAll(authorizedPermissions)));
      final response = await appClient.checkPermissions(ResourceType.robot, 'resourceId', [Permission.controlRobot]);
      expect(response, equals(expected));
    });

    test('getRegistryItem', () async {
      final expected = RegistryItem()
        ..organizationId = 'organizationId'
        ..name = 'name'
        ..type = PackageType.PACKAGE_TYPE_UNSPECIFIED;
      when(serviceClient.getRegistryItem(any)).thenAnswer((_) => MockResponseFuture.value(GetRegistryItemResponse()..item = expected));
      final response = await appClient.getRegistryItem('itemId');
      expect(response, equals(expected));
    });

    test('createRegistryItem', () async {
      final expected = CreateRegistryItemResponse();
      when(serviceClient.createRegistryItem(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.createRegistryItem('organizationId', 'name', PackageType.PACKAGE_TYPE_UNSPECIFIED);
      verify(serviceClient.createRegistryItem(any)).called(1);
    });

    test('updateRegistryItem', () async {
      final expected = UpdateRegistryItemResponse();
      when(serviceClient.updateRegistryItem(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.updateRegistryItem('itemId', PackageType.PACKAGE_TYPE_UNSPECIFIED, 'description', Visibility.VISIBILITY_UNSPECIFIED);
      verify(serviceClient.updateRegistryItem(any)).called(1);
    });

    test('listRegistryItems', () async {
      final expected = [
        RegistryItem()
          ..organizationId = 'organizationId'
          ..name = 'name'
          ..type = PackageType.PACKAGE_TYPE_UNSPECIFIED
      ];
      when(serviceClient.listRegistryItems(any))
          .thenAnswer((_) => MockResponseFuture.value(ListRegistryItemsResponse()..items.addAll(expected)));
      final response = await appClient.listRegistryItems([PackageType.PACKAGE_TYPE_UNSPECIFIED], [Visibility.VISIBILITY_UNSPECIFIED],
          ['platforms'], [RegistryItemStatus.REGISTRY_ITEM_STATUS_UNSPECIFIED]);
      expect(response, equals(expected));
    });

    test('deleteRegistryItem', () async {
      final expected = DeleteRegistryItemResponse();
      when(serviceClient.deleteRegistryItem(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteRegistryItem('itemId');
      verify(serviceClient.deleteRegistryItem(any)).called(1);
    });

    test('createModule', () async {
      final expected = CreateModuleResponse()
        ..moduleId = 'moduleId'
        ..url = 'url';
      when(serviceClient.createModule(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.createModule('organizationId', 'name');
      expect(response, equals(expected));
    });

    test('updateModule', () async {
      const expected = 'url';
      when(serviceClient.updateModule(any)).thenAnswer((_) => MockResponseFuture.value(UpdateModuleResponse()..url = expected));
      final response =
          await appClient.updateModule('moduleId', Visibility.VISIBILITY_UNSPECIFIED, 'url', 'description', [Model()], 'entrypoint');
      expect(response, equals(expected));
    });

    test('uploadModuleFile', () async {
      const expected = 'url';
      when(serviceClient.uploadModuleFile(any)).thenAnswer((_) => MockResponseFuture.value(UploadModuleFileResponse()..url = expected));
      final response = await appClient.uploadModuleFile(ModuleFileInfo(), [1, 2]);
      expect(response, equals(expected));
    });

    test('getModule', () async {
      final expected = Module()
        ..moduleId = 'moduleId'
        ..name = 'name';
      when(serviceClient.getModule(any)).thenAnswer((_) => MockResponseFuture.value(GetModuleResponse()..module = expected));
      final response = await appClient.getModule('moduleId');
      expect(response, equals(expected));
    });

    test('listModules', () async {
      final expected = [
        Module()
          ..moduleId = 'moduleId'
          ..name = 'name'
      ];
      when(serviceClient.listModules(any)).thenAnswer((_) => MockResponseFuture.value(ListModulesResponse()..modules.addAll(expected)));
      final response = await appClient.listModules('moduleId');
      expect(response, equals(expected));
    });

    test('createKey', () async {
      final expected = CreateKeyResponse()
        ..key = 'key'
        ..id = 'id';
      when(serviceClient.createKey(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final authorizations = [
        ViamAuthorization(
            authorizationId: AuthorizationId.robotOwner,
            resourceType: ResourceType.robot,
            resourceId: 'resourceId',
            organizationId: 'organizationId',
            identityType: IdentityType.user)
          ..authorizationType = 'authorizationType'
          ..identityId = 'identityId'
      ];
      final response = await appClient.createKey(authorizations, 'name');
      expect(response, equals(expected));
    });

    test('deleteKey', () async {
      final expected = DeleteKeyResponse();
      when(serviceClient.deleteKey(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      await appClient.deleteKey('id');
      verify(serviceClient.deleteKey(any)).called(1);
    });

    test('listKeys', () async {
      final apiKey = APIKey()
        ..id = 'id'
        ..key = 'key'
        ..name = 'name'
        ..createdOn = Timestamp.create();
      final expected = [APIKeyWithAuthorizations(authorizations: [])..apiKey = apiKey];
      when(serviceClient.listKeys(any)).thenAnswer((_) => MockResponseFuture.value(ListKeysResponse(apiKeys: expected)));
      final response = await appClient.listKeys('orgId');
      expect(response, equals(expected));
    });

    test('rotateKey', () async {
      final expected = RotateKeyResponse()
        ..id = 'id'
        ..key = 'key';
      when(serviceClient.rotateKey(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.rotateKey('id');
      expect(response, equals(expected));
    });

    test('createKeyFromExistingKeyAuthorizations', () async {
      final expected = CreateKeyFromExistingKeyAuthorizationsResponse()
        ..id = 'id'
        ..key = 'key';
      when(serviceClient.createKeyFromExistingKeyAuthorizations(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.createKeyFromExistingKeyAuthorizations('id');
      expect(response, equals(expected));
    });

    test('getOrganizationMetadata', () async {
      final expected = GetOrganizationMetadataResponse()
        ..metadata = (Struct()..fields['key'] = (Value()..stringValue = 'value'));
      when(serviceClient.getOrganizationMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected)); 
      final response = await appClient.getOrganizationMetadata('orgId');
      expect(response, equals(expected));
    });

    test('updateOrganizationMetadata', () async {
      final expected = UpdateOrganizationMetadataResponse()
        ..metadata = (Struct()..fields['key'] = (Value()..stringValue = 'newValue'));
      when(serviceClient.updateOrganizationMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.updateOrganizationMetadata('orgId', {'key': 'newValue'});
      expect(response, equals(expected));
    });

    test('getLocationMetadata', () async {
      final expected = GetLocationMetadataResponse()
        ..metadata = (Struct()..fields['locationKey'] = (Value()..stringValue = 'locationValue'));
      when(serviceClient.getLocationMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.getLocationMetadata('locationId');
      expect(response, equals(expected));
    });

    test('updateLocationMetadata', () async {
      final expected = UpdateLocationMetadataResponse()
        ..metadata = (Struct()..fields['locationKey'] = (Value()..stringValue = 'newLocationValue'));
      when(serviceClient.updateLocationMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.updateLocationMetadata('locationId', {'locationKey': 'newLocationValue'});
      expect(response, equals(expected));
    });

    test('getMachineMetadata', () async {
      final expected = GetRobotMetadataResponse()
        ..metadata = (Struct()..fields['robotKey'] = (Value()..stringValue = 'robotValue'));
      when(serviceClient.getMachineMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.getMachineMetadata('robotId');
      expect(response, equals(expected));
    });

    test('updateMachineMetadata', () async {
      final expected = UpdateRobotMetadataResponse()
        ..metadata = (Struct()..fields['robotKey'] = (Value()..stringValue = 'newRobotValue'));
      when(serviceClient.updateMachineMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.updateMachineMetadata('robotId', {'robotKey': 'newRobotValue'});
      expect(response, equals(expected));
    });

    test('getMachinePartMetadata', () async {
      final expected = GetRobotPartMetadataResponse()
        ..metadata = (Struct()..fields['partKey'] = (Value()..stringValue = 'partValue'));
      when(serviceClient.getMachinePartMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.getMachinePartMetadata('partId');
      expect(response, equals(expected));
    });

    test('updateMachinePartMetadata', () async {
      final expected = UpdateRobotPartMetadataResponse()
        ..metadata = (Struct()..fields['partKey'] = (Value()..stringValue = 'newPartValue'));
      when(serviceClient.updateMachinePartMetadata(any)).thenAnswer((_) => MockResponseFuture.value(expected));
      final response = await appClient.updateMachinePartMetadata('partId', {'partKey': 'newPartValue'});
      expect(response, equals(expected));
    });
  });
}