Merge pull request #89 from digitalocean/fixes-config-context

Fixes config context

Author: Zach Moody

pynetbox/lib/response.py

@@ -18,6 +18,10 @@
 
 from pynetbox.lib.query import Request
 
+# List of fields that contain a dict but are not to be converted into
+# Record objects.
+JSON_FIELDS = ('custom_fields', 'data', 'config_context')
+
 
 def get_return(lookup, return_fields=None):
     '''Returns simple representations for items passed to lookup.
@@ -147,7 +151,8 @@ def _parse_values(self, values):
         values within.
         """
         for k, v in values.items():
-            if k != 'custom_fields':
+
+            if k not in JSON_FIELDS:
                 if isinstance(v, dict):
                     lookup = getattr(self.__class__, k, None)
                     if lookup:

tests/fixtures/dcim/device.json

@@ -65,5 +65,8 @@
     },
     "primary_ip6": null,
     "comments": "",
-    "custom_fields": {}
+    "custom_fields": {},
+    "config_context": {
+        "test_key": "test_val"
+    }
 }

tests/test_dcim.py

@@ -161,6 +161,8 @@ def test_get(self, mock):
         self.assertTrue(isinstance(ret, self.ret))
         self.assertTrue(isinstance(ret.primary_ip, pynetbox.ipam.IpAddresses))
         self.assertTrue(isinstance(ret.primary_ip4, pynetbox.ipam.IpAddresses))
+        self.assertTrue(isinstance(ret.config_context, dict))
+        self.assertTrue(isinstance(ret.custom_fields, dict))
         mock.assert_called_with(
             'http://localhost:8000/api/{}/{}/1/'.format(
                 self.app,