feat: Add Infrahub Jinja2 transforms for VLANs, interfaces, and VXLAN (#20)
Add three GraphQL queries, Jinja2 templates, and integration tests for generating YANG-style JSON configuration payloads from Infrahub intent data, suitable for gNMI Set operations on Arista EOS devices. Queries (transforms/queries/): - vlan_intent.gql: Fetches VLANs for a device via VTEP mappings and SVI interfaces, including VNI associations. - interface_intent.gql: Fetches all interface types (loopback, ethernet, vlan, lag) with IP addresses and type-specific attributes. - vxlan_intent.gql: Fetches VTEP config, VLAN-to-VNI mappings, and VRF-to-VNI mappings (L3VNI) via VRF device assignments. Templates (transforms/templates/): - vlan_yang.j2: Merges VLANs from both VTEP and SVI sources, deduplicates by vlan_id, and emits a sorted JSON array. - interface_yang.j2: Emits a JSON array of interfaces sorted by name, with a "type" discriminator field for each interface kind. - vxlan_yang.j2: Emits a JSON object with vtep config, vlan_vni_mappings, and vrf_vni_mappings sections. Integration tests (transforms/tests/): - One test directory per transform with input.json (sample GraphQL response for leaf1), output.json (expected result), and test.yml config. - Test data reflects the lab topology: leaf1 VTEP 10.0.255.11, VLAN 40 / VNI 110040, MLAG VLANs 4090/4091, underlay Ethernet11/12. .infrahub.yml updated with queries and jinja2_transforms sections. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Damien
350
transforms/tests/interface_yang/input.json
Normal file
350
transforms/tests/interface_yang/input.json
Normal file
@@ -0,0 +1,350 @@
|
||||
{
|
||||
"data": {
|
||||
"InfraInterfaceLoopback": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Loopback0"
|
||||
},
|
||||
"description": {
|
||||
"value": "Router-ID"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"address": {
|
||||
"value": "10.0.250.11/32"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Loopback1"
|
||||
},
|
||||
"description": {
|
||||
"value": "VTEP"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"address": {
|
||||
"value": "10.0.255.11/32"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"InfraInterfaceEthernet": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet1"
|
||||
},
|
||||
"description": {
|
||||
"value": "host1"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"speed": {
|
||||
"value": null
|
||||
},
|
||||
"mode": {
|
||||
"value": "trunk"
|
||||
},
|
||||
"lag": {
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Port-Channel1"
|
||||
}
|
||||
}
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet10"
|
||||
},
|
||||
"description": {
|
||||
"value": "mlag peer link"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"speed": {
|
||||
"value": null
|
||||
},
|
||||
"mode": {
|
||||
"value": "trunk"
|
||||
},
|
||||
"lag": {
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Port-Channel999"
|
||||
}
|
||||
}
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet11"
|
||||
},
|
||||
"description": {
|
||||
"value": "spine1"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": 9214
|
||||
},
|
||||
"speed": {
|
||||
"value": null
|
||||
},
|
||||
"mode": {
|
||||
"value": "routed"
|
||||
},
|
||||
"lag": null,
|
||||
"ip_addresses": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"address": {
|
||||
"value": "10.0.1.1/31"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet12"
|
||||
},
|
||||
"description": {
|
||||
"value": "spine2"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": 9214
|
||||
},
|
||||
"speed": {
|
||||
"value": null
|
||||
},
|
||||
"mode": {
|
||||
"value": "routed"
|
||||
},
|
||||
"lag": null,
|
||||
"ip_addresses": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"address": {
|
||||
"value": "10.0.2.1/31"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"InfraInterfaceVlan": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Vlan4090"
|
||||
},
|
||||
"description": {
|
||||
"value": "MLAG Peer-Link"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"virtual_router_address": {
|
||||
"value": null
|
||||
},
|
||||
"autostate": {
|
||||
"value": false
|
||||
},
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 4090
|
||||
}
|
||||
}
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"address": {
|
||||
"value": "10.255.255.0/31"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Vlan4091"
|
||||
},
|
||||
"description": {
|
||||
"value": "MLAG iBGP Peering"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": 9214
|
||||
},
|
||||
"virtual_router_address": {
|
||||
"value": null
|
||||
},
|
||||
"autostate": {
|
||||
"value": true
|
||||
},
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 4091
|
||||
}
|
||||
}
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"InfraInterfaceLag": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Port-Channel1"
|
||||
},
|
||||
"description": {
|
||||
"value": "host1"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"lacp_mode": {
|
||||
"value": "active"
|
||||
},
|
||||
"mlag_id": {
|
||||
"value": 1
|
||||
},
|
||||
"members": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet1"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Port-Channel999"
|
||||
},
|
||||
"description": {
|
||||
"value": "MLAG Peer"
|
||||
},
|
||||
"enabled": {
|
||||
"value": true
|
||||
},
|
||||
"mtu": {
|
||||
"value": null
|
||||
},
|
||||
"lacp_mode": {
|
||||
"value": "active"
|
||||
},
|
||||
"mlag_id": {
|
||||
"value": null
|
||||
},
|
||||
"members": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Ethernet10"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"ip_addresses": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
106
transforms/tests/interface_yang/output.json
Normal file
106
transforms/tests/interface_yang/output.json
Normal file
@@ -0,0 +1,106 @@
|
||||
[
|
||||
{
|
||||
"type": "ethernet",
|
||||
"name": "Ethernet1",
|
||||
"description": "host1",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"speed": null,
|
||||
"mode": "trunk",
|
||||
"lag": "Port-Channel1",
|
||||
"ip_addresses": []
|
||||
},
|
||||
{
|
||||
"type": "ethernet",
|
||||
"name": "Ethernet10",
|
||||
"description": "mlag peer link",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"speed": null,
|
||||
"mode": "trunk",
|
||||
"lag": "Port-Channel999",
|
||||
"ip_addresses": []
|
||||
},
|
||||
{
|
||||
"type": "ethernet",
|
||||
"name": "Ethernet11",
|
||||
"description": "spine1",
|
||||
"enabled": true,
|
||||
"mtu": 9214,
|
||||
"speed": null,
|
||||
"mode": "routed",
|
||||
"lag": null,
|
||||
"ip_addresses": ["10.0.1.1/31"]
|
||||
},
|
||||
{
|
||||
"type": "ethernet",
|
||||
"name": "Ethernet12",
|
||||
"description": "spine2",
|
||||
"enabled": true,
|
||||
"mtu": 9214,
|
||||
"speed": null,
|
||||
"mode": "routed",
|
||||
"lag": null,
|
||||
"ip_addresses": ["10.0.2.1/31"]
|
||||
},
|
||||
{
|
||||
"type": "loopback",
|
||||
"name": "Loopback0",
|
||||
"description": "Router-ID",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"ip_addresses": ["10.0.250.11/32"]
|
||||
},
|
||||
{
|
||||
"type": "loopback",
|
||||
"name": "Loopback1",
|
||||
"description": "VTEP",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"ip_addresses": ["10.0.255.11/32"]
|
||||
},
|
||||
{
|
||||
"type": "lag",
|
||||
"name": "Port-Channel1",
|
||||
"description": "host1",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"lacp_mode": "active",
|
||||
"mlag_id": 1,
|
||||
"members": ["Ethernet1"],
|
||||
"ip_addresses": []
|
||||
},
|
||||
{
|
||||
"type": "lag",
|
||||
"name": "Port-Channel999",
|
||||
"description": "MLAG Peer",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"lacp_mode": "active",
|
||||
"mlag_id": null,
|
||||
"members": ["Ethernet10"],
|
||||
"ip_addresses": []
|
||||
},
|
||||
{
|
||||
"type": "vlan",
|
||||
"name": "Vlan4090",
|
||||
"description": "MLAG Peer-Link",
|
||||
"enabled": true,
|
||||
"mtu": null,
|
||||
"vlan_id": 4090,
|
||||
"virtual_router_address": null,
|
||||
"autostate": false,
|
||||
"ip_addresses": ["10.255.255.0/31"]
|
||||
},
|
||||
{
|
||||
"type": "vlan",
|
||||
"name": "Vlan4091",
|
||||
"description": "MLAG iBGP Peering",
|
||||
"enabled": true,
|
||||
"mtu": 9214,
|
||||
"vlan_id": 4091,
|
||||
"virtual_router_address": null,
|
||||
"autostate": true,
|
||||
"ip_addresses": []
|
||||
}
|
||||
]
|
||||
13
transforms/tests/interface_yang/test.yml
Normal file
13
transforms/tests/interface_yang/test.yml
Normal file
@@ -0,0 +1,13 @@
|
||||
---
|
||||
# Integration test for interface_yang_transform
|
||||
# Usage: infrahubctl test transforms/tests/interface_yang/test.yml
|
||||
apiVersion: infrahub.app/v1
|
||||
kind: IntegrationTest
|
||||
spec:
|
||||
kind: jinja2-transform-integration
|
||||
name: interface_yang_transform_leaf1
|
||||
transform: interface_yang_transform
|
||||
params:
|
||||
device_name: leaf1
|
||||
input: input.json
|
||||
output: output.json
|
||||
116
transforms/tests/vlan_yang/input.json
Normal file
116
transforms/tests/vlan_yang/input.json
Normal file
@@ -0,0 +1,116 @@
|
||||
{
|
||||
"data": {
|
||||
"InfraVTEP": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"source_address": {
|
||||
"value": "10.0.255.11"
|
||||
},
|
||||
"udp_port": {
|
||||
"value": 4789
|
||||
},
|
||||
"vlan_vni_mappings": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 40
|
||||
},
|
||||
"name": {
|
||||
"value": "test-l2-vxlan"
|
||||
},
|
||||
"status": {
|
||||
"value": "active"
|
||||
},
|
||||
"vlan_type": {
|
||||
"value": "standard"
|
||||
},
|
||||
"trunk_groups": {
|
||||
"value": []
|
||||
},
|
||||
"stp_enabled": {
|
||||
"value": true
|
||||
},
|
||||
"vni": {
|
||||
"node": {
|
||||
"vni": {
|
||||
"value": 110040
|
||||
},
|
||||
"vni_type": {
|
||||
"value": "l2vni"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"InfraInterfaceVlan": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 4090
|
||||
},
|
||||
"name": {
|
||||
"value": "mlag-peer"
|
||||
},
|
||||
"status": {
|
||||
"value": "active"
|
||||
},
|
||||
"vlan_type": {
|
||||
"value": "mlag_peer"
|
||||
},
|
||||
"trunk_groups": {
|
||||
"value": ["mlag-peer"]
|
||||
},
|
||||
"stp_enabled": {
|
||||
"value": false
|
||||
},
|
||||
"vni": null
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"node": {
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 4091
|
||||
},
|
||||
"name": {
|
||||
"value": "mlag-ibgp"
|
||||
},
|
||||
"status": {
|
||||
"value": "active"
|
||||
},
|
||||
"vlan_type": {
|
||||
"value": "mlag_ibgp"
|
||||
},
|
||||
"trunk_groups": {
|
||||
"value": ["mlag-peer"]
|
||||
},
|
||||
"stp_enabled": {
|
||||
"value": false
|
||||
},
|
||||
"vni": null
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
32
transforms/tests/vlan_yang/output.json
Normal file
32
transforms/tests/vlan_yang/output.json
Normal file
@@ -0,0 +1,32 @@
|
||||
[
|
||||
{
|
||||
"vlan_id": 40,
|
||||
"name": "test-l2-vxlan",
|
||||
"status": "ACTIVE",
|
||||
"vlan_type": "standard",
|
||||
"trunk_groups": [],
|
||||
"stp_enabled": true,
|
||||
"vni": 110040,
|
||||
"vni_type": "l2vni"
|
||||
},
|
||||
{
|
||||
"vlan_id": 4090,
|
||||
"name": "mlag-peer",
|
||||
"status": "ACTIVE",
|
||||
"vlan_type": "mlag_peer",
|
||||
"trunk_groups": ["mlag-peer"],
|
||||
"stp_enabled": false,
|
||||
"vni": null,
|
||||
"vni_type": null
|
||||
},
|
||||
{
|
||||
"vlan_id": 4091,
|
||||
"name": "mlag-ibgp",
|
||||
"status": "ACTIVE",
|
||||
"vlan_type": "mlag_ibgp",
|
||||
"trunk_groups": ["mlag-peer"],
|
||||
"stp_enabled": false,
|
||||
"vni": null,
|
||||
"vni_type": null
|
||||
}
|
||||
]
|
||||
13
transforms/tests/vlan_yang/test.yml
Normal file
13
transforms/tests/vlan_yang/test.yml
Normal file
@@ -0,0 +1,13 @@
|
||||
---
|
||||
# Integration test for vlan_yang_transform
|
||||
# Usage: infrahubctl test transforms/tests/vlan_yang/test.yml
|
||||
apiVersion: infrahub.app/v1
|
||||
kind: IntegrationTest
|
||||
spec:
|
||||
kind: jinja2-transform-integration
|
||||
name: vlan_yang_transform_leaf1
|
||||
transform: vlan_yang_transform
|
||||
params:
|
||||
device_name: leaf1
|
||||
input: input.json
|
||||
output: output.json
|
||||
62
transforms/tests/vxlan_yang/input.json
Normal file
62
transforms/tests/vxlan_yang/input.json
Normal file
@@ -0,0 +1,62 @@
|
||||
{
|
||||
"data": {
|
||||
"InfraVTEP": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"source_address": {
|
||||
"value": "10.0.255.11"
|
||||
},
|
||||
"udp_port": {
|
||||
"value": 4789
|
||||
},
|
||||
"learn_restrict": {
|
||||
"value": "any"
|
||||
},
|
||||
"source_interface": {
|
||||
"node": {
|
||||
"name": {
|
||||
"value": "Loopback1"
|
||||
}
|
||||
}
|
||||
},
|
||||
"vlan_vni_mappings": {
|
||||
"edges": [
|
||||
{
|
||||
"node": {
|
||||
"description": {
|
||||
"value": "VLAN 40 <-> VNI 110040"
|
||||
},
|
||||
"vlan": {
|
||||
"node": {
|
||||
"vlan_id": {
|
||||
"value": 40
|
||||
},
|
||||
"name": {
|
||||
"value": "test-l2-vxlan"
|
||||
}
|
||||
}
|
||||
},
|
||||
"vni": {
|
||||
"node": {
|
||||
"vni": {
|
||||
"value": 110040
|
||||
},
|
||||
"vni_type": {
|
||||
"value": "l2vni"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"InfraVRFDeviceAssignment": {
|
||||
"edges": []
|
||||
}
|
||||
}
|
||||
}
|
||||
17
transforms/tests/vxlan_yang/output.json
Normal file
17
transforms/tests/vxlan_yang/output.json
Normal file
@@ -0,0 +1,17 @@
|
||||
{
|
||||
"vtep": {
|
||||
"source_address": "10.0.255.11",
|
||||
"source_interface": "Loopback1",
|
||||
"udp_port": 4789,
|
||||
"learn_restrict": "any",
|
||||
"vlan_vni_mappings": [
|
||||
{
|
||||
"vlan_id": 40,
|
||||
"vlan_name": "test-l2-vxlan",
|
||||
"vni": 110040,
|
||||
"vni_type": "l2vni"
|
||||
}
|
||||
],
|
||||
"vrf_vni_mappings": []
|
||||
}
|
||||
}
|
||||
13
transforms/tests/vxlan_yang/test.yml
Normal file
13
transforms/tests/vxlan_yang/test.yml
Normal file
@@ -0,0 +1,13 @@
|
||||
---
|
||||
# Integration test for vxlan_yang_transform
|
||||
# Usage: infrahubctl test transforms/tests/vxlan_yang/test.yml
|
||||
apiVersion: infrahub.app/v1
|
||||
kind: IntegrationTest
|
||||
spec:
|
||||
kind: jinja2-transform-integration
|
||||
name: vxlan_yang_transform_leaf1
|
||||
transform: vxlan_yang_transform
|
||||
params:
|
||||
device_name: leaf1
|
||||
input: input.json
|
||||
output: output.json
|
||||
Reference in New Issue
Block a user