nested-lookup/test_lookup_api.py

from unittest import TestCase

from nested_lookup import nested_update, nested_delete


class BaseLookUpApi(TestCase):
    def setUp(self):
        self.sample_data1 = {
            "build_version": {
                "model_name": 'MacBook Pro',
                "build_version": {
                    "processor_name": 'Intel Core i7',
                    "processor_speed": '2.7 GHz',
                    "core_details": {
                        "build_version": '4',
                        "l2_cache(per_core)": '256 KB'
                    }
                },
                "number_of_cores": '4',
                "memory": '256 KB'
            },
            "os_details": {
                "product_version": '10.13.6',
                "build_version": '17G65'
            },
            "name": 'Test',
            "date": 'YYYY-MM-DD HH:MM:SS'
        }

        self.sample_data2 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "processor_details": [
                    {
                        "processor_name": 'Intel Core i7',
                        "processor_speed": '2.7 GHz'
                    },
                    {
                        "total_number_of_cores": '4',
                        "l2_cache(per_core)": '256 KB'
                    }
                ],
                "total_number_of_cores": '5',
                "memory": '16 GB'
            }
        }

        self.sample_data3 = {
            "values": [{
                "checks": [{
                    "monitoring_zones":
                    ["mzdfw", "mzfra", "mzhkg", "mziad",
                     "mzlon", "mzord", "mzsyd"]
                }]
            }]
        }

        self.sample_data4 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "total_number_of_cores": 0,
                "memory": False
            }
        }


class TestNestedDelete(BaseLookUpApi):
    def test_sample_data1(self):
        result = {
            "os_details": {
                "product_version": '10.13.6'
            },
            "name": 'Test',
            "date": 'YYYY-MM-DD HH:MM:SS'
        }
        self.assertEqual(
            result, nested_delete(self.sample_data1, 'build_version')
        )

    def test_sample_data2(self):
        result = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "total_number_of_cores": '5',
                "memory": '16 GB'
            }
        }
        self.assertEqual(
            result, nested_delete(self.sample_data2, 'processor_details')
        )

    def test_sample_data3(self):
        result = {"values": [{"checks": [{}]}]}
        self.assertEqual(
            result, nested_delete(self.sample_data3, 'monitoring_zones')
        )

    def test_sample_data4(self):
        result1 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "memory": False
            }
        }
        self.assertEqual(
            result1, nested_delete(self.sample_data4, 'total_number_of_cores')
        )
        result2 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "total_number_of_cores": 0
            }
        }
        self.assertEqual(
            result2, nested_delete(self.sample_data4, 'memory')
        )

    def test_nested_delete_in_place_false(self):
        """
            nested_delete with in_place argument set to 'False'
            should mutate and return a copy of the original document
        """
        before_id = id(self.sample_data1)
        result = nested_delete(
            self.sample_data1, 'build_version', in_place=False)
        after_id = id(result)
        # the object ids should _not_ match.
        self.assertNotEqual(before_id, after_id)

    def test_nested_delete_in_place_true(self):
        """
            nested_delete with in_place argument set to 'True'
            should mutate and return the original document
        """
        before_id = id(self.sample_data1)
        result = nested_delete(
            self.sample_data1, 'build_version', in_place=True)
        after_id = id(result)
        # the object ids should match.
        self.assertEqual(before_id, after_id)


class TestNestedUpdate(BaseLookUpApi):
    def test_sample_data1(self):
        result = {
            "build_version": "Test1",
            "os_details": {
                "product_version": '10.13.6',
                "build_version": 'Test1'
            },
            "name": 'Test',
            "date": 'YYYY-MM-DD HH:MM:SS'
        }
        self.assertEqual(
            result, nested_update(self.sample_data1, 'build_version', 'Test1')
        )

    def test_nested_update_in_place_false(self):
        """
            nested_update with in_place argument set to 'False'
            should mutate and return a copy of the original document
        """
        before_id = id(self.sample_data1)
        result = nested_update(
            self.sample_data1, 'build_version', 'Test2', in_place=False)
        after_id = id(result)
        # the object ids should _not_ match.
        self.assertNotEqual(before_id, after_id)

    def test_nested_update_in_place_true(self):
        """
            nested_update with in_place argument set to 'True'
            should mutate and return the original document
        """
        before_id = id(self.sample_data1)
        result = nested_update(
            self.sample_data1, 'build_version', 'Test2', in_place=True)
        after_id = id(result)
        # the object ids should match.
        self.assertEqual(before_id, after_id)

    def test_sample_data2(self):
        result = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "processor_details": {
                    'test_key1': 'test_value1'
                },
                "total_number_of_cores": '5',
                "memory": '16 GB'
            }
        }
        self.assertEqual(
            result, nested_update(
                self.sample_data2, 'processor_details',
                {'test_key1': 'test_value1'}
            )
        )

    def test_sample_data3(self):
        result = {
            "values": [{
                "checks": {
                    'key1': ['value1'],
                    'key2': 'value2'
                }
            }]
        }
        self.assertEqual(
            result, nested_update(
                self.sample_data3, 'checks',
                {
                    'key1': ['value1'],
                    'key2': 'value2'
                }
            )
        )

    def test_sample_data4(self):
        result1 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "total_number_of_cores": 1,
                "memory": False
            }
        }
        self.assertEqual(
            result1, nested_update(
                self.sample_data4, key='total_number_of_cores',
                value=1
            )
        )
        result2 = {
            "hardware_details": {
                "model_name": 'MacBook Pro',
                "total_number_of_cores": 0,
                "memory": True
            }
        }
        self.assertEqual(
            result2, nested_update(
                self.sample_data4, key='memory', value=True
            )
        )