NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-14 09:51:22 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Suppress deprecation warnings in ScriptTest

Browse Source
This commit is contained in:
Jeremy Stretch 2025-05-16 12:20:59 -04:00
parent f166e8d8ce
commit 3d6e6bc283
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
netbox/extras/tests/test_scripts.py
View File

@ -1,3 +1,4 @@
import logging
import tempfile import tempfile
from datetime import date, datetime, timezone from datetime import date, datetime, timezone
@ -7,6 +8,7 @@ from netaddr import IPAddress, IPNetwork
from dcim.models import DeviceRole from dcim.models import DeviceRole
from extras.scripts import * from extras.scripts import *
from utilities.testing import disable_logging
CHOICES = ( CHOICES = (
('ff0000', 'Red'), ('ff0000', 'Red'),
@ -39,7 +41,8 @@ class ScriptTest(TestCase):
datafile.write(bytes(YAML_DATA, 'UTF-8')) datafile.write(bytes(YAML_DATA, 'UTF-8'))
datafile.seek(0) datafile.seek(0)
data = Script().load_yaml(datafile.name) with disable_logging(level=logging.WARNING):
data = Script().load_yaml(datafile.name)
self.assertEqual(data, { self.assertEqual(data, {
'Foo': 123, 'Foo': 123,
'Bar': 456, 'Bar': 456,
@ -51,7 +54,8 @@ class ScriptTest(TestCase):
datafile.write(bytes(JSON_DATA, 'UTF-8')) datafile.write(bytes(JSON_DATA, 'UTF-8'))
datafile.seek(0) datafile.seek(0)
data = Script().load_json(datafile.name) with disable_logging(level=logging.WARNING):
data = Script().load_json(datafile.name)
self.assertEqual(data, { self.assertEqual(data, {
'Foo': 123, 'Foo': 123,
'Bar': 456, 'Bar': 456,