blob: 1d594be40a91866da5112ccd1e85185f607189b1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
# Copyright (c) 2024 Arista Networks, Inc.
# Use of this source code is governed by the Apache License 2.0
# that can be found in the LICENSE file.
"""Script that merge a collection of catalogs into one AntaCatalog."""
from pathlib import Path
from anta.catalog import AntaCatalog
from anta.models import AntaTest
CATALOG_SUFFIX = "-catalog.yml"
CATALOG_DIR = "intended/test_catalogs/"
if __name__ == "__main__":
catalogs = []
for file in Path(CATALOG_DIR).glob("*" + CATALOG_SUFFIX):
device = str(file).removesuffix(CATALOG_SUFFIX).removeprefix(CATALOG_DIR)
print(f"Loading test catalog for device {device}")
catalog = AntaCatalog.parse(file)
# Add the device name as a tag to all tests in the catalog
for test in catalog.tests:
test.inputs.filters = AntaTest.Input.Filters(tags={device})
catalogs.append(catalog)
# Merge all catalogs
merged_catalog = AntaCatalog.merge_catalogs(catalogs)
# Save the merged catalog to a file
with Path("anta-catalog.yml").open("w") as f:
f.write(merged_catalog.dump().yaml())
|
