osm_ro/tests/test_utils.py

# -*- coding: utf-8 -*-
# pylint: disable=E1101

import unittest

from ..utils import (
    get_arg,
    inject_args,
    remove_extra_items,
)


class TestUtils(unittest.TestCase):
    def test_inject_args_curries_arguments(self):
        fn = inject_args(lambda a=None, b=None: a + b, a=3, b=5)
        self.assertEqual(fn(), 8)

    def test_inject_args_doesnt_add_arg_if_not_needed(self):
        fn = inject_args(lambda: 7, a=1, b=2)
        self.assertEqual(fn(), 7)
        fn = inject_args(lambda a=None: a, b=2)
        self.assertEqual(fn(1), 1)

    def test_inject_args_knows_how_to_handle_arg_order(self):
        fn = inject_args(lambda a=None, b=None: b - a, a=3)
        self.assertEqual(fn(b=4), 1)
        fn = inject_args(lambda b=None, a=None: b - a, a=3)
        self.assertEqual(fn(b=4), 1)

    def test_inject_args_works_as_decorator(self):
        fn = inject_args(x=1)(lambda x=None: x)
        self.assertEqual(fn(), 1)

    def test_get_arg__positional(self):
        def _fn(x, y, z):
            return x + y + z

        x = get_arg("x", _fn, (1, 3, 4), {})
        self.assertEqual(x, 1)
        y = get_arg("y", _fn, (1, 3, 4), {})
        self.assertEqual(y, 3)
        z = get_arg("z", _fn, (1, 3, 4), {})
        self.assertEqual(z, 4)

    def test_get_arg__keyword(self):
        def _fn(x, y, z=5):
            return x + y + z

        z = get_arg("z", _fn, (1, 2), {"z": 3})
        self.assertEqual(z, 3)



    def test_remove_extra_items__keep_aditional_properties(self):
        schema = {
            "type": "object",
            "properties": {
                "a": {
                    "type": "object",
                    "properties": {
                        "type": "object",
                        "properties": {"b": "string"},
                    },
                    "additionalProperties": True,
                }
            },
        }

        example = {"a": {"b": 1, "c": 2}, "d": 3}
        deleted = remove_extra_items(example, schema)
        self.assertIn("d", deleted)
        self.assertIs(example.get("d"), None)
        self.assertEqual(example["a"]["c"], 2)


if __name__ == "__main__":
    unittest.main()