Merge pull request #85 from digitalocean/fix-circuit-term-repr

Fix circuit term repr
netbox-community · Aug 3, 2018 · 70b88e8 · 70b88e8
2 parents f450ebf + 111185e
commit 70b88e8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pynetbox/circuits.py b/pynetbox/circuits.py
@@ -20,3 +20,9 @@ class Circuits(Record):
 
     def __str__(self):
         return self.cid
+
+
+class CircuitTerminations(Record):
+
+    def __str__(self):
+        return self.circuit.cid
diff --git a/tests/test_circuits.py b/tests/test_circuits.py
@@ -111,3 +111,11 @@ class CircuitTypeTestCase(unittest.TestCase, GenericTest):
 
 class CircuitTerminationsTestCase(unittest.TestCase, GenericTest):
     name = 'circuit_terminations'
+
+    @patch(
+        'pynetbox.lib.query.requests.get',
+        return_value=Response(fixture='circuits/circuit_termination.json')
+    )
+    def test_repr(self, _):
+        test = nb.circuit_terminations.get(1)
+        self.assertEqual(str(test), '123456')