From 8f8141875388e49c50e09a3ec3b6f3fbb3fd1286 Mon Sep 17 00:00:00 2001
From: Arthur <worldnomad@gmail.com>
Date: Wed, 29 May 2024 12:56:07 -0700
Subject: [PATCH] 16261 fixup test

---
 netbox/ipam/tests/test_api.py   | 2 +-
 netbox/utilities/testing/api.py | 9 ++++++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/netbox/ipam/tests/test_api.py b/netbox/ipam/tests/test_api.py
index 20ba35c6b..2cf7a2f1c 100644
--- a/netbox/ipam/tests/test_api.py
+++ b/netbox/ipam/tests/test_api.py
@@ -649,7 +649,7 @@ class IPAddressTest(APIViewTestCases.APIViewTestCase):
         'description': 'New description',
     }
     graphql_filter = {
-        'address': '192.168.0.1/24',
+        'address': {'lookup': 'i_exact', 'value': '192.168.0.1/24'},
     }
 
     @classmethod
diff --git a/netbox/utilities/testing/api.py b/netbox/utilities/testing/api.py
index 019d6e6ca..b7e9f9c7e 100644
--- a/netbox/utilities/testing/api.py
+++ b/netbox/utilities/testing/api.py
@@ -496,7 +496,14 @@ class APIViewTestCases:
             Create a filtered query: i.e. ip_address_list(filters: {address: "1.1.1.1/24"}){.
             """
             if filters:
-                filter_string = ', '.join(f'{k}: "{v}"' for k, v in filters.items())
+                for field_name, params in filters.items():
+                    lookup = params['lookup']
+                    value = params['value']
+                    if lookup:
+                        query = f'{{{lookup}: "{value}"}}'
+                        filter_string = f'{field_name}: {query}'
+                    else:
+                        filter_string = f'{field_name}: "{value}"'
                 filter_string = f'(filters: {{{filter_string}}})'
             else:
                 filter_string = ''