ironic-python-agent/ironic_python_agent/tests/unit/test_encoding.py
Dmitry Tantsur a4855c544c Fix serializing ironic-lib exceptions
Change-Id: If1408e4b81d263c56b4bbab618dd0737db5f762e
Story: #2007889
Task: #40268
2020-07-02 12:18:53 +02:00

83 lines
2.7 KiB
Python

# Copyright (C) 2015 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import json
from ironic_lib import exception as lib_exc
from ironic_python_agent import encoding
from ironic_python_agent.tests.unit import base
class SerializableTesting(encoding.Serializable):
serializable_fields = ('jack', 'jill')
def __init__(self, jack, jill):
self.jack = jack
self.jill = jill
class SerializableComparableTesting(encoding.SerializableComparable):
serializable_fields = ('jack', 'jill')
def __init__(self, jack, jill):
self.jack = jack
self.jill = jill
class TestSerializable(base.IronicAgentTest):
def test_baseclass_serialize(self):
obj = encoding.Serializable()
self.assertEqual({}, obj.serialize())
def test_childclass_serialize(self):
expected = {'jack': 'hello', 'jill': 'world'}
obj = SerializableTesting('hello', 'world')
self.assertEqual(expected, obj.serialize())
class TestSerializableComparable(base.IronicAgentTest):
def test_childclass_equal(self):
obj1 = SerializableComparableTesting('hello', 'world')
obj2 = SerializableComparableTesting('hello', 'world')
self.assertEqual(obj1, obj2)
def test_childclass_notequal(self):
obj1 = SerializableComparableTesting('hello', 'world')
obj2 = SerializableComparableTesting('hello', 'world2')
self.assertNotEqual(obj1, obj2)
def test_childclass_hash(self):
# Ensure __hash__ is None
obj = SerializableComparableTesting('hello', 'world')
self.assertIsNone(obj.__hash__)
class TestEncoder(base.IronicAgentTest):
encoder = encoding.RESTJSONEncoder()
def test_encoder(self):
expected = {'jack': 'hello', 'jill': 'world'}
obj = SerializableTesting('hello', 'world')
self.assertEqual(expected, json.loads(self.encoder.encode(obj)))
def test_ironic_lib(self):
obj = lib_exc.InstanceDeployFailure(reason='boom')
encoded = json.loads(self.encoder.encode(obj))
self.assertEqual(500, encoded['code'])
self.assertEqual('InstanceDeployFailure', encoded['type'])
self.assertIn('boom', encoded['message'])