| Mike Frysinger | f1ba7ad | 2022-09-12 05:42:57 -0400 | [diff] [blame] | 1 | # Copyright 2019 The ChromiumOS Authors |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """Test cros_extract_deps.""" |
| 6 | |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 7 | from chromite.lib import cros_test_lib |
| 8 | from chromite.scripts import cros_extract_deps |
| 9 | |
| Mike Frysinger | 807d828 | 2022-04-28 22:45:17 -0400 | [diff] [blame] | 10 | |
| Greg Edelston | a4c9b3b | 2020-01-07 17:51:13 -0700 | [diff] [blame] | 11 | pytestmark = cros_test_lib.pytestmark_inside_only |
| 12 | |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 13 | |
| 14 | class FlattenDepTreeTest(cros_test_lib.TestCase): |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 15 | """Tests for cros_extract_deps.FlattenDepTree.""" |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 16 | |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 17 | def testFlattenDepTreeSimple(self): |
| 18 | dep_tree = { |
| 19 | "deathstar/darthvader-2.3": { |
| 20 | "action": "merge", |
| 21 | "deps": { |
| 22 | "deathstar/trooper-1.2": { |
| 23 | "action": "merge", |
| 24 | "deps": { |
| 25 | "weapon/blasterpistol-2.1": { |
| 26 | "action": "merge", |
| 27 | "deps": {}, |
| 28 | "deptype": "runtime", |
| 29 | } |
| 30 | }, |
| 31 | "deptype": "runtime", |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 32 | }, |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 33 | "deathstar/pilot-2.3": { |
| 34 | "action": "merge", |
| 35 | "deps": {}, |
| 36 | "deptype": "runtime", |
| 37 | }, |
| 38 | "deathstar/commander-2.3": { |
| 39 | "action": "merge", |
| 40 | "deps": {}, |
| 41 | "deptype": "runtime", |
| 42 | }, |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 43 | }, |
| 44 | }, |
| Ned Nguyen | 3bbd207 | 2019-01-28 19:33:41 -0700 | [diff] [blame] | 45 | } |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 46 | flatten_dep_tree = { |
| 47 | "weapon/blasterpistol-2.1": { |
| 48 | "rev_deps": ["deathstar/trooper-1.2"], |
| 49 | "category": "weapon", |
| 50 | "version": "2.1", |
| 51 | "name": "blasterpistol", |
| 52 | "deps": [], |
| 53 | "action": "merge", |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 54 | "full_name": "weapon/blasterpistol-2.1", |
| 55 | }, |
| 56 | "deathstar/darthvader-2.3": { |
| 57 | "rev_deps": [], |
| 58 | "category": "deathstar", |
| 59 | "version": "2.3", |
| 60 | "name": "darthvader", |
| 61 | "deps": [ |
| 62 | "deathstar/commander-2.3", |
| 63 | "deathstar/pilot-2.3", |
| 64 | "deathstar/trooper-1.2", |
| 65 | ], |
| 66 | "action": "merge", |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 67 | "full_name": "deathstar/darthvader-2.3", |
| 68 | }, |
| 69 | "deathstar/pilot-2.3": { |
| 70 | "rev_deps": ["deathstar/darthvader-2.3"], |
| 71 | "category": "deathstar", |
| 72 | "version": "2.3", |
| 73 | "name": "pilot", |
| 74 | "deps": [], |
| 75 | "action": "merge", |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 76 | "full_name": "deathstar/pilot-2.3", |
| 77 | }, |
| 78 | "deathstar/commander-2.3": { |
| 79 | "rev_deps": ["deathstar/darthvader-2.3"], |
| 80 | "category": "deathstar", |
| 81 | "version": "2.3", |
| 82 | "name": "commander", |
| 83 | "deps": [], |
| 84 | "action": "merge", |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 85 | "full_name": "deathstar/commander-2.3", |
| 86 | }, |
| 87 | "deathstar/trooper-1.2": { |
| 88 | "rev_deps": ["deathstar/darthvader-2.3"], |
| 89 | "category": "deathstar", |
| 90 | "version": "1.2", |
| 91 | "name": "trooper", |
| 92 | "deps": ["weapon/blasterpistol-2.1"], |
| 93 | "action": "merge", |
| Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 94 | "full_name": "deathstar/trooper-1.2", |
| 95 | }, |
| 96 | } |
| 97 | self.assertEqual( |
| 98 | cros_extract_deps.FlattenDepTree(dep_tree), flatten_dep_tree |
| 99 | ) |