Ocultación de información

from securemessenger import SecureMessenger

# Gestor de casos de prueba
test_case = input()
if test_case == "default_test":
    # Caso de prueba estándar del problema original
    messenger = SecureMessenger("user1")
    
    # Intenta añadir mensajes antes de iniciar sesión
    print(messenger.add_message("Hello World!"))
    print(messenger.add_message("Secure Message"))
    
    # Intenta iniciar sesión con una contraseña incorrecta
    print(messenger.login("wrong_pass"))
    
    # Inicia sesión con la contraseña correcta
    print(messenger.login("secure123"))
    
    # Añade mensajes después de iniciar sesión correctamente
    print(messenger.add_message("Hello World!"))
    print(messenger.add_message("Secure Message"))
    
    # Recupera los mensajes
    print(messenger.get_messages())
    
    # Comprueba los intentos de inicio de sesión
    print(messenger.get_login_attempts())
elif test_case == "custom_password":
    # Prueba con una contraseña personalizada
    messenger = SecureMessenger("admin", "admin123")
    print(messenger.login("wrong_password"))
    print(messenger.login("admin123"))
    print(messenger.get_login_attempts())
elif test_case == "multiple_login_attempts":
    # Prueba varios intentos de inicio de sesión
    messenger = SecureMessenger("user2")
    print(messenger.login("attempt1"))
    print(messenger.login("attempt2"))
    print(messenger.login("attempt3"))
    print(messenger.login("secure123"))
    print(messenger.get_login_attempts())
elif test_case == "message_management":
    # Gestión de Test message
    messenger = SecureMessenger("user3")
    print(messenger.login("secure123"))
    
    # Añade varios mensajes
    messages = ["Message 1", "Message 2", "Message 3", "Message 4", "Message 5"]
    for msg in messages:
        print(messenger.add_message(msg))
    
    # Verifica todos los mensajes
    print(messenger.get_messages())
elif test_case == "empty_messages":
    # Prueba el caso de mensajes vacíos
    messenger = SecureMessenger("user4")
    print(messenger.login("secure123"))
    print(messenger.get_messages())
elif test_case == "access_without_login":
    # Prueba el acceso sin iniciar sesión
    messenger = SecureMessenger("user5")
    print(messenger.add_message("Unauthorized message"))
    print(messenger.get_messages())
elif test_case == "attribute_privacy":
    # Prueba la privacidad de los atributos
    messenger = SecureMessenger("user6")
    
    # Atributo público
    print(f"Public username: {messenger.username}")
    
    # Atributos privados: estos deberían generar AttributeError
    try:
        print(messenger.__password)
    except AttributeError as e:
        print("Password is private: AttributeError")
    
    try:
        print(messenger.__messages)
    except AttributeError as e:
        print("Messages are private: AttributeError")
    
    try:
        print(messenger.__login_attempts)
    except AttributeError as e:
        print("Login attempts are private: AttributeError")
    
    try:
        print(messenger.__is_logged_in)
    except AttributeError as e:
        print("Login status is private: AttributeError")
elif test_case == "login_logout_sequence":
    # Esta prueba requiere ampliar la clase con un método logout
    # Para realizar la prueba, crearemos una subclase con funcionalidad de cierre de sesión
    class ExtendedMessenger(SecureMessenger):
        def logout(self):
            self._SecureMessenger__is_logged_in = False
            return "Logged out successfully"
    
    messenger = ExtendedMessenger("user7")
    print(messenger.login("secure123"))
    print(messenger.add_message("Test message"))
    print(messenger.get_messages())
    print(messenger.logout())
    print(messenger.get_messages())
elif test_case == "stress_test":
    # Prueba de estrés con muchos mensajes
    messenger = SecureMessenger("user8")
    print(messenger.login("secure123"))
    
    # Añade 100 mensajes
    for i in range(1, 101):
        messenger.add_message(f"Message {i}")
    
    # Verifica la cantidad de mensajes
    messages = messenger.get_messages()
    newline_char = '\n'  # Guarda el carácter de nueva línea en una variable
    message_count = len(messages.split(newline_char))
    print(f"Added 100 messages. Retrieved {message_count} messages.")