Merge pull request #238 from minitriga/elevation

Rack Elevation Endpoint

Author: Zach Moody

pynetbox/models/dcim.py

@@ -119,6 +119,23 @@ def units(self):
         """
         return RODetailEndpoint(self, "units", custom_return=RUs)
 
+    @property
+    def elevation(self):
+        """ Represents the ``elevation`` detail endpoint.
+
+        Returns a DetailEndpoint object that is the interface for
+        viewing response from the elevation endpoint updated in Netbox version 2.8.
+
+        :returns: :py:class:`.DetailEndpoint`
+
+        :Examples:
+
+        >>> rack = nb.dcim.racks.get(123)
+        >>> rack.elevation.list()
+        {"get_facts": {"interface_list": ["ge-0/0/0"]}}
+
+        """
+        return RODetailEndpoint(self, "elevation", custom_return=RUs)
 
 class Termination(Record):
     def __str__(self):

tests/test_dcim.py

@@ -401,6 +401,28 @@ def test_get_units(self, mock):
             isinstance(ret[0].device, pynetbox.models.dcim.Devices)
         )
 
+    @patch(
+        'pynetbox.core.query.requests.sessions.Session.get',
+        side_effect=[
+            Response(fixture='dcim/rack.json'),
+            Response(fixture='dcim/rack_u.json'),
+        ]
+    )
+    def test_get_elevation(self, mock):
+        test = nb.racks.get(1)
+        ret = test.elevation.list()
+        mock.assert_called_with(
+            'http://localhost:8000/api/dcim/racks/1/elevation/',
+            params={},
+            json=None,
+            headers=HEADERS,
+            verify=True,
+        )
+        self.assertTrue(ret)
+        self.assertTrue(
+            isinstance(ret[0].device, pynetbox.models.dcim.Devices)
+        )
+
 
 class RackRoleTestCase(Generic.Tests):
     name = 'rack_roles'