def testKernelRelease(self): mock_client = _MockClient(kernel='foobar') with mock.patch.object(magics_impl._state, 'client', mock_client): kernel_release = magics_impl.grr_uname_impl(kernel_release=True) self.assertEqual(kernel_release, 'foobar')
def testMachine(self): mock_client = _MockClient(arch='foobar') with mock.patch.object(magics_impl._state, 'client', mock_client): machine = magics_impl.grr_uname_impl(machine=True) self.assertEqual(machine, 'foobar')
def grr_uname(line: Text) -> Text: """Returns certain system infornamtion. Args: line: A string representing arguments passed to the magic command. Returns: String representing some system information. Raises: NoClientSelectedError: Client is not selected to perform this operation. """ args = grr_uname.parser.parse_args(shlex.split(line)) return magics_impl.grr_uname_impl(args.machine, args.kernel_release)
def testNoOptionsProvided(self): mock_client = _MockClient() with mock.patch.object(magics_impl._state, 'client', mock_client): with self.assertRaises(ValueError): magics_impl.grr_uname_impl()
def testNoClientSelected(self): with self.assertRaises(magics_impl.NoClientSelectedError): magics_impl.grr_uname_impl()