T6948: Smoketest for DHCP lease sync

Author: indrajitr

smoketest/scripts/cli/test_service_dhcp-server.py

@@ -20,8 +20,13 @@
 from json import loads
 
 from base_vyostest_shim import VyOSUnitTestSHIM
+from time import sleep
 
+from vyos.hostsd_client import Client as hostsd_client
 from vyos.configsession import ConfigSessionError
+from vyos.kea import kea_add_lease
+from vyos.kea import kea_delete_lease
+from vyos.utils.process import run
 from vyos.utils.process import process_named_running
 from vyos.utils.file import read_file
 from vyos.template import inc_ip
@@ -842,6 +847,88 @@ def test_dhcp_on_interface_with_vrf(self):
         self.cli_delete(['vrf', 'name', 'SMOKE-DHCP'])
         self.cli_commit()
 
+    def test_dhcp_hostsd_lease_sync(self):
+        hc = hostsd_client()
+        shared_net_name = 'SMOKE-LEASE-SYNC'
+        domain_name = 'sync.private'
+
+        client_range = range(1, 4)
+        subnet_range_start = inc_ip(subnet, 10)
+        subnet_range_stop  = inc_ip(subnet, 20)
+
+        def local_cleanup():
+            print('Cleaning up started')
+            for seq in client_range:
+                ip_addr = inc_ip(subnet, seq)
+                kea_delete_lease(4, ip_addr)
+                # zmq.Socket(zmq.REQ) throws unclosed socket warning
+                # hc.delete_hosts([f'dhcp-server-{ip_addr}'])
+                # hc.apply()
+                run(f'/usr/bin/vyos-hostsd-client --delete-hosts --tag dhcp-server-{ip_addr} --apply')
+            print('Cleaning up ended')
+
+        self.addCleanup(local_cleanup)
+
+        pool = base_path + ['shared-network-name', shared_net_name, 'subnet', subnet]
+        self.cli_set(pool + ['subnet-id', '1'])
+        self.cli_set(pool + ['option', 'domain-name', domain_name])
+        self.cli_set(pool + ['range', '0', 'start', subnet_range_start])
+        self.cli_set(pool + ['range', '0', 'stop', subnet_range_stop])
+
+        # commit changes
+        self.cli_commit()
+
+        config = read_file(KEA4_CONF)
+        obj = loads(config)
+
+        self.verify_config_value(obj, ['Dhcp4', 'shared-networks'], 'name', shared_net_name)
+        self.verify_config_value(obj, ['Dhcp4', 'shared-networks', 0, 'subnet4'], 'subnet', subnet)
+        self.verify_config_value(obj, ['Dhcp4', 'shared-networks', 0, 'subnet4'], 'id', 1)
+
+        # Verify options
+        self.verify_config_object(
+                obj,
+                ['Dhcp4', 'shared-networks', 0, 'subnet4', 0, 'option-data'],
+                {'name': 'domain-name', 'data': domain_name})
+        self.verify_config_object(
+                obj,
+                ['Dhcp4', 'shared-networks', 0, 'subnet4', 0, 'pools'],
+                {'pool': f'{subnet_range_start} - {subnet_range_stop}'})
+
+        # Check for running process
+        self.assertTrue(process_named_running(PROCESS_NAME))
+
+        # All up and running, now test vyos-hostsd-client
+
+        # 1. Inject leases into kea
+        for seq in client_range:
+            client = f'client{seq}'
+            mac = f'00:50:00:00:00:{seq:02}'
+            ip = inc_ip(subnet, seq)
+            kea_add_lease(4, ip, host_name=client, mac_address=mac)
+
+        # 2. Verify that leases do not exist in hostsd state
+        for seq in client_range:
+            tag = f'dhcp-server-{inc_ip(subnet, seq)}'
+            print(f'Verifying tag: {tag}')
+            host_json = hc.get_hosts(tag)
+            self.assertFalse(host_json)
+
+        # 3. Restart the service to trigger vyos-hostsd-dhcp4-lease-sync.service
+        print('Restarting service in 30s')
+        sleep(30)
+        self.op_mode(['restart', 'dhcp', 'server'])
+
+        # 4. Wait a while and verify that leases are now in hostsd state
+        print('Waiting to verify in 30s')
+        sleep(30)
+
+        for seq in client_range:
+            tag = f'dhcp-server-{inc_ip(subnet, seq)}'
+            host_json = hc.get_hosts(tag)
+            self.assertTrue(host_json)
+
+        print('All done')
 
 if __name__ == '__main__':
     unittest.main(verbosity=2)