mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
232 lines
9 KiB
Python
232 lines
9 KiB
Python
import sys
|
|
import time
|
|
sys.path.append('modem')
|
|
|
|
import unittest
|
|
import unittest.mock
|
|
from config import CONFIG
|
|
import helpers
|
|
import queue
|
|
import threading
|
|
import base64
|
|
from command_arq_raw import ARQRawCommand
|
|
from state_manager import StateManager
|
|
from frame_dispatcher import DISPATCHER
|
|
import random
|
|
import structlog
|
|
import numpy as np
|
|
from event_manager import EventManager
|
|
from data_frame_factory import DataFrameFactory
|
|
import codec2
|
|
import arq_session_irs
|
|
class TestModem:
|
|
def __init__(self, event_q):
|
|
self.data_queue_received = queue.Queue()
|
|
self.demodulator = unittest.mock.Mock()
|
|
self.event_manager = EventManager([event_q])
|
|
self.logger = structlog.get_logger('Modem')
|
|
|
|
def getFrameTransmissionTime(self, mode):
|
|
samples = 0
|
|
c2instance = codec2.open_instance(mode.value)
|
|
samples += codec2.api.freedv_get_n_tx_preamble_modem_samples(c2instance)
|
|
samples += codec2.api.freedv_get_n_tx_modem_samples(c2instance)
|
|
samples += codec2.api.freedv_get_n_tx_postamble_modem_samples(c2instance)
|
|
time = samples / 8000
|
|
return time
|
|
|
|
def transmit(self, mode, repeats: int, repeat_delay: int, frames: bytearray) -> bool:
|
|
|
|
# Simulate transmission time
|
|
tx_time = self.getFrameTransmissionTime(mode) + 0.1 # PTT
|
|
self.logger.info(f"TX {tx_time} seconds...")
|
|
threading.Event().wait(tx_time)
|
|
|
|
transmission = {
|
|
'mode': mode,
|
|
'bytes': frames,
|
|
}
|
|
self.data_queue_received.put(transmission)
|
|
|
|
class TestARQSession(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
config_manager = CONFIG('modem/config.ini.example')
|
|
cls.config = config_manager.read()
|
|
cls.logger = structlog.get_logger("TESTS")
|
|
cls.frame_factory = DataFrameFactory(cls.config)
|
|
|
|
# ISS
|
|
cls.iss_state_manager = StateManager(queue.Queue())
|
|
cls.iss_event_manager = EventManager([queue.Queue()])
|
|
cls.iss_event_queue = queue.Queue()
|
|
cls.iss_modem = TestModem(cls.iss_event_queue)
|
|
cls.iss_frame_dispatcher = DISPATCHER(cls.config,
|
|
cls.iss_event_manager,
|
|
cls.iss_state_manager,
|
|
cls.iss_modem)
|
|
|
|
# IRS
|
|
cls.irs_state_manager = StateManager(queue.Queue())
|
|
cls.irs_event_manager = EventManager([queue.Queue()])
|
|
cls.irs_event_queue = queue.Queue()
|
|
cls.irs_modem = TestModem(cls.irs_event_queue)
|
|
cls.irs_frame_dispatcher = DISPATCHER(cls.config,
|
|
cls.irs_event_manager,
|
|
cls.irs_state_manager,
|
|
cls.irs_modem)
|
|
|
|
# Frame loss probability in %
|
|
cls.loss_probability = 30
|
|
|
|
cls.channels_running = True
|
|
|
|
def channelWorker(self, modem_transmit_queue: queue.Queue, frame_dispatcher: DISPATCHER):
|
|
while self.channels_running:
|
|
# Transfer data between both parties
|
|
try:
|
|
transmission = modem_transmit_queue.get(timeout=1)
|
|
if random.randint(0, 100) < self.loss_probability:
|
|
self.logger.info(f"[{threading.current_thread().name}] Frame lost...")
|
|
continue
|
|
|
|
frame_bytes = transmission['bytes']
|
|
frame_dispatcher.new_process_data(frame_bytes, None, len(frame_bytes), 0, 0)
|
|
except queue.Empty:
|
|
continue
|
|
self.logger.info(f"[{threading.current_thread().name}] Channel closed.")
|
|
|
|
def waitForSession(self, q, outbound = False):
|
|
key = 'arq-transfer-outbound' if outbound else 'arq-transfer-inbound'
|
|
while True:
|
|
ev = q.get()
|
|
if key in ev and ('success' in ev[key] or 'ABORTED' in ev[key]):
|
|
self.logger.info(f"[{threading.current_thread().name}] {key} session ended.")
|
|
break
|
|
|
|
def establishChannels(self):
|
|
self.channels_running = True
|
|
self.iss_to_irs_channel = threading.Thread(target=self.channelWorker,
|
|
args=[self.iss_modem.data_queue_received,
|
|
self.irs_frame_dispatcher],
|
|
name = "ISS to IRS channel")
|
|
self.iss_to_irs_channel.start()
|
|
|
|
self.irs_to_iss_channel = threading.Thread(target=self.channelWorker,
|
|
args=[self.irs_modem.data_queue_received,
|
|
self.iss_frame_dispatcher],
|
|
name = "IRS to ISS channel")
|
|
self.irs_to_iss_channel.start()
|
|
|
|
def waitAndCloseChannels(self):
|
|
self.waitForSession(self.iss_event_queue, True)
|
|
self.waitForSession(self.irs_event_queue, False)
|
|
self.channels_running = False
|
|
|
|
def testARQSessionSmallPayload(self):
|
|
# set Packet Error Rate (PER) / frame loss probability
|
|
self.loss_probability = 0
|
|
|
|
self.establishChannels()
|
|
params = {
|
|
'dxcall': "XX1XXX-1",
|
|
'data': base64.b64encode(bytes("Hello world!", encoding="utf-8")),
|
|
'type': "raw_lzma"
|
|
}
|
|
cmd = ARQRawCommand(self.config, self.iss_state_manager, self.iss_event_queue, params)
|
|
cmd.run(self.iss_event_queue, self.iss_modem)
|
|
self.waitAndCloseChannels()
|
|
del cmd
|
|
|
|
def DisabledtestARQSessionLargePayload(self):
|
|
# set Packet Error Rate (PER) / frame loss probability
|
|
self.loss_probability = 0
|
|
|
|
self.establishChannels()
|
|
params = {
|
|
'dxcall': "XX1XXX-1",
|
|
'data': base64.b64encode(np.random.bytes(1000)),
|
|
'type': "raw_lzma"
|
|
}
|
|
cmd = ARQRawCommand(self.config, self.iss_state_manager, self.iss_event_queue, params)
|
|
cmd.run(self.iss_event_queue, self.iss_modem)
|
|
|
|
self.waitAndCloseChannels()
|
|
del cmd
|
|
|
|
def testARQSessionAbortTransmissionISS(self):
|
|
# set Packet Error Rate (PER) / frame loss probability
|
|
self.loss_probability = 0
|
|
|
|
self.establishChannels()
|
|
params = {
|
|
'dxcall': "XX1XXX-1",
|
|
'data': base64.b64encode(np.random.bytes(100)),
|
|
}
|
|
cmd = ARQRawCommand(self.config, self.iss_state_manager, self.iss_event_queue, params)
|
|
cmd.run(self.iss_event_queue, self.iss_modem)
|
|
|
|
threading.Event().wait(np.random.randint(1,10))
|
|
for id in self.iss_state_manager.arq_iss_sessions:
|
|
self.iss_state_manager.arq_iss_sessions[id].abort_transmission()
|
|
|
|
self.waitAndCloseChannels()
|
|
del cmd
|
|
|
|
def testARQSessionAbortTransmissionIRS(self):
|
|
# set Packet Error Rate (PER) / frame loss probability
|
|
self.loss_probability = 0
|
|
|
|
self.establishChannels()
|
|
params = {
|
|
'dxcall': "XX1XXX-1",
|
|
'data': base64.b64encode(np.random.bytes(100)),
|
|
}
|
|
cmd = ARQRawCommand(self.config, self.iss_state_manager, self.iss_event_queue, params)
|
|
cmd.run(self.iss_event_queue, self.iss_modem)
|
|
|
|
threading.Event().wait(np.random.randint(1,10))
|
|
for id in self.irs_state_manager.arq_irs_sessions:
|
|
self.irs_state_manager.arq_irs_sessions[id].abort_transmission()
|
|
|
|
self.waitAndCloseChannels()
|
|
del cmd
|
|
|
|
def testSessionCleanupISS(self):
|
|
|
|
params = {
|
|
'dxcall': "XX1XXX-1",
|
|
'data': base64.b64encode(np.random.bytes(100)),
|
|
}
|
|
cmd = ARQRawCommand(self.config, self.iss_state_manager, self.iss_event_queue, params)
|
|
cmd.run(self.iss_event_queue, self.iss_modem)
|
|
for session_id in self.iss_state_manager.arq_iss_sessions:
|
|
session = self.iss_state_manager.arq_iss_sessions[session_id]
|
|
ISS_States = session.state_enum
|
|
session.state = ISS_States.FAILED
|
|
session.session_ended = time.time() - 1000
|
|
if session.is_session_outdated():
|
|
self.logger.info(f"session [{session_id}] outdated - deleting it")
|
|
self.iss_state_manager.remove_arq_iss_session(session_id)
|
|
break
|
|
del cmd
|
|
|
|
def testSessionCleanupIRS(self):
|
|
session = arq_session_irs.ARQSessionIRS(self.config,
|
|
self.irs_modem,
|
|
'AA1AAA-1',
|
|
random.randint(0, 255))
|
|
self.irs_state_manager.register_arq_irs_session(session)
|
|
for session_id in self.irs_state_manager.arq_irs_sessions:
|
|
session = self.irs_state_manager.arq_irs_sessions[session_id]
|
|
irs_States = session.state_enum
|
|
session.state = irs_States.FAILED
|
|
session.session_ended = time.time() - 1000
|
|
if session.is_session_outdated():
|
|
self.logger.info(f"session [{session_id}] outdated - deleting it")
|
|
self.irs_state_manager.remove_arq_irs_session(session_id)
|
|
break
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|