l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[modules/publicip] handle missing public ip more gracefully

Browse source 
If location does not throw, but reports an empty public IP, return
"n/a".

Since this caused a bug, also add a test for it.

fixes #853
This commit is contained in:
tobi-wan-kenobi 2022-02-14 14:58:01 +01:00
parent b1f49f6a1e
commit 3aadab5628
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/modules/contrib/test_publicip.py
View file

@ -26,6 +26,15 @@ class PublicIPTest(TestCase):
assert widget(module).full_text() == '5.12.220.2'
@mock.patch('util.location.public_ip')
def test_public_ip(self, public_ip_mock):
public_ip_mock.return_value = None
module = build_module()
module.update()
assert widget(module).full_text() == 'n/a'
@mock.patch('util.location.public_ip')
def test_public_ip_with_exception(self, public_ip_mock):
public_ip_mock.side_effect = Exception