python-kasa/pyHS100/tests/test_protocol.py

from unittest import TestCase
from ..protocol import TPLinkSmartHomeProtocol
import json


class TestTPLinkSmartHomeProtocol(TestCase):
    def test_encrypt(self):
        d = json.dumps({'foo': 1, 'bar': 2})
        encrypted = TPLinkSmartHomeProtocol.encrypt(d)
        # encrypt adds a 4 byte header
        encrypted = encrypted[4:]
        self.assertEqual(d, TPLinkSmartHomeProtocol.decrypt(encrypted))

    def test_encrypt_unicode(self):
        d = "{'snowman': '\u2603'}"

        e = bytes([208, 247, 132, 234, 133, 242, 159, 254, 144, 183,
                   141, 173, 138, 104, 240, 115, 84, 41])

        encrypted = TPLinkSmartHomeProtocol.encrypt(d)
        # encrypt adds a 4 byte header
        encrypted = encrypted[4:]

        self.assertEqual(e, encrypted)

    def test_decrypt_unicode(self):
        e = bytes([208, 247, 132, 234, 133, 242, 159, 254, 144, 183,
                   141, 173, 138, 104, 240, 115, 84, 41])

        d = "{'snowman': '\u2603'}"

        self.assertEqual(d, TPLinkSmartHomeProtocol.decrypt(e))