FM / message

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

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

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
message.py	1522	0644	edit dl rm
server_bound.py	19536	0644	edit dl rm
test_message.py	2201	0644	edit dl rm
__init__.py	152	0644	edit dl rm

Edit: /usr/lib/python3/dist-packages/landscape/message_schemas/test_message.py (2201B)

import unittest

from landscape.lib.schema import Constant, 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}))

cmd: