mirror of
https://github.com/jlu5/PyLink.git
synced 2024-11-01 09:19:23 +01:00
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
"""
|
|
Tests for IRC parsers.
|
|
"""
|
|
from pathlib import Path
|
|
import unittest
|
|
|
|
import yaml
|
|
|
|
PARSER_DATA_PATH = Path(__file__).parent.resolve() / 'parser-tests' / 'tests'
|
|
print(PARSER_DATA_PATH)
|
|
'''
|
|
spec = importlib.util.spec_from_file_location("parser_tests", PARSER_DATA_PATH / 'data.py')
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
|
|
print(parser_tests)
|
|
'''
|
|
|
|
from pylinkirc.protocols.ircs2s_common import IRCCommonProtocol
|
|
|
|
class MessageParserTest(unittest.TestCase):
|
|
|
|
def testMessageSplit(self):
|
|
with open(PARSER_DATA_PATH / 'msg-split.yaml') as f:
|
|
splittest_data = yaml.safe_load(f)
|
|
|
|
for testdata in splittest_data['tests']:
|
|
inp = testdata['input']
|
|
atoms = testdata['atoms']
|
|
|
|
with self.subTest():
|
|
expected = []
|
|
has_source = False
|
|
if 'source' in atoms:
|
|
has_source = True
|
|
expected.append(atoms['source'])
|
|
|
|
if 'verb' in atoms:
|
|
expected.append(atoms['verb'])
|
|
|
|
if 'params' in atoms:
|
|
expected.extend(atoms['params'])
|
|
|
|
if 'tags' in atoms:
|
|
# HACK: in PyLink, message tags are processed in handle_events(), which is a dynamic
|
|
# method that relies on command handlers being present. So we can't reasonably test
|
|
# them here (plus handle_events() outputs params as a command-specific dict instead of)
|
|
# lists)
|
|
self.assertEqual(atoms['tags'], IRCCommonProtocol.parse_message_tags(inp.split(" ")),
|
|
"Parse test failed for message tags: %r" % inp)
|
|
_, inp = inp.split(" ", 1)
|
|
if has_source:
|
|
parts = IRCCommonProtocol.parse_prefixed_args(inp)
|
|
else:
|
|
parts = IRCCommonProtocol.parse_args(inp)
|
|
self.assertEqual(expected, parts, "Parse test failed for string: %r" % inp)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|