irctokens/test/format.py

43 lines
1.5 KiB
Python

import unittest
import irctokens
class FormatTestTags(unittest.TestCase):
def test(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello"],
tags={"id": "\\" + " " + ";" + "\r\n"})
self.assertEqual(line, r"@id=\\\s\:\r\n PRIVMSG #channel hello")
def test_missing(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello"])
self.assertEqual(line, "PRIVMSG #channel hello")
def test_none_value(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello"],
tags={"a": None})
self.assertEqual(line, "@a PRIVMSG #channel hello")
def test_empty_value(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello"],
tags={"a": ""})
self.assertEqual(line, "@a PRIVMSG #channel hello")
class FormatTestSource(unittest.TestCase):
def test(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello"],
source="nick!user@host")
self.assertEqual(line, ":nick!user@host PRIVMSG #channel hello")
class FormatTestCommand(unittest.TestCase):
def test_lowercase(self):
line = irctokens.format("privmsg")
self.assertEqual(line, "PRIVMSG")
class FormatTestTrailing(unittest.TestCase):
def test_space(self):
line = irctokens.format("PRIVMSG", ["#channel", "hello world"])
self.assertEqual(line, "PRIVMSG #channel :hello world")
def test_no_space(self):
line = irctokens.format("PRIVMSG", ["#channel", "helloworld"])
self.assertEqual(line, "PRIVMSG #channel helloworld")