Merge pull request #370 from digitalocean/fix-create-list

Account for list returning from create

Author: Zach Moody

pynetbox/core/endpoint.py

@@ -308,6 +308,8 @@ def create(self, *args, **kwargs):
             http_session=self.api.http_session,
         ).post(args[0] if args else kwargs)
 
+        if isinstance(req, list):
+            return [self.return_obj(i, self.api, self) for i in req]
         return self.return_obj(req, self.api, self)
 
     def choices(self):

tests/integration/test_dcim.py

@@ -90,10 +90,12 @@ def init(self, request, site):
 
     @pytest.fixture(scope="class")
     def add_sites(self, api):
-        sites = [
-            api.dcim.sites.create(name="test{}".format(i), slug="test{}".format(i))
-            for i in range(2, 20)
-        ]
+        sites = api.dcim.sites.create(
+            [
+                {"name": "test{}".format(i), "slug": "test{}".format(i)}
+                for i in range(2, 20)
+            ]
+        )
         yield
         for i in sites:
             i.delete()