13 changed files with 262 additions and 26 deletions
@ -0,0 +1,41 @@
|
||||
import pytest |
||||
from contextlib import closing |
||||
from tests.lib import * # NOQA - fixtures |
||||
from time import sleep |
||||
import datetime |
||||
|
||||
|
||||
@pytest.fixture |
||||
def rbot(fakebot): |
||||
""" |
||||
Provide a bot loaded with the Calc module. Clear the database. |
||||
""" |
||||
fakebot.botconfig["module_configs"]["Remind"] = {"mytimezone": "US/Pacific", "precision": 0.2} |
||||
fakebot.loadmodule("SQLite") |
||||
with closing(fakebot.moduleInstances["SQLite"].opendb("remind.db")) as db: |
||||
db.query("DROP TABLE IF EXISTS `reminders`;") |
||||
fakebot.loadmodule("Remind") |
||||
return fakebot |
||||
|
||||
|
||||
@pytest.mark.slow |
||||
def test_remind_in(rbot): |
||||
rbot.feed_line(".in 3s frig off") |
||||
rbot.act_PRIVMSG.assert_called_once_with('#test', 'chatter: Ok, talk to you in approx 0h0m') |
||||
rbot.act_PRIVMSG.reset_mock() |
||||
sleep(2.5) |
||||
rbot.act_PRIVMSG.assert_not_called() |
||||
sleep(1) |
||||
rbot.act_PRIVMSG.assert_called_once_with('#test', 'chatter: Reminder: frig off') |
||||
|
||||
|
||||
@pytest.mark.slow |
||||
def test_remind_at(rbot): |
||||
then = datetime.datetime.now() + datetime.timedelta(seconds=3) |
||||
rbot.feed_line(".at {} frig off".format(then.strftime("%H:%M:%SPDT"))) |
||||
rbot.act_PRIVMSG.assert_called_once_with('#test', 'chatter: Ok, will do. Approx 0h0m to go.') |
||||
rbot.act_PRIVMSG.reset_mock() |
||||
sleep(2) |
||||
rbot.act_PRIVMSG.assert_not_called() |
||||
sleep(2) |
||||
rbot.act_PRIVMSG.assert_called_once_with('#test', 'chatter: Reminder: frig off') |
||||
@ -0,0 +1,87 @@
|
||||
from tests.lib import * # NOQA - fixtures |
||||
|
||||
from unittest.mock import MagicMock, call |
||||
from pyircbot.rpc import BotRPC |
||||
from pyircbot.rpcclient import connect |
||||
from random import randint |
||||
from time import sleep |
||||
|
||||
|
||||
def test_rpc(monkeypatch): |
||||
port = randint(40000, 65000) |
||||
m = MagicMock() |
||||
m.botconfig = {"bot": {"rpcbind": "127.0.0.1", "rpcport": port}} |
||||
server = BotRPC(m) |
||||
sleep(0.05) |
||||
|
||||
calltrack = MagicMock() |
||||
|
||||
def fake(*args): |
||||
calltrack(*args) |
||||
return args |
||||
|
||||
for k, v in server.server.funcs.items(): |
||||
server.server.funcs[k] = fake |
||||
|
||||
methods = [["importModule", "foo"], |
||||
["deportModule", "foo"], |
||||
["loadModule", "foo"], |
||||
["unloadModule", "foo"], |
||||
["reloadModule", "foo"], |
||||
["redoModule", "foo"], |
||||
["getLoadedModules"], |
||||
["pluginCommand", "foo", "foo", "foo"], |
||||
["setPluginVar", "foo", "foo"], |
||||
["getPluginVar", "foo", "foo", "foo"], |
||||
["eval", "foo"], |
||||
["exec", "foo"], |
||||
["quit", "foo"]] |
||||
|
||||
client = connect("127.0.0.1", port) |
||||
|
||||
for test in methods: |
||||
method = test[0] |
||||
args = test[1:] |
||||
server.server.funcs[method] = fake |
||||
print("Calling {} with: {}".format(method, args)) |
||||
getattr(client, method)(*args) |
||||
calltrack.assert_called_once_with(*args) |
||||
calltrack.reset_mock() |
||||
|
||||
|
||||
def test_rpc_internal(monkeypatch): |
||||
port = randint(40000, 65000) |
||||
m = MagicMock() |
||||
m.botconfig = {"bot": {"rpcbind": "127.0.0.1", "rpcport": port}} |
||||
server = BotRPC(m) |
||||
|
||||
methods = [["importModule", "foo"], |
||||
["deportModule", "foo"], |
||||
["loadModule", "foo"], |
||||
["unloadModule", "foo"], |
||||
["redoModule", "foo"],] |
||||
|
||||
for test in methods: |
||||
method = test[0] |
||||
args = test[1:] |
||||
getattr(server, method)(*args) |
||||
getattr(m, method.lower()).assert_called_once_with(*args) |
||||
getattr(m, method.lower()).reset_mock() |
||||
|
||||
m.moduleInstances = {"Foo": None, "Bar": None} |
||||
assert server.getLoadedModules() == ["Foo", "Bar"] |
||||
|
||||
m.reset_mock() |
||||
|
||||
server.reloadModule("Foo") |
||||
m.unloadmodule.assert_called_once_with("Foo") |
||||
m.loadmodule.assert_called_once_with("Foo") |
||||
|
||||
m.reset_mock() |
||||
|
||||
# ["pluginCommand", "foo", "foo", "foo"], |
||||
# ["setPluginVar", "foo", "foo"], |
||||
# ["getPluginVar", "foo", "foo", "foo"] |
||||
# ["eval", "foo"], |
||||
# ["exec", "foo"], |
||||
# ["quit", "foo"]] |
||||
Loading…
Reference in new issue