Forward port subordinate utils (#880)

* Add subordinate utilities for unit.Unit

Author: cderici

juju/application.py

@@ -64,6 +64,11 @@ def units(self):
             if unit.application == self.name
         ]
 
+    @property
+    def subordinate_units(self):
+        """Returns the subordinate units of this application"""
+        return [u for u in self.units if u.is_subordinate]
+
     @property
     def relations(self):
         return [rel for rel in self.model.relations if rel.matches(self.name)]

juju/model.py

@@ -169,6 +169,11 @@ def units(self):
         """
         return self._live_entity_map('unit')
 
+    @property
+    def subordinate_units(self):
+        """Return a map of unit-id:Unit for all subordinate units"""
+        return {u_name: u for u_name, u in self.units.items() if u.is_subordinate}
+
     @property
     def relations(self):
         """Return a map of relation-id:Relation for all relations currently in
@@ -1078,6 +1083,14 @@ def units(self):
         """
         return self.state.units
 
+    @property
+    def subordinate_units(self):
+        """Return a map of unit-id:Unit for all subordiante units currently in
+        the model.
+
+        """
+        return self.state.subordinate_units
+
     @property
     def relations(self):
         """Return a list of all Relations currently in the model.

juju/unit.py

@@ -26,6 +26,20 @@ def agent_status_since(self):
         """
         return pyrfc3339.parse(self.safe_data['agent-status']['since'])
 
+    @property
+    def is_subordinate(self):
+        """True if the unit is subordinate of another unit
+
+        """
+        return self.safe_data['subordinate']
+
+    @property
+    def principal_unit(self):
+        """Returns the name of the unit of which this unit is a subordinate to.
+        Returns '' for principal units themselves.
+        """
+        return self.safe_data['principal']
+
     @property
     def agent_status_message(self):
         """Get the agent status message.
@@ -77,6 +91,14 @@ def public_address(self):
     def tag(self):
         return tag.unit(self.name)
 
+    def get_subordinates(self):
+        """Returns the unit objects that are subordinates to this unit
+
+        :return [Unit]
+        """
+        return [u for u_name, u in self.model.units.items() if u.is_subordinate and
+                u.principal_unit == self.name]
+
     async def destroy(self):
         """Destroy this unit.
 

tests/integration/test_unit.py

@@ -255,3 +255,33 @@ async def test_unit_introspect(event_loop):
                            series='jammy',
                            to='0',
                            )
+
+
+@base.bootstrapped
+@pytest.mark.asyncio
+async def test_subordinate_units(event_loop):
+    async with base.CleanModel() as model:
+        u_app = await model.deploy('ubuntu')
+        n_app = await model.deploy('ntp')
+        await model.relate('ubuntu', 'ntp')
+        await model.wait_for_idle()
+
+        # model subordinates
+        model_subs = model.subordinate_units
+        assert len(model_subs) == 1
+        assert 'ntp/0' in model_subs
+        assert 'ubuntu/0' not in model_subs
+
+        n_unit = model_subs['ntp/0']
+        u_unit = u_app.units[0]
+
+        # application subordinates
+        app_sub_names = [u.name for u in n_app.subordinate_units]
+        assert n_unit.name in app_sub_names
+        assert u_unit.name not in app_sub_names
+
+        assert n_unit.is_subordinate
+        assert not u_unit.is_subordinate
+        assert n_unit.principal_unit == 'ubuntu/0'
+        assert u_unit.principal_unit == ''
+        assert [u.name for u in u_unit.get_subordinates()] == [n_unit.name]