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

Improve network traffic module tests

Browse source
This commit is contained in:
Thaynã Moretti 2020-09-01 22:12:54 -03:00
parent 1b66e17780
commit 032a651efa
2 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
bumblebee_status/modules/contrib/network_traffic.py
View file

@ -97,9 +97,6 @@ class BandwidthInfo(object):
"""Return default active network adapter""" """Return default active network adapter"""
gateway = netifaces.gateways()["default"] gateway = netifaces.gateways()["default"]
if not gateway:
raise "No default gateway found"
return gateway[netifaces.AF_INET][1] return gateway[netifaces.AF_INET][1]
@classmethod @classmethod

12
tests/modules/contrib/test_network_traffic.py
View file

@ -89,3 +89,15 @@ class TestNetworkTrafficUnit(TestCase):
assert download_widget(module).full_text() == '30.00MiB/s' assert download_widget(module).full_text() == '30.00MiB/s'
assert upload_widget(module).full_text() == '512.00KiB/s' assert upload_widget(module).full_text() == '512.00KiB/s'
def test_widget_states(self):
module = build_module()
assert module.state(download_widget(module)) == 'rx'
assert module.state(upload_widget(module)) == 'tx'
def test_invalid_widget_state(self):
module = build_module()
invalid_widget = core.widget.Widget(name='invalid')
assert module.state(invalid_widget) == None