traffic rules: TestVLAN: vifRule

Signed-off-by: Sebastien Marie <semarie@kapouay.eu.org>

Author: semarie

tests/network/test_traffic_rules.py

@@ -2,9 +2,12 @@
 
 import pytest
 
+import logging
+
 from lib.common import safe_split
 from lib.host import Host
 from lib.network import Network
+from lib.vlan import VLAN
 from lib.vm import VM
 from lib.xo import xo_cli
 
@@ -276,5 +279,41 @@ def test_networkRule(self, connected_hosts_with_xo: list[Host], hostA2: Host, lo
                 'protocol': 'icmp',
             })
 
-        assert count_of(hostA1, hostBr) == 0, "1 OF after migrate (on hostA1)"
-        assert count_of(hostA2, hostBr) == 0, "1 OF after migrate (on hostA2)"
+        assert count_of(hostA1, hostBr) == 0, "no OF at end (on hostA1)"
+        assert count_of(hostA2, hostBr) == 0, "no OF at end (on hostA2)"
+
+@pytest.mark.complex_prerequisites
+@pytest.mark.small_vm
+@pytest.mark.parametrize("vlan_device", ["eth1"])
+@pytest.mark.parametrize("vlan_tag", [42])
+class TestVLAN:
+    def test_vifRule(self, connected_hosts_with_xo: list[Host], imported_vm: VM, empty_network: Network, vlan: VLAN):
+        host = connected_hosts_with_xo[0]
+        network = empty_network
+        vm = imported_vm.clone()
+
+        # get bridge of the network of the tagged PIF
+        hostBr = Network(host, vlan.tagged_PIF().network_uuid()).bridge()
+        logging.info(f"host bridge for vlan: {hostBr}")
+
+        try:
+            vif = vm.create_vif(1, network_uuid=network.uuid)
+
+            vm.start()
+
+            xo_cli('sdnController.addRule', {
+                'vifId': vif.uuid,
+                'ipRange': '0.0.0.0/0',
+                'direction': 'to',
+                'protocol': 'tcp',
+                'port': 'json:81',
+                'allow': 'true',
+            })
+
+            vm.wait_for_os_booted()
+
+            assert count_of(host, hostBr) == 1, "1 OF after started"
+        finally:
+            vm.destroy()
+
+        assert count_of(host, hostBr) == 0, "no OF at end"