def testWithClientSet(self): mock_client = _MockClient(client_id='foo') with mock.patch.object(magics_impl._state, 'client', mock_client): client_id = magics_impl.grr_id_impl() self.assertEqual(client_id, 'foo')
def grr_id(line: Text) -> Text: """Returns ID of the selected client. Args: line: A string representing arguments passed to the magic command. Returns: String representing ID of a client. Raises: NoClientSelectedError: Client is not selected to perform this operation. """ del line # Unused. return magics_impl.grr_id_impl()
def testNoClientSelected(self): with self.assertRaises(magics_impl.NoClientSelectedError): magics_impl.grr_id_impl()