1 # Licensed to the Apache Software Foundation (ASF) under one or more
2 # contributor license agreements. See the NOTICE file distributed with
3 # this work for additional information regarding copyright ownership.
4 # The ASF licenses this file to You under the Apache License, Version 2.0
5 # (the "License"); you may not use this file except in compliance with
6 # the License. You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
18 from aria.orchestrator.workflows.api import task
19 from aria.orchestrator.workflows.builtin.uninstall import uninstall
21 from tests import mock
22 from tests import storage
24 from . import assert_node_uninstall_operations
29 context = mock.context.simple(str(tmpdir),
30 topology=mock.topology.create_simple_topology_three_nodes)
32 storage.release_sqlite_storage(context.model)
35 def test_uninstall(ctx):
37 uninstall_tasks = list(task.WorkflowTask(uninstall, ctx=ctx).topological_order(True))
39 assert len(uninstall_tasks) == 3
40 dependent_node_subgraph, dependency_node_subgraph1, dependency_node_subgraph2 = uninstall_tasks
41 dependent_node_tasks = list(dependent_node_subgraph.topological_order(reverse=True))
42 dependency_node1_tasks = list(dependency_node_subgraph1.topological_order(reverse=True))
43 dependency_node2_tasks = list(dependency_node_subgraph2.topological_order(reverse=True))
45 assert_node_uninstall_operations(operations=dependency_node1_tasks)
46 assert_node_uninstall_operations(operations=dependency_node2_tasks)
47 assert_node_uninstall_operations(operations=dependent_node_tasks, relationships=2)