From ec0d9026128c707d9b1dbb3965983d0e06126182 Mon Sep 17 00:00:00 2001 From: Andrew Turner Date: Thu, 10 Oct 2024 11:31:08 -0500 Subject: [PATCH] fix black --- pynautobot/core/endpoint.py | 1 + tests/unit/test_endpoint.py | 1 + 2 files changed, 2 insertions(+) diff --git a/pynautobot/core/endpoint.py b/pynautobot/core/endpoint.py index 7b730450..14b4109d 100644 --- a/pynautobot/core/endpoint.py +++ b/pynautobot/core/endpoint.py @@ -743,6 +743,7 @@ def run_and_wait(self, *args, api_version=None, interval=5, max_rechecks=50, **k raise ValueError("Did not receieve completed job result for job.") + class GraphqlEndpoint(Endpoint): """Extend Endpoint class to support run method for graphql queries.""" diff --git a/tests/unit/test_endpoint.py b/tests/unit/test_endpoint.py index 27599c1a..b4fb437f 100644 --- a/tests/unit/test_endpoint.py +++ b/tests/unit/test_endpoint.py @@ -294,6 +294,7 @@ def test_run_and_wait_no_complete(self, mock_post, mock_get): with self.assertRaises(ValueError): test_obj.run_and_wait(job_id="test", interval=1, max_rechecks=2) + class GraphqlEndPointTestCase(unittest.TestCase): def test_invalid_arg(self): with self.assertRaises(