diff options
author | Zuul <zuul@review.opendev.org> | 2021-04-21 19:46:36 +0000 |
---|---|---|
committer | Gerrit Code Review <review@openstack.org> | 2021-04-21 19:46:36 +0000 |
commit | b75a126ec639d96e3fd76816fdd8f2ae70d3048b (patch) | |
tree | b5f2dd10f19f680ed795a6033ceb7e9c6136956f | |
parent | fda85f1f1c19e5a72a7de1f687683b673206e4a1 (diff) | |
parent | 4f1608846265ef76b4aad7922b3f43ed766bdee7 (diff) | |
download | oslo-vmware-b75a126ec639d96e3fd76816fdd8f2ae70d3048b.tar.gz |
Merge "Add serialize_object() helper function"
-rw-r--r-- | oslo_vmware/tests/test_vim_util.py | 30 | ||||
-rw-r--r-- | oslo_vmware/vim_util.py | 18 |
2 files changed, 48 insertions, 0 deletions
diff --git a/oslo_vmware/tests/test_vim_util.py b/oslo_vmware/tests/test_vim_util.py index 43c02bc..e35294c 100644 --- a/oslo_vmware/tests/test_vim_util.py +++ b/oslo_vmware/tests/test_vim_util.py @@ -527,3 +527,33 @@ class VimUtilTest(base.TestCase): self.assertEqual({"test_name_0": "test_val_0", "test_name_1": "test_val_1"}, vim_util.propset_dict(mock_propset)) + + def test_serialize_object(self): + self.assertEqual({}, vim_util.serialize_object({})) + + mobj1 = mock.MagicMock() + mobj1.__keylist__ = ['asdf'] + mobj1.keys = lambda: ['asdf'] + mobj1.__getitem__.side_effect = [1] + + mobj2 = mock.Mock() + + mobj3 = mock.MagicMock() + mobj3.__keylist__ = ['subkey1', 'subkey2'] + mobj3.keys = lambda: ['subkey1', 'subkey2'] + mobj3.__getitem__.side_effect = ['subvalue1', True] + + mobj4 = 12 + + obj = { + 'foo': mobj1, + 'bar': [mobj2, mobj3], + 'baz': mobj4 + } + + expected = { + 'foo': {'asdf': 1}, + 'bar': [mobj2, {'subkey1': 'subvalue1', 'subkey2': True}], + 'baz': 12 + } + self.assertEqual(expected, vim_util.serialize_object(obj)) diff --git a/oslo_vmware/vim_util.py b/oslo_vmware/vim_util.py index b2c9d2c..945c98a 100644 --- a/oslo_vmware/vim_util.py +++ b/oslo_vmware/vim_util.py @@ -694,3 +694,21 @@ def storage_placement_spec(client_factory, spec.resourcePool = res_pool_ref spec.host = host_ref return spec + + +def serialize_object(obj): + """Convert Suds object into serializable format - a dict.""" + d = {} + for k, v in dict(obj).items(): + if hasattr(v, '__keylist__'): + d[k] = serialize_object(v) + elif isinstance(v, list): + d[k] = [] + for item in v: + if hasattr(item, '__keylist__'): + d[k].append(serialize_object(item)) + else: + d[k].append(item) + else: + d[k] = v + return d |