def KillFleetspeak(grr_id: Text, force: bool) -> None: """Kills Fleespeak on the given client.""" die_req = fs_system_pb2.DieRequest(force=force) fs_msg = fs_common_pb2.Message() fs_msg.message_type = "Die" fs_msg.destination.client_id = GRRIDToFleetspeakID(grr_id) fs_msg.destination.service_name = "system" fs_msg.data.Pack(die_req) fleetspeak_connector.CONN.outgoing.InsertMessage(fs_msg)
def testKillFleetspeak(self, mock_conn): fleetspeak_utils.KillFleetspeak("C.1000000000000000", True) mock_conn.outgoing.InsertMessage.assert_called_once() insert_args, _ = mock_conn.outgoing.InsertMessage.call_args fs_message = insert_args[0] self.assertEqual(fs_message.message_type, "Die") self.assertEqual(fs_message.destination.client_id, b"\x10\x00\x00\x00\x00\x00\x00\x00") self.assertEqual(fs_message.destination.service_name, "system") die_req = fs_system_pb2.DieRequest() fs_message.data.Unpack(die_req) self.assertTrue(die_req.force)