Viewing file:  test_message.py (2.15 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
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}))
|