test_message.py

import unittest

from landscape.lib.schema import Constant
from landscape.lib.schema import Int
from landscape.message_schemas.message import Message

class MessageTest(unittest.TestCase):
    def test_coerce(self):
        """The L{Message} schema should be very similar to KeyDict."""
        schema = Message("foo", {"data": Int()})
        self.assertEqual(
            schema.coerce({"type": "foo", "data": 3}),
            {"type": "foo", "data": 3},
        )

def test_coerce_bytes_to_str(self):
        """
        The L{Constant} schema type recognizes bytestrings that decode to
        matching strings.
        """
        constant = Constant("register")
        self.assertEqual(constant.coerce(b"register"), "register")

def test_timestamp(self):
        """L{Message} schemas should accept C{timestamp} keys."""
        schema = Message("bar", {})
        self.assertEqual(
            schema.coerce({"type": "bar", "timestamp": 0.33}),
            {"type": "bar", "timestamp": 0.33},
        )

def test_api(self):
        """L{Message} schemas should accept C{api} keys."""
        schema = Message("baz", {})
        self.assertEqual(
            schema.coerce({"type": "baz", "api": b"whatever"}),
            {"type": "baz", "api": b"whatever"},
        )

def test_api_none(self):
        """L{Message} schemas should accept None for C{api}."""
        schema = Message("baz", {})
        self.assertEqual(
            schema.coerce({"type": "baz", "api": None}),
            {"type": "baz", "api": None},
        )

def test_optional(self):
        """The L{Message} schema should allow additional optional keys."""
        schema = Message("foo", {"data": Int()}, optional=["data"])
        self.assertEqual(schema.coerce({"type": "foo"}), {"type": "foo"})

def test_type(self):
        """The C{type} should be introspectable on L{Message} objects."""
        schema = Message("foo", {})
        self.assertEqual(schema.type, "foo")

def test_with_unknown_fields(self):
        """
        The L{Message} schema discards unknown fields when coercing values.
        """
        schema = Message("foo", {})
        self.assertEqual(
            {"type": "foo"},
            schema.coerce({"type": "foo", "crap": 123}),
        )

docroot: /var/www/469ed2f6-4d4b-4a8a-9f5e-ee4ef04d3bc8/public_html

curdir: /usr/lib/python3/dist-packages/landscape/message_schemas

	.	4096
	..	4096
X	__init__.py	151
	__pycache__	4096
X	message.py	1624
X	server_bound.py	24408
X	test_message.py	2282