tr069/tests/models_tests.py

# SPDX-FileCopyrightText: 2020 The Magma Authors.
# SPDX-FileCopyrightText: 2022 Open Networking Foundation <support@opennetworking.org>
#
# SPDX-License-Identifier: BSD-3-Clause

import unittest

from tr069.models import DeviceIdStruct
from spyne import ComplexModelBase


class DeviceIdStructTests(unittest.TestCase):
    def test_as_dict_memory_leak(self):
        """
        Test to ensure as_dict() doesn't leak model instances
        """
        thing = DeviceIdStruct(
            Manufacturer='abc',
            OUI='def',
            ProductClass='ghi',
            SerialNumber='jkl',
        )
        res = thing.as_dict()
        self.assertEqual(
            {
                'Manufacturer': 'abc',
                'OUI': 'def',
                'ProductClass': 'ghi',
                'SerialNumber': 'jkl',
            },
            res,
        )
        # inspect the spyne.util.memoize object that wraps the staticmethod
        self.assertEqual(1, len(ComplexModelBase.get_flat_type_info.memo))

        # should produce a different result and not grow the size of memo
        thing.OUI = 'aaaa'
        res = thing.as_dict()
        self.assertEqual(
            {
                'Manufacturer': 'abc',
                'OUI': 'aaaa',
                'ProductClass': 'ghi',
                'SerialNumber': 'jkl',
            },
            res,
        )
        self.assertEqual(1, len(ComplexModelBase.get_flat_type_info.memo))

        # use a different object this time. Again should not grow memo
        thing = DeviceIdStruct(
            Manufacturer='abc',
            OUI='def',
            ProductClass='ghi',
            SerialNumber='jkl',
        )
        res = thing.as_dict()
        self.assertEqual(
            {
                'Manufacturer': 'abc',
                'OUI': 'def',
                'ProductClass': 'ghi',
                'SerialNumber': 'jkl',
            },
            res,
        )
        self.assertEqual(1, len(ComplexModelBase.get_flat_type_info.memo))