mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fixes #3937: Suppress warning messages in tests for requests expected to yield a 4XX response

Browse Source
This commit is contained in:
Jeremy Stretch
2020-01-17 14:53:33 -05:00
parent 439fa731ba
commit 302f87e108
4 changed files with 30 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/utilities/testing.py
View File

@ -1,3 +1,6 @@
import logging
from contextlib import contextmanager
from django.contrib.auth.models import Permission, User
from rest_framework.test import APITestCase as _APITestCase
@ -62,3 +65,15 @@ def choices_to_dict(choices_list):
return {
choice['value']: choice['label'] for choice in choices_list
}
@contextmanager
def disable_warnings(logger_name):
"""
Temporarily suppress expected warning messages to keep the test output clean.
"""
logger = logging.getLogger(logger_name)
current_level = logger.level
logger.setLevel(logging.ERROR)
yield
logger.setLevel(current_level)