22 lines
694 B
Python
22 lines
694 B
Python
import unittest
|
|
|
|
from secure_sms.infrastructure.gsm import _has_terminal_status, _sanitize_protocol_body
|
|
|
|
|
|
class GsmGatewayParsingTests(unittest.TestCase):
|
|
def test_terminal_status_requires_final_ok_line(self):
|
|
partial = '\r\n+CMGR: "REC READ","+9891","",""\r\n@S:SYM|abcOKxyz'
|
|
complete = partial + "\r\n\r\nOK\r\n"
|
|
|
|
self.assertFalse(_has_terminal_status(partial))
|
|
self.assertTrue(_has_terminal_status(complete))
|
|
|
|
def test_protocol_body_sanitizer_removes_control_bytes(self):
|
|
corrupted = "@S:SYM|abc\x11\x11DEF-_="
|
|
|
|
self.assertEqual(_sanitize_protocol_body(corrupted), "@S:SYM|abcDEF-_=")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|