Support node-attributes in Azure driver

This is intended to be supported by all drivers, but we missed it
in the recent refactor.

Change-Id: Ib3f8c090b06c8458eb2f492db33e77a763aa64d0

doc/source/azure.rst

@@ -392,6 +392,12 @@ section of the configuration.
 
           A unique name within the provider for this pool of resources.
 
+       .. attr:: node-attributes
+          :type: dict
+
+          A dictionary of key-value pairs that will be stored with the node data
+          in ZooKeeper. The keys and values can be any arbitrary string.
+
        .. attr:: ipv4
           :type: bool
 

nodepool/driver/azure/config.py

@@ -166,6 +166,7 @@ class AzurePool(ConfigPool):
         self.load(pool_config)
 
     def load(self, pool_config):
+        super().load(pool_config)
         self.name = pool_config['name']
         self.max_servers = pool_config['max-servers']
         self.public_ipv4 = pool_config.get('public-ipv4',

nodepool/tests/fixtures/azure-diskimage.yaml

@@ -35,6 +35,9 @@ providers:
     pools:
       - name: main
         max-servers: 10
+        node-attributes:
+          key1: value1
+          key2: value2
         labels:
           - name: bionic
             diskimage: fake-image

nodepool/tests/fixtures/azure.yaml

@@ -36,6 +36,9 @@ providers:
     pools:
       - name: main
         max-servers: 10
+        node-attributes:
+          key1: value1
+          key2: value2
         labels:
           - name: bionic
             cloud-image: bionic
@@ -44,4 +47,4 @@ providers:
             tags:
               department: R&D
               team: DevOps
-              systemPurpose: CI
+              systemPurpose: CI

nodepool/tests/unit/test_driver_azure.py

@@ -54,6 +54,8 @@ class TestDriverAzure(tests.DBTestCase):
         self.assertEqual(node.state, zk.READY)
         self.assertIsNotNone(node.launcher)
         self.assertEqual(node.connection_type, 'ssh')
+        self.assertEqual(node.attributes,
+                         {'key1': 'value1', 'key2': 'value2'})
 
     def test_azure_diskimage(self):
         configfile = self.setup_config(
@@ -83,3 +85,5 @@ class TestDriverAzure(tests.DBTestCase):
         self.assertEqual(node.state, zk.READY)
         self.assertIsNotNone(node.launcher)
         self.assertEqual(node.connection_type, 'ssh')
+        self.assertEqual(node.attributes,
+                         {'key1': 'value1', 'key2': 'value2'})

releasenotes/notes/azure-node-attributes-8c3bb8bea3aea287.yaml

@@ -0,0 +1,4 @@
+---
+features:
+  - |
+    The node-attributes setting has been added to the Azure driver.