72 lines
3.0 KiB
Python
72 lines
3.0 KiB
Python
# 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 karbor.services.protection.graph as graph
|
|
from karbor.services.protection.resource_graph import ResourceGraphContext
|
|
from karbor.services.protection.resource_graph \
|
|
import ResourceGraphWalkerListener
|
|
|
|
from karbor.tests import base
|
|
from karbor.tests.unit.protection.fakes import FakeProtectionPlugin
|
|
from karbor.tests.unit.protection.fakes import plan_resources
|
|
from karbor.tests.unit.protection.fakes import resource_graph
|
|
from karbor.tests.unit.protection.fakes import resource_map
|
|
|
|
|
|
class ResourceGraphWalkerListenerTest(base.TestCase):
|
|
def test_resource_graph_walker_listener_plan(self):
|
|
expected_calls = [
|
|
("on_resource_start", 'A', True),
|
|
("on_resource_start", 'C', True),
|
|
("on_resource_start", 'D', True),
|
|
("on_resource_end", 'D'),
|
|
("on_resource_start", 'E', True),
|
|
("on_resource_end", 'E'),
|
|
("on_resource_end", 'C'),
|
|
("on_resource_end", 'A'),
|
|
("on_resource_start", 'B', True),
|
|
("on_resource_start", 'C', False),
|
|
("on_resource_start", 'D', False),
|
|
("on_resource_end", 'D'),
|
|
("on_resource_start", 'E', False),
|
|
("on_resource_end", 'E'),
|
|
("on_resource_end", 'C'),
|
|
("on_resource_end", 'B'),
|
|
]
|
|
|
|
fake_cntxt = "fake_cntxt"
|
|
fake_protection_plugin = FakeProtectionPlugin(expected_calls)
|
|
fake_context = ResourceGraphContext(
|
|
fake_cntxt,
|
|
plugin_map={"fake_plugin": fake_protection_plugin})
|
|
listener = ResourceGraphWalkerListener(fake_context)
|
|
|
|
walker = graph.GraphWalker()
|
|
walker.register_listener(listener)
|
|
walker.walk_graph(graph.build_graph(plan_resources,
|
|
resource_map.__getitem__))
|
|
self.assertEqual(len(listener.context.status_getters), 5)
|
|
|
|
def tearDown(self):
|
|
super(ResourceGraphWalkerListenerTest, self).tearDown()
|
|
|
|
|
|
class SerializeResourceGraphTest(base.TestCase):
|
|
def test_serialize_deserialize_packed_resource_graph(self):
|
|
serialized_resource_graph = graph.serialize_resource_graph(
|
|
resource_graph)
|
|
deserialized_resource_graph = graph.deserialize_resource_graph(
|
|
serialized_resource_graph)
|
|
self.assertEqual(len(resource_graph), len(deserialized_resource_graph))
|
|
for start_node in resource_graph:
|
|
self.assertIn(start_node, deserialized_resource_graph)
|