tests/test_aaa.py - dev - kotkov/eco

tests/test_aaa.py
54 строки2 KB
Arseny Kotkov
Add multithreading logging to file with log.ecolog
04 мар 2025, 08:01
04 мар 2025, 08:0103de008
from threading import Thread

from aaa import AAA

from log import ecolog
from rpc import Message
from scheduler import Scheduler
from slurm import Slurm
from utils import Socket


def test_registration():
    try:
        # Создаём экземпляр центра регистрации агентов.
        aaa = AAA()
        aaa_socket = Socket('127.0.0.1', 5000)

        # Создаём экземпляр агента доступа к вычислительным ресурсам.
        slurm = Slurm()
        slurm.name = "slurm"
        slurm.aaa = aaa_socket

        # Создаём экземпляр агента планирования сценариев.
        scheduler = Scheduler()
        scheduler.name = "scheduler"
        scheduler.aaa = aaa_socket

        # Запускаем RPC-сервер центра регистрации агентов.
        t0 = Thread(target=aaa.serve, args=(('127.0.0.1', 5000),))
        t0.daemon = True
        t0.start()

        # Запускаем RPC-сервер агента доступа к вычислительным ресурсам.
        t1 = Thread(target=slurm.serve, args=(('127.0.0.1', 50010),))
        t1.daemon = True
        t1.start()

        # Запускаем RPC-сервер агента планирования сценариев.
        t2 = Thread(target=scheduler.serve, args=(('127.0.0.1', 50020),))
        t2.daemon = True
        t2.start()

        ecolog()  # Пустая строка для повышения удобочитаемости вывода.

        # Регистрируем агентов в центре регистрации.
        ecolog(scheduler.register())
        ecolog(slurm.register())

        # Просим планировщика обратиться к агенту доступа к вычислительным
        # ресурсам по зарегистрированному имени и что-нибудь сделать.
        ecolog(scheduler.call_agent('slurm', Message(method='do_slurm_stuff', params=[])))

    except KeyboardInterrupt:
        exit(0)