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)
