Пример #1
0
    def testGetIamPolicy(self):
        policy = self.msgs.GoogleIamV1Policy(etag=b'abcd')
        get_iam_policy = self.StartObjectPatch(self.models_client,
                                               'GetIamPolicy',
                                               return_value=policy,
                                               autospec=True)

        response = models_util.GetIamPolicy(self.models_client, 'myModel')

        self.assertEqual(response, policy)
        get_iam_policy.assert_called_once_with(self.model_ref)
def _Run(args):
    with endpoint_util.MlEndpointOverrides(region=args.region):
        return models_util.GetIamPolicy(models.ModelsClient(), args.model)
Пример #3
0
 def Run(self, args):
   return models_util.GetIamPolicy(models.ModelsClient(), args.model)