| # |
| # Copyright (C) 2018 The Android Open Source Project |
| # |
| # 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. |
| # |
| """Test for ndk.deps.""" |
| from typing import Set |
| import unittest |
| |
| from ndk.deps import CyclicDependencyError |
| from ndk.deps import DependencyManager |
| from ndk.builds import Module |
| |
| |
| class MockModule(Module): |
| """A no-op module base.""" |
| def validate(self) -> None: |
| pass |
| |
| def build(self) -> None: |
| pass |
| |
| def install(self) -> None: |
| pass |
| |
| |
| # A basic cycle. The cycle logic is tested more thoroughly in test_graph.py, |
| # but we want to ensure that CyclicDependencyError is formatted nicely. |
| class CycleA(MockModule): |
| name = 'cycleA' |
| deps = {'cycleB'} |
| |
| |
| class CycleB(MockModule): |
| name = 'cycleB' |
| deps = {'cycleA'} |
| |
| |
| # A module with no dependents or dependencies. Should be immediately buildable. |
| class Isolated(MockModule): |
| name = 'isolated' |
| deps: Set[str] = set() |
| |
| |
| # A module that is not present in the build graph. |
| class Unknown(MockModule): |
| name = 'unknown' |
| deps: Set[str] = set() |
| |
| |
| # A simple chain of two modules. The first should be immediately buildable, and |
| # the second should become buildable after it completes. |
| class SimpleA(MockModule): |
| name = 'simpleA' |
| deps: Set[str] = set() |
| |
| |
| class SimpleB(MockModule): |
| name = 'simpleB' |
| deps = {'simpleA'} |
| |
| |
| # Slightly more complex module graph. |
| class ComplexA(MockModule): |
| name = 'complexA' |
| deps: Set[str] = set() |
| |
| |
| class ComplexB(MockModule): |
| name = 'complexB' |
| deps = {'complexA'} |
| |
| |
| class ComplexC(MockModule): |
| name = 'complexC' |
| deps = {'complexA'} |
| |
| |
| class ComplexD(MockModule): |
| name = 'complexD' |
| deps = {'complexA', 'complexB'} |
| |
| |
| class DependencyManagerTest(unittest.TestCase): |
| def test_cyclic_dependency_message(self) -> None: |
| """Test that a cycle raises the proper exception.""" |
| pattern = '^Detected cyclic dependency: cycleA -> cycleB -> cycleA$' |
| with self.assertRaisesRegex(CyclicDependencyError, pattern): |
| DependencyManager([CycleA(), CycleB()]) |
| |
| def test_empty_raises(self) -> None: |
| """Test that an empty module list raises.""" |
| with self.assertRaises(ValueError): |
| DependencyManager([]) |
| |
| def test_complete_invalid_module_raises(self) -> None: |
| """Test that completing an unknown module raises.""" |
| isolated = Isolated() |
| unknown = Unknown() |
| deps = DependencyManager([isolated]) |
| with self.assertRaises(KeyError): |
| deps.complete(unknown) |
| |
| def test_isolated(self) -> None: |
| """Test module graph with a single isolated vertex.""" |
| isolated = Isolated() |
| deps = DependencyManager([isolated]) |
| self.assertSetEqual({isolated}, deps.buildable_modules) |
| self.assertSetEqual({isolated}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(isolated) |
| |
| def test_simple(self) -> None: |
| """Test module graph with a simple module chain.""" |
| simpleA = SimpleA() |
| simpleB = SimpleB() |
| deps = DependencyManager([simpleA, simpleB]) |
| self.assertSetEqual({simpleB}, set(deps.blocked_modules.keys())) |
| self.assertSetEqual({simpleA}, deps.buildable_modules) |
| self.assertSetEqual({simpleA}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(simpleA) |
| self.assertSetEqual(set(), set(deps.blocked_modules.keys())) |
| self.assertSetEqual({simpleB}, deps.buildable_modules) |
| self.assertSetEqual({simpleB}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(simpleB) |
| |
| def test_complex(self) -> None: |
| """Test module graph with a more complex module chain.""" |
| complexA = ComplexA() |
| complexB = ComplexB() |
| complexC = ComplexC() |
| complexD = ComplexD() |
| deps = DependencyManager([complexA, complexB, complexC, complexD]) |
| self.assertSetEqual( |
| {complexB, complexC, complexD}, set(deps.blocked_modules.keys())) |
| self.assertSetEqual({complexA}, deps.buildable_modules) |
| self.assertSetEqual({complexA}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(complexA) |
| self.assertSetEqual({complexD}, set(deps.blocked_modules.keys())) |
| self.assertSetEqual({complexB, complexC}, deps.buildable_modules) |
| self.assertSetEqual({complexB, complexC}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(complexC) |
| self.assertSetEqual({complexD}, set(deps.blocked_modules.keys())) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| self.assertSetEqual(set(), deps.get_buildable()) |
| deps.complete(complexB) |
| self.assertSetEqual(set(), set(deps.blocked_modules.keys())) |
| self.assertSetEqual({complexD}, deps.buildable_modules) |
| self.assertSetEqual({complexD}, deps.get_buildable()) |
| self.assertSetEqual(set(), deps.buildable_modules) |
| deps.complete(complexD) |