mirror of
https://github.com/craigerl/aprsd.git
synced 2024-11-18 14:22:01 -05:00
Hemna
fe0d71de4d
This patch replaces the ratelimiter library with rush for rate limiting as the ratelimiter package doesn't work with python 3.11. This patch also refactors the flask.pu to admin_web.py and aprsd.py to main.py
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import typing as t
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from click.testing import CliRunner
|
|
from oslo_config import cfg
|
|
|
|
from aprsd import conf # noqa : F401
|
|
from aprsd.cmds import send_message # noqa
|
|
from aprsd.main import cli
|
|
|
|
from .. import fake
|
|
|
|
|
|
CONF = cfg.CONF
|
|
F = t.TypeVar("F", bound=t.Callable[..., t.Any])
|
|
|
|
|
|
class TestSendMessageCommand(unittest.TestCase):
|
|
|
|
def config_and_init(self, login=None, password=None):
|
|
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
|
CONF.trace_enabled = False
|
|
CONF.watch_list.packet_keep_count = 1
|
|
if login:
|
|
CONF.aprs_network.login = login
|
|
if password:
|
|
CONF.aprs_network.password = password
|
|
|
|
CONF.admin.user = "admin"
|
|
CONF.admin.password = "password"
|
|
|
|
@mock.patch("aprsd.logging.log.setup_logging")
|
|
def test_no_tocallsign(self, mock_logging):
|
|
"""Make sure we get an error if there is no tocallsign."""
|
|
|
|
self.config_and_init(
|
|
login="something",
|
|
password="another",
|
|
)
|
|
runner = CliRunner()
|
|
|
|
result = runner.invoke(
|
|
cli, ["send-message"],
|
|
catch_exceptions=False,
|
|
)
|
|
assert result.exit_code == 2
|
|
assert "Error: Missing argument 'TOCALLSIGN'" in result.output
|
|
|
|
@mock.patch("aprsd.logging.log.setup_logging")
|
|
def test_no_command(self, mock_logging):
|
|
"""Make sure we get an error if there is no command."""
|
|
|
|
self.config_and_init(
|
|
login="something",
|
|
password="another",
|
|
)
|
|
runner = CliRunner()
|
|
|
|
result = runner.invoke(
|
|
cli, ["send-message", "WB4BOR"],
|
|
catch_exceptions=False,
|
|
)
|
|
assert result.exit_code == 2
|
|
assert "Error: Missing argument 'COMMAND...'" in result.output
|